Python 处理TCP连接 - Cloud Notes

import socket
# 连接到本机的12345端口
tcp_addr = ("127.0.0.1", 12345)
# 建立与服务端的TCP连接
tcp_client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
tcp_client.connect(tcp_addr)
# 注意发送的数据必须是字节流，一般使用UTF-8格式
tcp_client.sendall("hello world".encode("utf-8"))
# 关闭与服务端的TCP连接
tcp_client.close()
同时处理多个客户端请求¶
如果需要TCP服务端同时处理多个客户端的请求，可以采取多线程的方式，对于一段时间不活动的客户端连接，可以主动关闭连接节省系统资源，代码实现如下
Python
 1
61 import signal
import socket
import threading
import time
import typing
def handle_sigterm(*args):
    """接收中断信号，安全关闭tcp连接"""
    raise KeyboardInterrupt()
class TcpServer(object):
    def __init__(self, host: str = "0.0.0.0", port: int = 12345, timeout: int = 60):
        self.host = host
        self.port = port
        self.timeout = timeout
        self._addr = (host, port)
        self.server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.connections = dict()
    def start(self):
        try:
            signal.signal(signal.SIGTERM, handle_sigterm)
            self._start()
        except KeyboardInterrupt:
            self.server.close()
            print(f"TCP server closed, bye.")
    def _start(self):
        self.server.bind(self._addr)
        self.server.listen()
        print(f"TCP server started, listening on {self._addr}")
        while True:
            client_conn, client_addr = self.server.accept()
            print(f"New TCP client connected, address: {client_addr}")
            client_thread = threading.Thread(
                target=self._handle_new_client, args=(client_conn, client_addr)
            client_thread.start()
            self.connections[client_addr] = client_thread
    def _handle_new_client(
            self, client_socket: socket.socket, client_addr: typing.Tuple[str, int]
        last_communication = time.time()
        while time.time() - last_communication < self.timeout:
            if data := client_socket.recv(10240).decode("utf-8"):
                print(f"New data from client {client_addr}: {data}")
                last_communication = time.time()
            else:
                time.sleep(1)
        client_socket.close()
        print(f"TCP client {client_addr} closed due to timeout")
        self.connections.pop(client_addr)
if __name__ == "__main__":
    tcp_server = TcpServer()
    tcp_server.start()
Socket Programming in Python (Guide) – Real Python
How to Create Socket Server with Multiple Clients in Python
sockets - What is AF_INET, and why do I need it? - Stack Overflow