一、并发TCP服务器

  我们使用线程的方式实现并发 TCP 服务器。

from socket import socket
from socket import AF_INET, SOCK_STREAM, SOL_SOCKET, SO_REUSEADDR
from time import ctime
from threading import Thread

HOST = "127.0.0.1"
PORT = 8080
ADDRESS = (HOST, PORT)

tcp_server = socket(AF_INET, SOCK_STREAM)               # 创建服务器套接字
tcp_server.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)      # 解决端口占用问题
tcp_server.bind(ADDRESS)                                # 套接字与地址绑定
tcp_server.listen(5)                                    # 监听连接，最大挂起连接数为5

class TCPServer(Thread):
    def run(self):
        print("waiting for connection...")

        while True:                                     # 监听连接
            connection, address = tcp_server.accept()   # 接收客户端连接
            print("...connection from: ", address)

            # 创建一个新的子线程，专门为刚刚创建的客户端服务
            handle_data_thread = HandleData(connection, address)
            handle_data_thread.start()

    def __del__(self):
        tcp_server.close()                              # 关闭服务器套接字

class HandleData(Thread):
    def __init__(self, connection, address):
        super().__init__()
        self.connection = connection
        self.address = address

    def run(self):
        while True:                                     # 通信循环
            try:
                recv_data = self.connection.recv(1024)  # 服务端接收消息，单次最大接收为1024个字节
                # 在Linux系统中，一旦data收到空，意味着是一种异常的行为：客户端非法断开连接
                if not recv_data:                       # 适用于Linux系统
                    break
                print(f"收到客户端【{self.address}】发送的数据：{recv_data.decode('utf-8')}")
                self.connection.send(f"【{ctime()}】 {recv_data.decode('utf-8')}".encode("utf-8"))  # 服务端发送消息
            except ConnectionResetError:                # 适用于Windows系统
                break

        print(f"客户端【{self.address}】断开连接！")
        self.connection.close()                         # 关闭连接

if __name__ == "__main__":
    server = TCPServer()
    server.start()

二、TCP服务器

from socket import socket
from socket import AF_INET, SOCK_STREAM

HOST = "127.0.0.1"
PORT = 8080
ADDRESS = (HOST, PORT)

tcp_client = socket(AF_INET, SOCK_STREAM)               # 创建客户端套接字
tcp_client.connect(ADDRESS)                             # 尝试连接服务器

while True:                                             # 通信循环
    send_data = input("请输入要发送的数据: ").strip()
    if not send_data:                                   # 按空格、回车等键结束连接
        break
    tcp_client.send(send_data.encode("utf-8"))          # 客户端发送数据，不允许发送空数据

    recv_data = tcp_client.recv(1024)                   # 客户端接收数据，单次最大接收为1024个字节
    if not recv_data:
        break
    print(f"收到服务端【{ADDRESS}】返回的数据：{recv_data.decode('utf-8')}")

tcp_client.close()                                      # 关闭客户端套接字