python的select服务端的代码和客户端的代码

时间：2023-03-09 16:25:10
python的select服务端的代码和客户端的代码

服务端的代码

import socket

import queue

import select

ip_bind = ("127.0.0.1",9000)

message_queue = {}

#保存客户端发送过来的信息，将消息放入到队列中

input_list = []

output_list = []

if __name__ == '__main__':

    server = socket.socket()

    server.bind(ip_bind)

    server.listen(10)

    server.setblocking(False)

    #设置socket服务端为非阻塞的

    input_list.append(server)

    #初始化将服务端加入到监听列表中

    while True:

        print("waiting for new connection")

        stdinput,stdoutput,stderr = select.select(input_list,output_list,input_list,2)

        #开始select监听，对input_list中的服务端server进行监听

        for obj in stdinput:

            #判断是否有客户端连接进来，当有客户端连接进来时select将触发，就会进入循环

            if obj == server:  #代表一个新的连接进来

                #判断当前触发的是不是服务端的对象，当触发的对象是服务端的对象时候，说明

                #有新的客户端连接进来了

                conn,addr = server.accept()

                print("Client {0} is connected".format(addr))

                conn.setblocking(False)

                #把这个新连接的实例设置为非阻塞

                #这个时候我们不能立即接受数据，因为在select中，是单线程的，如果这里直接开始接受数据，那么其他的客户端就无法连接进来了，所以只能

                #先把这个链接存起来，放到一个input_list中，为什么放到input_list中呢，因为我们想监控这个链接，如果这个客户端有消息过来，那么select

                #就会触发，如果没有数据返回，则select就不会被触发

                input_list.append(conn)

                #将和这个客户端的连接的服务端的实例也加入到监听列表中，当客户端发送消息的

                #时候，select就会被触发

                message_queue[conn] = queue.Queue()

                #为这个连接的客户端单独创建一个独一无二的消息队列，用来保存客户端发送的消息。

            else:  #代表一个有数据的链接进来了，这个时候我就可以开始收数据了

                #由于客户端连接进来时服务端接受客户端的连接请求，将这个客户端的服务端的

                #也加入到监听列表中，这个客户端如果发送消息，则会触发select

                try:

                    recv_data = obj.recv(1024)

                    if recv_data:

                        #客户端未断开

                        print("received {0} from client {1}".format(str(recv_data,encoding="utf-8"),addr))

                        #将收到的信息放入该客户端的队列中

                        message_queue[obj].put(recv_data)

                        #此时也不能直接给客户端返回数据，因为一旦返回数据，就又可能造成阻塞，所有将回复操作放到output列表中，让select监听

                        if obj not in output_list:

                            output_list.append(obj)

                except ConnectionResetError:

                    input_list.remove(obj)

                    del message_queue[obj]

                    print("client {0} is disconnected".format(addr))

        for sendobj in stdoutput: #这里处理的是返回的消息，output_list是我们自己维护的，我们自己往里面放数据

            try:

                if not message_queue[sendobj].empty():

                #如果消息队列中有消息，从消息队列中获取要发送的消息

                    send_data = message_queue[sendobj].get()

                    #从该客户端对象的消息队列中获取消息

                    sendobj.sendall(send_data)

                else:

                    output_list.remove(sendobj)

                    #将监听移除等待瞎猜疑客户端发送消息

            except ConnectionResetError:

                #客户端连接断开

                del  message_queue[sendobj]

                output_list.remove(sendobj)

                print("Client {0} disconnected".format(addr))

        for obj in stderr:  #如果出错的处理

            # print()

            input_list.remove(obj)

            if obj in output_list:

                output_list.remove(obj)

            obj.close()

            del message_queue[obj]

　　

相关文章