mirror of
https://github.com/newnius/YAO-agent.git
synced 2025-06-06 05:21:55 +00:00
upgrade to multithreading http service
This commit is contained in:
parent
f43af24ecd
commit
b5ff2de041
7
agent.py
7
agent.py
@ -11,6 +11,7 @@ import multiprocessing
|
||||
import psutil
|
||||
import math
|
||||
from http.server import BaseHTTPRequestHandler, HTTPServer
|
||||
from socketserver import ThreadingMixIn
|
||||
import cgi
|
||||
import docker
|
||||
from urllib import parse
|
||||
@ -276,6 +277,10 @@ class MyHandler(BaseHTTPRequestHandler):
|
||||
self.send_error(404, 'File Not Found: %s' % self.path)
|
||||
|
||||
|
||||
class ThreadingSimpleServer(ThreadingMixIn, HTTPServer):
|
||||
pass
|
||||
|
||||
|
||||
def event_trigger():
|
||||
global event_counter
|
||||
for event in client.events(decode=True, filters={'event': 'die'}):
|
||||
@ -377,7 +382,7 @@ def listener():
|
||||
try:
|
||||
# Create a web server and define the handler to manage the
|
||||
# incoming request
|
||||
server = HTTPServer(('', PORT), MyHandler)
|
||||
server = ThreadingSimpleServer(('', PORT), MyHandler)
|
||||
print('Started http server on port ', PORT)
|
||||
|
||||
# Wait forever for incoming http requests
|
||||
|
Loading…
Reference in New Issue
Block a user