1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
import os
import yaml
from gevent.lock import Semaphore
from gevent.server import StreamServer
class TarantoolInspector(StreamServer):
"""
Tarantool inspector daemon. Usage:
inspector = TarantoolInspector('localhost', 8080)
inspector.start()
# run some tests
inspector.stop()
"""
def __init__(self, host, port):
super(TarantoolInspector, self).__init__((host, port))
self.parser = None
def start(self):
super(TarantoolInspector, self).start()
os.environ['INSPECTOR'] = str(self.server_port)
def stop(self):
del os.environ['INSPECTOR']
def set_parser(self, parser):
self.parser = parser
self.sem = Semaphore()
@staticmethod
def readline(socket, delimiter='\n', size=4096):
result = ''
data = True
while data:
try:
data = socket.recv(size)
except IOError:
# catch instance halt connection refused errors
data = ''
result += data
while result.find(delimiter) != -1:
line, result = result.split(delimiter, 1)
yield line
return
def handle(self, socket, addr):
if self.parser is None:
raise AttributeError('Parser is not defined')
self.sem.acquire()
for line in self.readline(socket):
try:
result = self.parser.parse_preprocessor(line)
except Exception, e:
print('error', e)
import traceback
traceback.print_exc()
result = { "error": repr(e) }
if result == None:
result = True
result = yaml.dump(result)
if not result.endswith('...\n'):
result = result + '...\n'
socket.sendall(result)
self.sem.release()
|
