def create_poller(): context = zmq.Context() socket = context.socket(zmq.PULL) socket.bind(address) sockets = { "master": {"socket": socket, "receive": socket.recv, "send": socket.send}, } time_func = time.time return Poller(sockets, time_func)
def create_poller(frontend_address): from cloudasr import Poller context = zmq.Context() frontend_socket = context.socket(zmq.REP) frontend_socket.bind(frontend_address) sockets = { "frontend": { "socket": frontend_socket, "receive": frontend_socket.recv, "send": frontend_socket.send }, } time_func = time.time return Poller(sockets, time_func)
def create_poller(worker_address, frontend_address): context = zmq.Context() worker_socket = context.socket(zmq.PULL) worker_socket.bind(worker_address) frontend_socket = context.socket(zmq.REP) frontend_socket.bind(frontend_address) sockets = { "worker": { "socket": worker_socket, "receive": worker_socket.recv, "send": worker_socket.send_json }, "frontend": { "socket": frontend_socket, "receive": frontend_socket.recv, "send": frontend_socket.send }, } time_func = time.time return Poller(sockets, time_func)