Python网络编程Socket之协程

一、服务端

__author__ = "Jent Zhang"

import socket
import gevent
from gevent import monkey

monkey.patch_all()          # 把当前程序中的所有IO操作单独做上标记


def server(port):
    s = socket.socket()
    s.bind(("0.0.0.0", port))
    s.listen(5000)
    while True:
        conn, addr = s.accept()
        gevent.spawn(handle_request, conn)


def handle_request(conn):
    try:
        while True:
            data = conn.recv(1024)
            print(data.decode("utf8"))
            conn.send(data.upper())
            if not data:
                conn.shutdowm(socket.SHUT_WR)

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


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

二、客户端

__author__ = "Jent Zhang"

import socket

HOST = "localhost"
PORT = 8001

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect((HOST, PORT))

while True:
    msg = bytes(input(">>:"), encoding="utf8")
    if not msg: continue
    s.sendall(msg)
    data = s.recv(1024)

    print("Received", repr(data.decode("utf8")))

s.close()