复习ZMQ的 REQ-ROUTER

最新推荐文章于 2024-07-20 00:05:07 发布

weixin_34138377

最新推荐文章于 2024-07-20 00:05:07 发布

阅读量429

点赞数

文章标签： python 网络

原文链接：https://my.oschina.net/cppblog/blog/536202

版权

2019独角兽企业重金招聘Python工程师标准>>>

服务器端代码：

import gevent
import zmq.green as zmq

def serve():
    context = zmq.Context()
    socket = context.socket(zmq.ROUTER)
    socket.bind("tcp://*:5560")    
    while True:
        client_id, _ ,msg = socket.recv_multipart()
        print ("Received request: ", msg)
        gevent.sleep(0.3)
        socket.send_multipart([client_id ,b"", msg + b"World"])
            
gevent.spawn(serve).join()

客户端代码(模拟WSGI每个连接一个单独的线程的情况)

# -*- coding: utf-8 -*-
import gevent
import zmq.green as zmq


import logging
logging.basicConfig(level=logging.DEBUG,
                    #%(filename)s [line:%(lineno)d] %(levelname)s 
                    format='%(levelname)s  %(asctime)s %(message)s',
                    datefmt='%Y %H:%M:%S',
                    )

def client(request):
    # None of these operations will block, regardless of peer:
    context = zmq.Context()
    socket = context.socket(zmq.REQ)
    #socket.setsockopt(zmq.LINGER, 0)
    socket.connect( "tcp://localhost:5560" )
    
    socket.send( "Hello {0}  ".format(request).encode() )
    poller = zmq.Poller()
    poller.register(socket, zmq.POLLIN)
    if poller.poll(1*1000): # 1s timeout in milliseconds
        msg = socket.recv()
        logging.info ( "Received reply {0} [{1} ]".format(request, msg.decode() )  )
    else:
        logging.info ( "Timeout processing auth request"  )
       

gevent.joinall( [ gevent.spawn(client, i) for i in range(10)] )

转载于:https://my.oschina.net/cppblog/blog/536202