init commit

2022-12-21 20:30:44 +08:00
parent 178bc16c68
commit e492836c57
36 changed files with 4026 additions and 0 deletions
@@ -0,0 +1,57 @@
+import json
+import threading
+import socketserver
+
+
+class ReceiveMsgSocketServer(socketserver.BaseRequestHandler):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+    def handle(self):
+        conn = self.request
+        while True:
+            try:
+                ptr_data = b""
+                while True:
+                    data = conn.recv(1024)
+                    ptr_data += data
+                    if len(data) == 0 or data[-1] == 0xA:
+                        break
+
+                msg = json.loads(ptr_data)
+                ReceiveMsgSocketServer.msg_callback(msg)
+
+            except OSError:
+                break
+            except json.JSONDecodeError:
+                pass
+            conn.sendall("200 OK".encode())
+        conn.close()
+
+    @staticmethod
+    def msg_callback(msg):
+        print(msg)
+
+
+def start_socket_server(port: int = 19099,
+                        request_handler=ReceiveMsgSocketServer,
+                        main_thread: bool = True) -> int or None:
+    ip_port = ("127.0.0.1", port)
+    try:
+        s = socketserver.ThreadingTCPServer(ip_port, request_handler)
+        if main_thread:
+            s.serve_forever()
+        else:
+            socket_server = threading.Thread(target=s.serve_forever)
+            socket_server.setDaemon(True)
+            socket_server.start()
+            return socket_server.ident
+    except KeyboardInterrupt:
+        pass
+    except Exception as e:
+        print(e)
+    return None
+
+
+if __name__ == '__main__':
+    start_socket_server()