mirror of https://github.com/MISP/misp-modules
Improve tornado parallel
parent
6ba2731eb5
commit
d6388e1c52
|
@ -166,15 +166,23 @@ class ListModules(tornado.web.RequestHandler):
|
||||||
self.write(json.dumps(ret))
|
self.write(json.dumps(ret))
|
||||||
|
|
||||||
|
|
||||||
|
@tornado.gen.coroutine
|
||||||
|
def async_module(request, write_fct):
|
||||||
|
jsonpayload = request.body.decode('utf-8')
|
||||||
|
x = json.loads(jsonpayload)
|
||||||
|
log.debug('MISP QueryModule request {0}'.format(jsonpayload))
|
||||||
|
ret = mhandlers[x['module']].handler(q=jsonpayload)
|
||||||
|
write_fct(json.dumps(ret))
|
||||||
|
|
||||||
|
|
||||||
class QueryModule(tornado.web.RequestHandler):
|
class QueryModule(tornado.web.RequestHandler):
|
||||||
@tornado.gen.coroutine
|
@tornado.gen.coroutine
|
||||||
def post(self):
|
def post(self):
|
||||||
global mhandlers
|
global mhandlers
|
||||||
jsonpayload = self.request.body.decode('utf-8')
|
try:
|
||||||
x = json.loads(jsonpayload)
|
yield async_module(self.request, self.write)
|
||||||
log.debug('MISP QueryModule request {0}'.format(jsonpayload))
|
except Exception:
|
||||||
ret = mhandlers[x['module']].handler(q=jsonpayload)
|
log.exception("Someting bad happened.")
|
||||||
self.write(json.dumps(ret))
|
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
|
|
Loading…
Reference in New Issue