Python中的Tcp协议应用之TCP服务端-协程版（推荐）

利用gevent第三方库，实现协程。

通过协程实现一个服务端服务多个客户端需求。

使用协程的好处是协程比线程更加节省内存资源。

gevent安装命令：

pip3 install gevent

注意：在安装gevent库之前，需要更新pip版本，只需要在终端窗口执行以下命令即可：

python -m pip install --upgrade pip 

TCP服务端-协程版本：

import socket
import gevent
from gevent import monkey

# 打补丁
monkey.patch_all()


def handle_client_socket(new_socket, ip_port):
    print("建立连接成功：", ip_port)
    while True:
        recv_data = new_socket.recv(1024)
        if recv_data:
            recv_data_decode = recv_data.decode("UTF-8")
            print("服务端接收的数据为：", recv_data_decode)
        else:
            print("客户端下线了", ip_port)
            break
        # 8.使用新的套接字发送数据
        send_data_input = input("请输入回复的消息：")
        send_data = send_data_input
        send_data_encode = send_data.encode("UTF-8")
        new_socket.send(send_data_encode)
        # 7.使用新的套接字接收数据，并设置最大数据接收长度为1024

    # 9.关闭套接字
    # 与客户端断开通信
    new_socket.close()


# 1.导入模块
if __name__ == '__main__':
    # 2.创建套接字对象
    tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    # 3.设置端口复用
    tcp_server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, True)

    # 4.绑定端口
    tcp_server_socket.bind(("", 9090))

    # 5.设置监听,并设置最大等待连接数为128
    tcp_server_socket.listen(128)
    while True:
        # 6.等待客户端建立连接,并接收accpet()返回的新的套接字对象，和客户端IP+端口号
        new_socket, ip_port = tcp_server_socket.accept()
        gevent.spawn(handle_client_socket, new_socket, ip_port)

    # 停止服务端服务
    # tcp_server_socket.close()