python 杂记 网络

参考资料：https://www.cnblogs.com/gareth-yu/p/9097943.html
import selectors
import socket

sel = selectors.DefaultSelector()

def accept(sock, mask):              # sock相当于key.fileobj
    conn, addr = sock.accept()  # Should be ready
    print(‘accepted‘, conn, ‘from‘, addr)
    conn.setblocking(False)
    sel.register(conn, selectors.EVENT_READ, read)

def read(conn, mask):
    data = conn.recv(1000)  # Should be ready
    if data:
        print(‘echoing‘, repr(data), ‘to‘, conn)
        conn.send(data)  # Hope it won‘t block
    else:
        print(‘closing‘, conn)
        sel.unregister(conn)
        conn.close()

sock = socket.socket()
sock.bind((‘localhost‘, 1234))
sock.listen(100)
sock.setblocking(False)
sel.register(sock, selectors.EVENT_READ, accept)

while True:
    events = sel.select()     # 根据系统是否支持，使用epoll还是select，优先epoll。默认阻塞，有活动连接就返回活动的连接列表
    for key, mask in events:
        callback = key.data    # 相当于回调accept
        callback(key.fileobj, mask)  # key.fileobj相当于建立连接后的conn