Колбы-розетки и ZeroMQ

Я играю с получением zeromq сообщения pub/sub отправляются в браузер с помощью веб-сокета. Следующее "работает" в том, что сообщения отправляются через веб-сокет. Тем не менее, попытка перезагрузить страницу просто зависает, так как я предполагаю, while True цикл блокирует. Я думал, что gevent.sleep() вызов позволил бы переключение контекста, но, очевидно, нет. Есть идеи, как заставить все эти части работать вместе?

import zmq
import json
import gevent
from flask_sockets import Sockets
from flask import Flask, render_template
import logging
from gevent import monkey

monkey.patch_all()

app = Flask(__name__)
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

sockets = Sockets(app)
context = zmq.Context()

ZMQ_LISTENING_PORT = 6557

@app.route('/')
def index():
    return render_template('index.html')

@sockets.route('/zeromq')
def send_data(ws):
    logger.info('Got a websocket connection, sending up data from zmq')
    socket = context.socket(zmq.SUB)
    socket.connect('tcp://localhost:{PORT}'.format(PORT=ZMQ_LISTENING_PORT))
    socket.setsockopt(zmq.SUBSCRIBE, "")
    gevent.sleep()
    while True:
        data = socket.recv_json()
        logger.info(data)
        ws.send(json.dumps(data))
        gevent.sleep()

if __name__ == '__main__':
    from gevent import pywsgi
    from geventwebsocket.handler import WebSocketHandler
    server = pywsgi.WSGIServer(('', 25000), app, handler_class=WebSocketHandler)
    server.serve_forever()

1 ответ

Решение

Фигу, только что понял надо сделать

import zmq.green as zmq

для того, чтобы использовать с совместимостью Gevent. Вот ссылка на документацию pyZeormq.

Другие вопросы по тегам