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

时间:2021-05-31 00:00:01
'''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画的图

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