Python學習之實現高並發socket server

客戶端
#!/usr/bin/env python
#-*-coding:utf8-*-

import socket

HOST = 'localhost'  # The remote host
PORT = 8001  # The same port as used by the server
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect((HOST, PORT))
while True:
    msg = bytes(input(">>:"), encoding="utf8")
    s.sendall(msg)
    data = s.recv(1024)
    # print(data)

    print('Received', repr(data))  #repr格式化輸出
s.close()

服務器
#!/usr/bin/env python
#-*-coding:utf8-*-

import sys
import socket
import time
import gevent

from gevent import socket, monkey

monkey.patch_all()


def server(port):
    s = socket.socket()
    s.bind(('0.0.0.0', port))
    s.listen(500)
    while True:
        cli, addr = s.accept()  #socket sever到這步默認會生成一個線程，把請求交給線程處理
        # 但這里是交給協程處理（cli就是客戶端連過來而在服務器端為其生成的一個連接實例）

        gevent.spawn(handle_request, cli) #啟動一個協程
        #把客戶端請求連接生成的實例cli交給handle_request方法


def handle_request(conn):  #負責處理和客戶端請求的所有交互
    try:
        while True:
            data = conn.recv(1024)
            print("recv:", data)
            conn.send(data)
            if not data:
                conn.shutdown(socket.SHUT_WR)

    except Exception as  ex:
        print(ex)
    finally:
        conn.close()


if __name__ == '__main__':
    server(8001)

 

（會生成一個線程）