Python-4.32 gevent实现并发的套接字通信

• 基于gevent实现

服务端

from gevent import monkey,spawn;monkey.patch_all()
from socket import *



def communicate(conn):
    while True:
        try:
            data = conn.recv(1024)
            if not data: break
            conn.send(data.upper())
        except ConnectionResetError:
            break
    conn.close()


def server(ip, port):
    server = socket(AF_INET, SOCK_STREAM)
    server.bind((ip, port))
    server.listen(5)

    while True:
        conn, addr = server.accept()
        spawn(communicate, conn)

    server.close()


if __name__ == '__main__':
    g = spawn(server, '127.0.0.1', 8078)
    g.join()

记住：from gevent import monkey;monkey.patch_all()必须放到被打补丁者的前面，如time，socket模块之前

客户端

from socket import *
from threading import Thread, currentThread


def client():
    client = socket(AF_INET, SOCK_STREAM)
    client.connect(('127.0.0.1', 8078))

    while True:
        client.send(('%s hello' % currentThread().getName()).encode('utf-8'))
        data = client.recv(1024)
        print(data.decode('utf-8'))

    client.close()


if __name__ == '__main__':
    for i in range(500):
        t = Thread(target=client)
        t.start()