阅读量:8
在使用socket进行网络通信时,可以设置socket为阻塞或非阻塞模式。
阻塞模式下,当执行socket的读取和写入操作时,如果没有数据可读或无法立即写入数据,当前线程会被阻塞,直到有数据可读或可以写入数据。
非阻塞模式下,当执行socket的读取和写入操作时,如果没有数据可读或无法立即写入数据,当前线程不会被阻塞,而是立即返回一个错误码,然后可以通过不断地轮询socket状态来判断是否有数据可读或可以写入数据。
要设置socket为非阻塞模式,可以使用socket.setblocking(False)
方法来实现。该方法接受一个布尔值参数,设置为False表示将socket设置为非阻塞模式。要设置回阻塞模式,可以使用socket.setblocking(True)
。
下面是一个简单的示例代码,演示如何设置socket为非阻塞模式:
import socket # 创建一个TCP socket sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # 将socket设置为非阻塞模式 sock.setblocking(False) # 连接到服务器 sock.connect(('www.example.com', 80)) # 发送数据 sock.send('GET / HTTP/1.1\r\nHost: www.example.com\r\n\r\n') # 接收数据 data = '' while True: try: chunk = sock.recv(1024) if not chunk: break data += chunk except socket.error: pass # 关闭socket sock.close() # 打印接收到的数据 print(data)
在这个示例中,通过sock.setblocking(False)
将socket设置为非阻塞模式,然后在接收数据时通过捕获socket.error异常来处理非阻塞模式下的错误码。