python socket 网络编程s

发布时间:2019-09-08 09:08:39编辑:auto阅读(1461)

    Server端:

    import socket
    import selectors

    class Server(object):
    def init(self,sel,sock):
    self.sel = sel
    self.sock = sock

    def run(self,host,port):
        self.sock.bind((host,port))
        self.sock.listen(50)
        self.sock.setblocking(False)#设置非阻塞
        self.sel.register(sock,selectors.EVENT_READ,self.accept)
        while True:
            events = self.sel.select()#默认是阻塞,有活动连接就返回活动连接列表
            for key,mask in events:
                callback = key.data#创建一个回调函数并获取
                callback(key.fileobj,mask)#调用回调函数
    
    def accept(self,sock,mask):
        conn,addr = sock.accept()#已经就绪,等待接收
        print ('连接来自于{}'.format(addr))
        conn.setblocking(False)
        # sock.send(str('thanks').decode())
        sel.register(conn,selectors.EVENT_READ,self.read)#注册事件
    
    def read(self,conn,mask):
        data = conn.recv(1024)#就绪,等待接收数据
        if data:#判断是否有数据过来,有就执行
            print ('来自客户端:',data)
            conn.send(data)
        else:
            print ('准备关闭连接',conn)
            sel.unregister(conn)
            conn.close()

    if name == 'main':
    sel = selectors.DefaultSelector()#默认的选择方式
    sock = socket.socket()
    sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR,1)
    host,port = '127.0.0.1',10011
    server_obj = Server(sel,sock)
    server_obj.run(host,port)

    Client端:

    import socket
    import selectors
    import random

    sel = selectors.DefaultSelector()

    def write(sock):
    sock.send(str(random.randint(0,99)).encode('utf-8'))#发送必须是一个byts的数据,需要转码
    sel.unregister(sock)#当发送成功后必须取消注册,用与接下来的接收(读)到数据的注册
    sel.register(sock,selectors.EVENT_READ,read)#注册一个读的事件

    def read(sock):
    data = sock.recv(1024)
    if not data:#判断接收到的数据是否为空数据
    sel.unregister(sock)#当接收完成后,依然是取消注册
    sock.close()#到此和服务器的请求基本处理完成,关闭套接字
    return
    print ('receiver server %s' %data)

    for i in range(1000):#开1000个客户端测试
    client = socket.socket()
    client.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)#防止编辑器的每次运行端口冲突问题
    client.connect(('localhost',10011))
    sel.register(client,selectors.EVENT_WRITE,write)#注册一个写的事件,接下来处理发送(写)事件的处理

    while True:
    all_event = sel.select()
    print (allevent)
    for event,
    in all_event:
    sock = event.fileobj
    callback = event.data#创建一个回调函数,方便触发事务执行
    callback(sock)#调用刚创建的回调函数

关键字