Commit c2dffb99 authored by Pieter Hintjens's avatar Pieter Hintjens

Problem: threadsafe test uses CLIENT-to-CLIENT

Solution: fix to use CLIENT-to-SERVER and increase number of messages
sent to 100K per thread, to better stress thread safety.
parent d416ffce
...@@ -29,71 +29,51 @@ ...@@ -29,71 +29,51 @@
#include "testutil.hpp" #include "testutil.hpp"
void worker1(void* s); // Client threads loop on send/recv until told to exit
void worker2(void* s); void client_thread (void *client)
{
char data = 0;
for (int count = 0; count < 100000; count++) {
int rc = zmq_send (client, &data, 1, 0);
assert (rc == 1);
}
data = 1;
int rc = zmq_send (client, &data, 1, 0);
assert (rc == 1);
}
int main (void) int main (void)
{ {
setup_test_environment(); setup_test_environment ();
void *ctx = zmq_ctx_new (); void *ctx = zmq_ctx_new ();
assert (ctx); assert (ctx);
void *client = zmq_socket (ctx, ZMQ_CLIENT); void *server = zmq_socket (ctx, ZMQ_SERVER);
void *client2 = zmq_socket (ctx, ZMQ_CLIENT); int rc = zmq_bind (server, "tcp://127.0.0.1:5560");
assert (rc == 0);
void *client = zmq_socket (ctx, ZMQ_CLIENT);
int thread_safe; int thread_safe;
size_t size = sizeof(int); size_t size = sizeof (int);
zmq_getsockopt (client, ZMQ_THREAD_SAFE, &thread_safe, &size); zmq_getsockopt (client, ZMQ_THREAD_SAFE, &thread_safe, &size);
assert (thread_safe == 1); assert (thread_safe == 1);
rc = zmq_connect (client, "tcp://127.0.0.1:5560");
int rc;
rc = zmq_bind (client, "tcp://127.0.0.1:5560");
assert (rc == 0); assert (rc == 0);
rc = zmq_connect (client2, "tcp://127.0.0.1:5560"); void *t1 = zmq_threadstart (client_thread, client);
assert (rc == 0); void *t2 = zmq_threadstart (client_thread, client);
void* t1 = zmq_threadstart(worker1, client2);
void* t2 = zmq_threadstart(worker2, client2);
char data[1];
data[0] = 0;
for (int i=0; i < 10; i++) {
rc = zmq_send_const(client, data, 1, 0);
assert (rc == 1);
rc = zmq_send_const(client, data, 1, 0);
assert(rc == 1);
char a, b; char data;
int threads_completed = 0;
rc = zmq_recv(client, &a, 1, 0); while (threads_completed < 2) {
assert(rc == 1); zmq_recv (server, &data, 1, 0);
if (data == 1)
rc = zmq_recv(client, &b, 1, 0); threads_completed++; // Thread ended
assert(rc == 1);
// make sure they came from different threads
assert((a == 1 && b == 2) || (a == 2 && b == 1));
} }
zmq_threadclose (t1);
zmq_threadclose (t2);
// make the thread exit rc = zmq_close (server);
data[0] = 1;
rc = zmq_send_const(client, data, 1, 0);
assert (rc == 1);
rc = zmq_send_const(client, data, 1, 0);
assert(rc == 1);
zmq_threadclose(t1);
zmq_threadclose(t2);
rc = zmq_close (client2);
assert (rc == 0); assert (rc == 0);
rc = zmq_close (client); rc = zmq_close (client);
...@@ -104,59 +84,3 @@ int main (void) ...@@ -104,59 +84,3 @@ int main (void)
return 0 ; return 0 ;
} }
void worker1(void* s)
{
const char worker_id = 1;
char c;
while (true)
{
int rc = zmq_recv(s, &c,1, 0);
assert(rc == 1);
if (c == 0)
{
msleep(100);
rc = zmq_send_const(s,&worker_id, 1, 0);
assert(rc == 1);
}
else
{
// we got exit request
break;
}
}
}
void worker2(void* s)
{
const char worker_id = 2;
char c;
while (true)
{
int rc = zmq_recv(s, &c,1, 0);
assert(rc == 1);
assert(c == 1 || c == 0);
if (c == 0)
{
msleep(100);
rc = zmq_send_const(s,&worker_id, 1, 0);
assert(rc == 1);
}
else
{
// we got exit request
break;
}
}
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment