所以我试图在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发生之前保存的,我想要一个真正的线程而不是绿色线程。
答案 0 :(得分:3)
两件事,
因为您在修改后的版本中使用了req / rep,所以“发送,发送,接收,发送...”将会出现 不行。发送/接收必须以'锁步'方式工作(发送,接收,发送,转发)。
ZMQ_NOBLOCK将引发EAGAIN的异常,这可能意味着“套接字连接 不完整,请稍后再回来。“尝试在绑定之后放置一个计时器/睡眠 发送/ recv的。这就是导致“资源暂时不可用”信息的原因。
希望这会有所帮助
先生。 onoffon
答案 1 :(得分:1)
不确定这会有所帮助,但这里有:
send()
阻止,这不仅仅是一个永远不会得到回复的问题吗?您可以使用ZMQ_NOBLOCK
致电send()
并查看是否引发了异常。如果是,那么send()
确实无法将消息排入队列。PAIR
套接字而不是REQ/REP
? The guide recommends this用于线程之间的inproc
通信。