def setup(self):
self.dbwriter = DBWriter(self.server.results_file_path)
class LogCollector(SocketServer.BaseRequestHandler):
def setup(self):
self.dbwriter = DBWriter(self.server.results_file_path)
def handle(self):
str_message = self.request[0].strip()
dic_message = json.loads(str_message)
dic_payload = dic_message.get('payload', {})
if dic_message.get('dataType') == 'transactionStart':
self.dbwriter.write_transaction_start(dic_payload)
elif dic_message.get('dataType') == 'transactionEnd':
self.dbwriter.write_transaction_end(dic_payload)
elif dic_message.get('dataType') == 'request':
self.dbwriter.write_request(dic_payload)
elif dic_message.get('dataType') == 'spawns':
self.dbwriter.write_spawns_info(dic_payload)
elif dic_message.get('dataType') == 'sysmonitor':
self.dbwriter.write_sysmonitor_info(dic_payload)
elif dic_message.get('dataType') == 'message':
self.dbwriter.write_message(dic_payload)
