summaryrefslogtreecommitdiff
path: root/perf/inproc_lat.cpp
diff options
context:
space:
mode:
authorMartin Lucina <mato@kotelna.sk>2011-05-09 11:20:08 +0200
committerMartin Lucina <martin@lucina.net>2012-01-23 08:53:47 +0100
commitf34d1599a651dd0b8feba2397f87629733988384 (patch)
tree07694c4ed96bf7b0ecadebc76d6da3f6c0125219 /perf/inproc_lat.cpp
parent8eea9178e40abf16c0e469465a58fba3beaaf754 (diff)
parentcbaa7cfa93893876e4fd8794b6ea39f4d245b6b5 (diff)
Imported Debian patch 2.1.6-1debian/2.1.6-1
Diffstat (limited to 'perf/inproc_lat.cpp')
-rw-r--r--perf/inproc_lat.cpp232
1 files changed, 232 insertions, 0 deletions
diff --git a/perf/inproc_lat.cpp b/perf/inproc_lat.cpp
new file mode 100644
index 0000000..512fa94
--- /dev/null
+++ b/perf/inproc_lat.cpp
@@ -0,0 +1,232 @@
+/*
+ Copyright (c) 2007-2011 iMatix Corporation
+ Copyright (c) 2007-2011 Other contributors as noted in the AUTHORS file
+
+ This file is part of 0MQ.
+
+ 0MQ is free software; you can redistribute it and/or modify it under
+ the terms of the GNU Lesser General Public License as published by
+ the Free Software Foundation; either version 3 of the License, or
+ (at your option) any later version.
+
+ 0MQ is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU Lesser General Public License for more details.
+
+ You should have received a copy of the GNU Lesser General Public License
+ along with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "../include/zmq.h"
+#include "../include/zmq_utils.h"
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include "../src/platform.hpp"
+
+#if defined ZMQ_HAVE_WINDOWS
+#include <windows.h>
+#include <process.h>
+#else
+#include <pthread.h>
+#endif
+
+static size_t message_size;
+static int roundtrip_count;
+
+#if defined ZMQ_HAVE_WINDOWS
+static unsigned int __stdcall worker (void *ctx_)
+#else
+static void *worker (void *ctx_)
+#endif
+{
+ void *s;
+ int rc;
+ int i;
+ zmq_msg_t msg;
+
+ s = zmq_socket (ctx_, ZMQ_REP);
+ if (!s) {
+ printf ("error in zmq_socket: %s\n", zmq_strerror (errno));
+ exit (1);
+ }
+
+ rc = zmq_connect (s, "inproc://lat_test");
+ if (rc != 0) {
+ printf ("error in zmq_connect: %s\n", zmq_strerror (errno));
+ exit (1);
+ }
+
+ rc = zmq_msg_init (&msg);
+ if (rc != 0) {
+ printf ("error in zmq_msg_init: %s\n", zmq_strerror (errno));
+ exit (1);
+ }
+
+ for (i = 0; i != roundtrip_count; i++) {
+ rc = zmq_recv (s, &msg, 0);
+ if (rc < 0) {
+ printf ("error in zmq_recv: %s\n", zmq_strerror (errno));
+ exit (1);
+ }
+ rc = zmq_send (s, &msg, 0);
+ if (rc < 0) {
+ printf ("error in zmq_send: %s\n", zmq_strerror (errno));
+ exit (1);
+ }
+ }
+
+ rc = zmq_msg_close (&msg);
+ if (rc != 0) {
+ printf ("error in zmq_msg_close: %s\n", zmq_strerror (errno));
+ exit (1);
+ }
+
+ rc = zmq_close (s);
+ if (rc != 0) {
+ printf ("error in zmq_close: %s\n", zmq_strerror (errno));
+ exit (1);
+ }
+
+#if defined ZMQ_HAVE_WINDOWS
+ return 0;
+#else
+ return NULL;
+#endif
+}
+
+int main (int argc, char *argv [])
+{
+#if defined ZMQ_HAVE_WINDOWS
+ HANDLE local_thread;
+#else
+ pthread_t local_thread;
+#endif
+ void *ctx;
+ void *s;
+ int rc;
+ int i;
+ zmq_msg_t msg;
+ void *watch;
+ unsigned long elapsed;
+ double latency;
+
+ if (argc != 3) {
+ printf ("usage: inproc_lat <message-size> <roundtrip-count>\n");
+ return 1;
+ }
+
+ message_size = atoi (argv [1]);
+ roundtrip_count = atoi (argv [2]);
+
+ ctx = zmq_init (1);
+ if (!ctx) {
+ printf ("error in zmq_init: %s\n", zmq_strerror (errno));
+ return -1;
+ }
+
+ s = zmq_socket (ctx, ZMQ_REQ);
+ if (!s) {
+ printf ("error in zmq_socket: %s\n", zmq_strerror (errno));
+ return -1;
+ }
+
+ rc = zmq_bind (s, "inproc://lat_test");
+ if (rc != 0) {
+ printf ("error in zmq_bind: %s\n", zmq_strerror (errno));
+ return -1;
+ }
+
+#if defined ZMQ_HAVE_WINDOWS
+ local_thread = (HANDLE) _beginthreadex (NULL, 0,
+ worker, ctx, 0 , NULL);
+ if (local_thread == 0) {
+ printf ("error in _beginthreadex\n");
+ return -1;
+ }
+#else
+ rc = pthread_create (&local_thread, NULL, worker, ctx);
+ if (rc != 0) {
+ printf ("error in pthread_create: %s\n", zmq_strerror (rc));
+ return -1;
+ }
+#endif
+
+ rc = zmq_msg_init_size (&msg, message_size);
+ if (rc != 0) {
+ printf ("error in zmq_msg_init_size: %s\n", zmq_strerror (errno));
+ return -1;
+ }
+ memset (zmq_msg_data (&msg), 0, message_size);
+
+ printf ("message size: %d [B]\n", (int) message_size);
+ printf ("roundtrip count: %d\n", (int) roundtrip_count);
+
+ watch = zmq_stopwatch_start ();
+
+ for (i = 0; i != roundtrip_count; i++) {
+ rc = zmq_send (s, &msg, 0);
+ if (rc < 0) {
+ printf ("error in zmq_send: %s\n", zmq_strerror (errno));
+ return -1;
+ }
+ rc = zmq_recv (s, &msg, 0);
+ if (rc < 0) {
+ printf ("error in zmq_recv: %s\n", zmq_strerror (errno));
+ return -1;
+ }
+ if (zmq_msg_size (&msg) != message_size) {
+ printf ("message of incorrect size received\n");
+ return -1;
+ }
+ }
+
+ elapsed = zmq_stopwatch_stop (watch);
+
+ rc = zmq_msg_close (&msg);
+ if (rc != 0) {
+ printf ("error in zmq_msg_close: %s\n", zmq_strerror (errno));
+ return -1;
+ }
+
+ latency = (double) elapsed / (roundtrip_count * 2);
+
+#if defined ZMQ_HAVE_WINDOWS
+ DWORD rc2 = WaitForSingleObject (local_thread, INFINITE);
+ if (rc2 == WAIT_FAILED) {
+ printf ("error in WaitForSingleObject\n");
+ return -1;
+ }
+ BOOL rc3 = CloseHandle (local_thread);
+ if (rc3 == 0) {
+ printf ("error in CloseHandle\n");
+ return -1;
+ }
+#else
+ rc = pthread_join (local_thread, NULL);
+ if (rc != 0) {
+ printf ("error in pthread_join: %s\n", zmq_strerror (rc));
+ return -1;
+ }
+#endif
+
+ printf ("average latency: %.3f [us]\n", (double) latency);
+
+ rc = zmq_close (s);
+ if (rc != 0) {
+ printf ("error in zmq_close: %s\n", zmq_strerror (errno));
+ return -1;
+ }
+
+ rc = zmq_term (ctx);
+ if (rc != 0) {
+ printf ("error in zmq_term: %s\n", zmq_strerror (errno));
+ return -1;
+ }
+
+ return 0;
+}
+