使用选择处理多个请求

Question

目前我正在研究聊天服务器/客户端项目。我正在努力使用 select 处理多个请求，我的服务器脚本使用 select 模块，但客户端脚本没有。结果是，当用户输入消息时，其他客户端必须编写自己的消息才能通读对话。我已经在网上搜索了很多示例，但只能找到带有 sys.stdin 的代码片段，这不是我想要的。

我很高兴收到任何指示/解释。

服务器代码：

import socket
import select

server_socket = socket.socket()
server_socket.bind(("0.0.0.0", 2855))
server_socket.listen(1)

open_client_sockets = [] # current clients handler
messages_to_send = [] # future message send handler

def send_waiting_messages(wlist):
    for message in messages_to_send:
        (client_socket, data) = message
        if client_socket in wlist: # if current socket in iteration has reading abilities
            client_socket.send(data)
            messages_to_send.remove(message) # remove from future send handler

def broadcast_message(sock, message):
    for socket in open_client_sockets:
        if socket != server_socket and socket != sock:
            socket.send(message)

while True:
    rlist, wlist, xlist = select.select([server_socket] + open_client_sockets, open_client_sockets, []) # apending reading n writing socket to list

    for current_socket in rlist: # sockets that can be read
        if current_socket is server_socket: # if there is a new client
            (new_socket, address) = server_socket.accept() 
            open_client_sockets.append(new_socket) # clients list
        else:
            data = current_socket.recv(1024)
            if len(data) == 0:
                open_client_sockets.remove(current_socket) # remove user if he quit.
                print "Connection with client closed."
                send_waiting_messages(wlist) # send message to specfic client
            else:
                broadcast_message(current_socket, "\r" + '<' + data + '> ')

    # send_waiting_messages(wlist) # send message to specfic client

server_socket.close()

客户端代码：

import socket
import msvcrt

client_socket = socket.socket()
client_socket.connect(("127.0.0.1", 2855))

data = ""
def read_message():
    msg = ""
    while True:
        if msvcrt.kbhit():
            key = msvcrt.getch()
            if key == '\r': # Enter key
                break
            else:
                msg = msg + "" + key

    return msg

while data != "quit":
    data = read_message()
    client_socket.send(data)
    data = client_socket.recv(1024)
    print data
client_socket.close()

Answer 1

服务器端和客户端都存在阻止您的应用程序真正实时的问题。到目前为止，我注意到以下几个：

>

客户端仅在将一些数据写入套接字后才从服务器连接读取数据。请考虑将要从套接字读取的逻辑放入单独的线程中。

在服务器中，在向客户端发送挂起的消息之前，遍历select（）返回的rlist；只有当客户端发送了消息时，客户端fds才会出现在rlist中。发送消息应该基于可写fds，通过在wlist上迭代来完成。但这还有其他问题。。。

您始终select（）所有客户端fd的可写性，即使没有要写入该客户端的挂起消息。最终结果是，您的select（）调用几乎总是立即返回，并且会浪费CPU（这有点违背了select的目的）

您的所有套接字IO都是在“阻塞”模式下完成的，因此如果您发送的数据超过了内核在远程端的接收缓冲区所能处理的数据（通常大约为10MB），则send（）可能会阻塞。

您最好使用异步框架（例如扭曲）来实现这种类型的应用程序。管理所有缓冲区可能既乏味又具有挑战性，而且这是以前已经完成的工作。

Answer 2

我的服务器脚本使用select模块，但客户端脚本不使用。

解决方案是在客户端也使用< code>select。在Windows上，不幸的是< code>select不处理< code>sys.stdin，但是我们可以使用超时参数来轮询键盘。

import socket
import select
import msvcrt
client_socket = socket.socket()
client_socket.connect(("localhost", 2855))
msg = ""
while True:
    ready = select.select([client_socket], [], [], .1)
    if client_socket in ready[0]:
        data = client_socket.recv(1024)
        print data, ' '*(len(msg)-len(data))
        print msg,
    if msvcrt.kbhit():
        key = msvcrt.getche()
        if key == '\r': # Enter key
            if msg == "quit":
                break
            client_socket.send(msg)
            msg = ""
            print
        else:
            msg = msg + "" + key
client_socket.close()