Python之基于socket和select模块实现IO多路复用

'''IO指的是输入输出，一部分指的是文件操作，还有一部分
网络传输操作，例如soekct就是其中之一；多路复用指的是
利用一种机制，同时使用多个IO，例如同时监听多个文件句
柄（socket对象一旦传送或者接收信息），一旦文件句柄出
现变化就会立刻感知到
'''
1、下面通过IO多路复用实现多人同时连接socket服务器

这是服务端代码

import socket

sk1 = socket.socket()#sk1,sk2,sk3这就是一个文件描述符
sk1.bind(('127.0.0.1',8002))
sk1.listen()

sk2 = socket.socket()
sk2.bind(('127.0.0.1',8003))
sk2.listen()

sk3 = socket.socket()
sk3.bind(('127.0.0.1',8004))
sk3.listen()

inputs = [sk1,sk2,sk3]
import select
while True:
    '''[sk1,sk2,sk3],select内部会自动监听sk1,sk21，sk3三个对象
    一旦某个句柄发生变化就会被监听到
    '''
    #如果有人链接sk1，则会被添加进列表，r_list = [sk1]
    r_list,w_list,e_list = select.select(inputs,[],[],1)#1表示等一秒，在while执行到这里的时候监测一秒，没有人来链接的话就接着循环
    #print(r_list,sk1.listen())
    for sk in r_list:
        conn,address = sk.accept()
        #print(conn,address)
        conn.sendall(bytes('你好',encoding='utf-8'))
        conn.close()

　　客户端的代码都是一样的，就差个端口

#客户端一
import socket

obj = socket.socket()
obj.connect(('127.0.0.1',8002))
content = str(obj.recv(1024),encoding='utf-8')
print(content)

obj.close()

#客户端二
import socket

obj = socket.socket()
obj.connect(('127.0.0.1',8003))
content = str(obj.recv(1024),encoding='utf-8')
print(content)

obj.close()

#客户端三
import socket

obj = socket.socket()
obj.connect(('127.0.0.1',8004))
content = str(obj.recv(1024),encoding='utf-8')
print(content)

obj.close()

　　执行结果：只要启动服务器端，然后不同的客户端多次启动都能收到信息，多个端口成功被监听

2、下面使用select模块实现多路复用，使同一个端口同时接收多个链接

import socket

sk1 = socket.socket()#sk1,sk2,sk3这就是一个文件描述符
sk1.bind(('127.0.0.1',8002))
sk1.listen()

#
# sk2 = socket.socket()
# sk2.bind(('127.0.0.1',8003))
# sk2.listen()
#
# sk3 = socket.socket()
# sk3.bind(('127.0.0.1',8004))
# sk3.listen()

inputs = [sk1]
import select

#epoll效率更高，但是Windows不支持，它是谁有问题就告诉它，不用循坏
while True:
    '''[sk1,sk2,sk3],select内部会自动监听sk1,sk21，sk3三个对象
    一旦某个句柄发生变化（某人来链接）就会被监听到
    '''
    #如果有人链接sk1，则会被添加进列表，r_list = [sk1]
    r_list,w_list,e_list = select.select(inputs,[],[],1)
    '''第三个参数是监听错误的，只要有错误出现，就会被监听到，返回e_list
       第二个参数返回给w_list，只要传了什么，就原封不动的传给w_list'''
    print('正在监听 %s 多少个对象' % len(inputs))
    for sk in r_list:
        if sk == sk1:
            #句柄跟服务器端的对象一样，表示有新用户来链接了
            conn,address = sk.accept()
            inputs.append(conn)#加入去之后，inputs有一个链接对象和服务器对象
        else:
            #有老用户发消息了
            try:
                data_byte =sk.recv(1024)

                data_str =str(data_byte,encoding='utf-8')
                sk.sendall(bytes(data_str+'收到了',encoding='utf-8'))
            except:#空信息表示客户端断开链接，所以要在监听中移除

                inputs.remove(sk)#这里的sk就是之前传进去的conn，因为r_list接收的是有变化的值

　启动这个服务端之后，就可以实现多路复用了，可以接收多个客户端同时连接

3、下面介绍一些多路操作里面的读写分离

import socket

sk1 = socket.socket()#sk1就是一个文件描述符
sk1.bind(('127.0.0.1',8002))
sk1.listen()
inputs = [sk1]#被检测发生变动的句柄放这里
outputs=[]#用来记录谁给服务端发过信息，以便回复
message_dict = {}#接收信息的，根据句柄形成键值对
import select

#epoll效率更高，但是Windows不支持，它是谁有问题就告诉它，不用循坏
while True:
    '''[sk1,sk2,sk3],select内部会自动监听sk1,sk21，sk3三个对象
    一旦某个句柄发生变化（某人来链接）就会被监听到
    '''
    #如果有人链接sk1，则会被添加进列表，r_list = [sk1]
    r_list,w_list,e_list = select.select(inputs,outputs,inputs,1)
    '''第三个参数是监听错误的，只要有错误出现，就会被监听到，返回e_list
       第二个参数返回给w_list，只要传了什么，就原封不动的传给w_list'''
    print('正在监听 %s 多少个对象' % len(inputs))
    for sk in r_list:
        if sk == sk1:
            #句柄跟服务器端的对象一样，表示有新用户来链接了
            conn,address = sk.accept()
            inputs.append(conn)#加入去之后，inputs有一个链接对象和服务器对象
            message_dict[conn]=[]#字典的key是具体的链接对象
        else:
            #有老用户发消息了
            try:
                data_byte =sk.recv(1024)

            except Exception as e:#空信息表示客户端断开链接，所以要在监听中移除
                print(e)
                inputs.remove(sk)#这里的sk就是之前传进去的conn，因为r_list接收的是有变化的值
            else:
                data_str = str(data_byte, encoding='utf-8')

                message_dict[sk].append(data_str)
                outputs.append(sk)#把传送数据的句柄添加进去第二个参数
    for conn in w_list:
        recv_str=message_dict[conn][0]#拿到我们收到了的信息
        del message_dict[conn][0]#删除信息，防止下次出现一样的消息
        print(recv_str)
        conn.sendall(bytes(recv_str+'收到了',encoding='utf-8'))
        outputs.remove(conn)#回复完成之后在列表删除
    for sk_or_conn in e_list:
        e_list.remove(sk_or_conn)

　　这样可以形成简单的读写分离操作

对于select里面的参数关系，这里有个武sir画的图