要設置非阻塞模式,可以使用fcntl
函數來設置套接字的屬性。
以下是一個使用recvfrom
設置非阻塞的示例:
import socket
import fcntl
import 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):
# 表示沒有數據可讀
pass
else:
print(e)
else:
# 處理接收到的數據
print('Received:', data.decode())
# 關閉套接字
sock.close()
在上述示例中,使用fcntl.fcntl
函數將套接字的屬性設置為非阻塞模式。這樣,在調用recvfrom
方法時,如果沒有數據可讀取,會立即返回一個錯誤。通過捕獲socket.error
異常并檢查錯誤代碼,可以判斷是否是非阻塞模式下的超時情況。