要设置非阻塞模式,可以使用fcntl
函数来设置套接字的属性。
以下是一个使用recvfrom
设置非阻塞的示例:
import socketimport fcntlimport os# 创建套接字sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)# 绑定地址和端口sock.bind(('127.0.0.1', 8888))# 设置非阻塞模式fcntl.fcntl(sock, fcntl.F_SETFL, os.O_NONBLOCK)# 接收数据try:data, addr = sock.recvfrom(1024)except socket.error as e:if isinstance(e.args, tuple):if e == socket.error(errno.EWOULDBLOCK):# 表示没有数据可读passelse:print(e)else:# 处理接收到的数据print('Received:', data.decode())# 关闭套接字sock.close()
在上述示例中,使用fcntl.fcntl
函数将套接字的属性设置为非阻塞模式。这样,在调用recvfrom
方法时,如果没有数据可读取,会立即返回一个错误。通过捕获socket.error
异常并检查错误代码,可以判断是否是非阻塞模式下的超时情况。