qq489498494 发表于 2018-8-7 06:42:33

Python selectors-90SirDB

#!/usr/bin/env python  
# -*- coding:utf-8 -*-
  
# author: Changhua Gong
  

  
import selectors
  
import socket
  

  
sel = selectors.DefaultSelector()
  
'''selectors根据系统自动select或epoll'''
  

  
def accept(sock, mask):
  
    conn, addr = sock.accept()# Should be ready
  
    print('accepted', conn, 'from', addr)
  
    conn.setblocking(False)# 设置连接不阻塞
  
    sel.register(conn, selectors.EVENT_READ, read) # 新连接注册read回调函数
  

  

  
def read(conn, mask):
  
    data = conn.recv(1024)# 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', 10000))
  
sock.listen(100)
  
sock.setblocking(False)
  
sel.register(sock, selectors.EVENT_READ, accept)# 注册sock,有连接进来则调用accept
  

  
while True:
  
    events = sel.select() # 默认阻塞,有活动连接就返回活动的连接列表
  
    for key, mask in events:
  
      callback = key.data # accept的内存地址或read的内存地址,即等于read或accept函数
  
      callback(key.fileobj, mask)# 调用对应的回调函数,key.fileobj为文件句柄
页: [1]
查看完整版本: Python selectors-90SirDB