关于 socket.recv 阻塞问题

在开发中遇到个问题。

标准的socket 客户端应该是这样的~_~ 我见网上书上都这样写。所以我说是“标准的”

# -*- coding: utf-8 -*-
import socket,time

host = "127.0.0.1"
port = 9009

s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.connect((host,port))

s.send("SELECT * FROM tt")
a = s.recv(655350)
print a
s.close()

客户端在 recv 服务端发送来的数据后，就直接 close socket 了。这样的确是没问题

但我在想，如果服务端一次发送的数据量很大很大~~ 超出了 recv(指定的bufsize) 那么一次接收不是有问题了吗？所以我改成了循环接收方式

# -*- coding: utf-8 -*-
import socket,time

host = "127.0.0.1"
port = 9009

s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.connect((host,port))

s.send("SELECT * FROM tt")
while True:
a = s.recv(1024)
if not len(a):break
print a
s.close()

的确。一样能接收，但这样的问题是。 recv 是阻塞的。也就是会一直等待服务端发送来的数据包。如果没有数据包到来，就一直会等待。

也就是后面的 if not len(a):break 无效，也就退不出循环。这个也是不符合我的需求的。 第一想到如何能让 recv 不阻塞了。通过查看手册发现两个办法

1 。 socket.setblocking(0)

2. 使用 socket.MSG_DONTWAIT 这个我们可以通过 man recv 来查看说明

但这样以后，还是有问题~，问题就是，recv不阻塞了。在 send 数据出去后，服务端还来没来得急返回数据，客户端已经跑到了recv这里，而又不阻塞。所以抛出了一个异常。这个在手册上有说明

Set blocking or non-blocking mode of the socket: if flag is 0, the socket is set to non-blocking, else to blocking mode. Initially all sockets are in blocking mode. In non-blocking mode, if a recv() call doesn’t find any data, or if a send() call can’t immediately dispose of the data, a error exception is raised;

可以在recv前加一个time.sleep(2)模拟一个阻塞就可以看到这个事实。加了 sleep 以后，recv就可以等到服务端返回的数据，顺利的在不阻塞的状态下执行下去了。