ZMQ:socket_send / recv阻塞

时间:2013-08-23 16:38:10

标签: c++ python python-2.7 zeromq pyzmq

所以我试图在C / C ++扩展中在Python和ZMQ之间建立一些简单的通信。 Python设置上下文,绑定inproc套接字,并将上下文和套接字名称传递给扩展。扩展设置自己的套接字,连接和侦听消息。 Python然后发送一个标题,然后将字典的字符串表示形式发送到扩展名。使用REQ / REP套接字很简单的东西。但是,由于某些原因我似乎无法找到,对socket.send的调用是阻塞的,并且扩展永远不会超过对zmq_recv的调用。我有一个测试环境,我几乎完全相同的场景播出,但套接字没有阻止,我已经三次检查代码,它应该以相同的方式工作。

PYTHON:

import zmq
import cppextension
# No lectures about using threading please. I'm restricted to, in essence
# using this function because of the code base I'm working with.
from thread import start_new_thread

socket = self.zmq_context.socket(zmq.REQ)
socket_name = "inproc://agl"
socket.bind(socket_name)
t = start_native_thread(cppextension.actor,
                        (self.zmq_context, socket_name))

test_send = {"foo": 1, "bar": 2}
# BLOCKS ON THIS LINE VVVVV
socket.send("TEST", flags=zmq.SNDMORE)
socket.send(str(test_send))
socket.recv()
socket.send("STOP")

C / C ++:

// Originally these used std::basic_string<Py_UNICODE> but I reverted
// back to normal std::string so I can use a JSON parsing library.
typedef string pystring;
typedef char pystring_t;

extern "C" PyObject *
actor(PyObject *self, PyObject *args) {
    PyObject *py_context, *py_connect_to;
    PyThreadState *_save;
    void *context;
    char *connect_to;
    void *socket;
    int rc;

    if(!PyArg_ParseTuple(args, "OO", &py_context, &py_connect_to)) {
        PyErr_SetString(PyExc_TypeError, "Expected two arguments (ZMQ context, name of socket to connect to)");
        return NULL;
    }
    py_context = PyObject_GetAttrString(py_context, "_handle");
    if(py_context == NULL) {
        PyErr_SetString(PyExc_TypeError, "Could not get '_handle' from context");
        return NULL;
    }
    if(!PyInt_Check(py_context)) {
        PyErr_SetString(PyExc_TypeError, "_handle was not an integer");
        return NULL;
    }
    context = (void*)PyInt_AsLong(py_context);
    connect_to = new char[PyString_Size(py_connect_to) + 1];
    strcpy(connect_to, PyString_AsString(py_connect_to));
    _save = PyEval_SaveThread();

    //
    // GIL-less operation BEGIN
    // ** WARNING: Do NOT call any functions that begin with 'Py', or touch any
    //    data structures that begin with 'Py' while in this section. It *WILL*
    //    blow up the Python interpreter.
    //
    socket = zmq_socket(context, ZMQ_REP);
    rc = zmq_connect(socket, connect_to);

    pystring TEST("TEST");
    pystring STOP("STOP");
    pystring SUCCESS("SUCCESS");
    pystring FAILURE("FAILURE");

    if(rc == 0) {
        int going = 1;
        // Should be able to hold a full megabyte of text, which should be enough
        // for any message being passed in.
        // Is there a way to query size of the incoming message...?
        char buffer[1000000];
        while(going) {
            // BLOCKS ON THIS LINE VVVVVV
            int size = zmq_recv(socket, buffer, 1000000, 0);
            if(size == -1) {
                // ERROR
                continue;
            }
            // Assume we don't get larger than 1MB of data. Should put a
            // check around this at some point, but not right now.
            buffer[size] = 0;

            pystring fullmsg(buffer);
            cout << "ZMQ RECIEVED: " << fullmsg << endl;
            if(fullmsg == TEST) {
                size = zmq_recv(socket, &buffer, 1000000, 0);
                if(size != -1) {
                    buffer[size] = 0;
                    pystring json_fullmsg(buffer);
                    cout << "ZMQ JSON: " << json_fullmsg << endl;
                    contacts.add(json_fullmsg);
                    zmq_send(socket, SUCCESS.c_str(), SUCCESS.size() + 1, 0);
                }
                else {
                    zmq_send(socket, FAILURE.c_str(), FAILURE.size() + 1, 0);
                }
            }
            else if(fullmsg == STOP) {
                going = 0;
                zmq_send(socket, SUCCESS.c_str(), SUCCESS.size() + 1, 0);
            }
        }
    }
    else {
        // ERROR
        int err = zmq_errno();
        switch(err) {
        case EINVAL:
            cout << "ZMQ CONNECT ERR: " << "Endpoint supplied is invalid" << endl;
            break;
        default:
            cout << "ZMQ CONNECT ERR: " << err << endl;
            break;
        }
    }
    zmq_close(socket);
    //
    // GIL-less operation END
    //

    PyEval_RestoreThread(_save);
    Py_INCREF(Py_None);
    return Py_None;
}

非常感谢任何帮助,弄清楚这里发生了什么。

编辑:另请注意,此代码将在gevent对标准库进行monkeypatched的环境中运行。这是我使用thread.start_new_thread的部分原因,因为它是在monkeypatching发生之前保存的,我想要一个真正的线程而不是绿色线程。

2 个答案:

答案 0 :(得分:3)

两件事,

因为您在修改后的版本中使用了req / rep,所以“发送,发送,接收,发送...”将会出现 不行。发送/接收必须以'锁步'方式工作(发送,接收,发送,转发)。

ZMQ_NOBLOCK将引发EAGAIN的异常,这可能意味着“套接字连接 不完整,请稍后再回来。“尝试在绑定之后放置一个计时器/睡眠 发送/ recv的。这就是导致“资源暂时不可用”信息的原因。

希望这会有所帮助

先生。 onoffon

答案 1 :(得分:1)

不确定这会有所帮助,但这里有:

  • 您确定send()阻止,这不仅仅是一个永远不会得到回复的问题吗?您可以使用ZMQ_NOBLOCK致电send()并查看是否引发了异常。如果是,那么send()确实无法将消息排入队列。
  • 您是否考虑过PAIR套接字而不是REQ/REPThe guide recommends this用于线程之间的inproc通信。