log zmq socket实例send/recv？

Question

我正在调试一些zeromq代码，我想记录套接字实例send/recv。换句话说，我有

context = zmq.Context() 
socket = context.socket(zmq.REQ) 
socket.connect("tcp://127.0.0.1:5000") 

我想以某种方式在代码中的其他地方记录套接字实例的send/recv。

这里的任何想法？我不确定从哪里开始。

Answer 1

下面是记录每个消息Socket对象的示例：

 

import logging 
import time 
import zmq 

class LogSocket(object): 
    socket = None 
    log = None 
    name = None 

    def __init__(self, socket, log=None, name=None): 
     """wrap a socket so send/recv_multipart log their messages""" 
     self.socket = socket 
     if log is None: 
      logging.basicConfig(level=logging.INFO) 
      log = logging.getLogger().info 
     self.log = log 

     if name is None: 
      name = s.identity or repr(s) 
     self.name = name 

    def send_multipart(self, msg, *args, **kwargs): 
     self.log("%s sending: %r" % (self.name, msg)) 
     return self.socket.send_multipart(msg, *args, **kwargs) 

    def recv_multipart(self, *args, **kwargs): 
     msg = self.socket.recv_multipart(*args, **kwargs) 
     self.log("%s received: %r" % (self.name, msg)) 
     return msg 

    def __getattr__(self, key): 
     return getattr(self.socket, key) 


ctx = zmq.Context.instance() 

s = ctx.socket(zmq.REQ) 
s.connect("tcp://127.0.0.1:5556") 

# wrap the zmq socket in the Logging object: 
# this object has exactly the same interface as 
s = LogSocket(s, name="REQ") 

for i in range(5): 
    time.sleep(1) 
    s.send_multipart(['hello', 'world']) 
    reply = s.recv_multipart() 
    # in this case, only _multipart methods are logged, so straight send/recv are not: 
    s.send("quiet") 
    s.recv() 

用于处理序列/记录/认证等在同一个地方的第二个目标的一个例子是在IPython中，所述Session object这比这个简单的例子复杂得多，但它允许你集中代码中的所有消息抽象。