YAO/YAO-agent
1
0
Fork 0
mirror of https://github.com/newnius/YAO-agent.git synced 2025-06-06 21:31:55 +00:00
Code Issues Releases Wiki Activity

upgrade to multithreading http service

Browse Source
This commit is contained in:
Newnius 2020-05-27 20:49:18 +08:00
parent f43af24ecd
commit b5ff2de041
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
agent.py
View File

@ -11,6 +11,7 @@ import multiprocessing
import psutil import psutil
import math import math
from http.server import BaseHTTPRequestHandler, HTTPServer from http.server import BaseHTTPRequestHandler, HTTPServer
from socketserver import ThreadingMixIn
import cgi import cgi
import docker import docker
from urllib import parse from urllib import parse
@ -276,6 +277,10 @@ class MyHandler(BaseHTTPRequestHandler):
self.send_error(404, 'File Not Found: %s' % self.path) self.send_error(404, 'File Not Found: %s' % self.path)
class ThreadingSimpleServer(ThreadingMixIn, HTTPServer):
pass
def event_trigger(): def event_trigger():
global event_counter global event_counter
for event in client.events(decode=True, filters={'event': 'die'}): for event in client.events(decode=True, filters={'event': 'die'}):
@ -377,7 +382,7 @@ def listener():
try: try:
# Create a web server and define the handler to manage the # Create a web server and define the handler to manage the
# incoming request # incoming request
server = HTTPServer(('', PORT), MyHandler) server = ThreadingSimpleServer(('', PORT), MyHandler)
print('Started http server on port ', PORT) print('Started http server on port ', PORT)
# Wait forever for incoming http requests # Wait forever for incoming http requests