use class to reconstruct server

Author: HFrost0

echo/__init__.py

@@ -0,0 +1,12 @@
+import socket
+
+
+class BaseServer:
+    def __init__(self, address):
+        self.listener = socket.socket(family=socket.AF_INET, type=socket.SOCK_STREAM)
+        self.listener.bind(address)
+        self.listener.listen()
+
+    def start_sering(self):
+        raise NotImplemented
+

echo/base_server.py

@@ -4,24 +4,25 @@
 import socket
 import threading
 
+from echo.base_server import BaseServer
 
-def accept(s: socket.socket):
-    while data := s.recv(1024):
-        s.send(data)
-        # sum(range(100000))  # cpu bound task
-        print(f'Echo: {data}')
-    s.close()
-    print(f'Remove: {s}')
+
+class ThreadServer(BaseServer):
+    def start_sering(self):
+        while True:
+            s, addr = self.listener.accept()
+            print(f'Connection: {addr}')
+            threading.Thread(target=self.handler, args=(s,)).start()
+
+    def handler(self, s: socket.socket):
+        while data := s.recv(1024):
+            s.send(data)
+            # sum(range(100000))  # cpu bound task
+            print(f'Echo: {data}')
+        s.close()
+        print(f'Remove: {s}')
 
 
 if __name__ == '__main__':
-    server_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-    server_sock.bind(('', 6666))
-    server_sock.listen()
-    threads = []
-    while True:
-        s, addr = server_sock.accept()
-        print(f'Connection: {addr}')
-        t = threading.Thread(target=accept, args=(s,))
-        threads.append(t)
-        t.start()
+    server = ThreadServer(("127.0.0.1", 6666))
+    server.start_sering()

echo/echo_server_thread.py

@@ -1,22 +1,29 @@
 import socket
 from concurrent.futures import ThreadPoolExecutor
 
+from echo.base_server import BaseServer
 
-def accept(s: socket.socket):
-    while data := s.recv(1024):
-        s.send(data)
-        # sum(range(100000))  # slow down respond
-        print(f'Echo: {data}')
-    s.close()
-    print(f'Remove: {s}')
+
+class ThreadPoolServer(BaseServer):
+    def __init__(self, address=("127.0.0.1", 6666)):
+        super(ThreadPoolServer, self).__init__(address)
+        self.pool = ThreadPoolExecutor()
+
+    def start_sering(self):
+        while True:
+            s, addr = self.listener.accept()
+            print(f'Connection: {addr}')
+            self.pool.submit(self.handler, s)
+
+    def handler(self, sock: socket.socket):
+        while data := sock.recv(1024):
+            sock.send(data)
+            # sum(range(100000))  # slow down respond
+            print(f'Echo: {data}')
+        sock.close()
+        print(f'Remove: {sock}')
 
 
 if __name__ == '__main__':
-    pool = ThreadPoolExecutor()
-    server_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-    server_sock.bind(('', 6666))
-    server_sock.listen()
-    while True:
-        s, addr = server_sock.accept()
-        print(f'Connection: {addr}')
-        pool.submit(accept, s)
+    server = ThreadPoolServer()
+    server.start_sering()