sockets - 改善PyMongo套接字接收

Question

我已经使用cProfile在我的代码中进行了一些分析(使用PyMongo 3.0.3进行了大量批量操作)，结果发现，套接字花费了大量时间:
14006 68.823 0.005 68.823 0.005 {method 'recv' of '_socket.socket' objects}
我发现了这个somewhat related answer，解决方法很简单。有什么方法可以将TCP_NOWAIT选项传递给PyMongo，而不必更改libs代码吗？

Answer 1

也许你可以猴子补丁 socket ...

#before you import or do anything...
import socket
socket.raw_socket = socket.socket

def my_socket(*args, **kws):
    sock = socket.socket(*args, **kws)
    sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1)
    return sock

socket.socket = my_socket
#cross fingers ...