温馨提示×

SOCKET中SELECT的用法是什么

小亿
214
2023-12-14 12:54:35
栏目: 编程语言

在Socket编程中,select() 是一个用于多路复用的函数,用于监视套接字集合的状态变化。它可以同时监视多个套接字,一旦其中任何一个套接字变得可读、可写或发生错误,select() 就会返回,告诉程序该套接字的状态发生了变化。

select() 函数的基本用法如下:

import select

# 创建套接字
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.bind((host, port))
sock.listen(5)

# 创建套接字集合
sockets_list = [sock]
read_sockets, write_sockets, error_sockets = select.select(sockets_list, [], [])

# 处理可读的套接字
for sock in read_sockets:
    # 如果是监听套接字,则表示有新的连接
    if sock == server_socket:
        client_socket, client_address = server_socket.accept()
        sockets_list.append(client_socket)
        print('New connection from {}'.format(client_address))
    # 否则,表示有数据可读
    else:
        data = sock.recv(1024)
        if data:
            print('Received data: {}'.format(data))
        else:
            # 如果没有接收到数据,则表示连接已关闭
            print('Connection closed')
            sockets_list.remove(sock)

在上面的代码中,select() 函数监视了 sockets_list 中的套接字,一旦其中任何一个套接字变得可读,select() 函数就会返回。然后程序可以通过遍历 read_sockets 进行处理,对于监听套接字,表示有新的连接;对于其他套接字,表示有数据可读。

0