summaryrefslogtreecommitdiff
path: root/src/app_thread.cpp
diff options
context:
space:
mode:
authorMartin Lucina <martin@lucina.net>2012-01-23 08:53:19 +0100
committerMartin Lucina <martin@lucina.net>2012-01-23 08:53:19 +0100
commita15854bd92db69fcd0b4444fe1b8fe3610a7acf6 (patch)
tree1214b945d0f0033ff318de367c70525ea141ef56 /src/app_thread.cpp
Imported Upstream version 2.0.7.dfsgupstream/2.0.7.dfsg
Diffstat (limited to 'src/app_thread.cpp')
-rw-r--r--src/app_thread.cpp195
1 files changed, 195 insertions, 0 deletions
diff --git a/src/app_thread.cpp b/src/app_thread.cpp
new file mode 100644
index 0000000..fbf034c
--- /dev/null
+++ b/src/app_thread.cpp
@@ -0,0 +1,195 @@
+/*
+ Copyright (c) 2007-2010 iMatix Corporation
+
+ This file is part of 0MQ.
+
+ 0MQ is free software; you can redistribute it and/or modify it under
+ the terms of the Lesser GNU 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
+ Lesser GNU General Public License for more details.
+
+ You should have received a copy of the Lesser GNU General Public License
+ along with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include <new>
+#include <algorithm>
+
+#include "../include/zmq.h"
+
+#include "platform.hpp"
+
+#if defined ZMQ_HAVE_WINDOWS
+#include "windows.hpp"
+#if defined _MSC_VER
+#include <intrin.h>
+#endif
+#else
+#include <unistd.h>
+#endif
+
+#include "app_thread.hpp"
+#include "ctx.hpp"
+#include "err.hpp"
+#include "pipe.hpp"
+#include "config.hpp"
+#include "socket_base.hpp"
+#include "pair.hpp"
+#include "pub.hpp"
+#include "sub.hpp"
+#include "req.hpp"
+#include "rep.hpp"
+#include "xreq.hpp"
+#include "xrep.hpp"
+#include "upstream.hpp"
+#include "downstream.hpp"
+
+// If the RDTSC is available we use it to prevent excessive
+// polling for commands. The nice thing here is that it will work on any
+// system with x86 architecture and gcc or MSVC compiler.
+#if (defined __GNUC__ && (defined __i386__ || defined __x86_64__)) ||\
+ (defined _MSC_VER && (defined _M_IX86 || defined _M_X64))
+#define ZMQ_DELAY_COMMANDS
+#endif
+
+zmq::app_thread_t::app_thread_t (ctx_t *ctx_,
+ uint32_t thread_slot_) :
+ object_t (ctx_, thread_slot_),
+ last_processing_time (0),
+ terminated (false)
+{
+}
+
+zmq::app_thread_t::~app_thread_t ()
+{
+ zmq_assert (sockets.empty ());
+}
+
+void zmq::app_thread_t::stop ()
+{
+ send_stop ();
+}
+
+zmq::signaler_t *zmq::app_thread_t::get_signaler ()
+{
+ return &signaler;
+}
+
+bool zmq::app_thread_t::process_commands (bool block_, bool throttle_)
+{
+ bool received;
+ command_t cmd;
+ if (block_) {
+ received = signaler.recv (&cmd, true);
+ zmq_assert (received);
+ }
+ else {
+
+#if defined ZMQ_DELAY_COMMANDS
+ // Optimised version of command processing - it doesn't have to check
+ // for incoming commands each time. It does so only if certain time
+ // elapsed since last command processing. Command delay varies
+ // depending on CPU speed: It's ~1ms on 3GHz CPU, ~2ms on 1.5GHz CPU
+ // etc. The optimisation makes sense only on platforms where getting
+ // a timestamp is a very cheap operation (tens of nanoseconds).
+ if (throttle_) {
+
+ // Get timestamp counter.
+#if defined __GNUC__
+ uint32_t low;
+ uint32_t high;
+ __asm__ volatile ("rdtsc" : "=a" (low), "=d" (high));
+ uint64_t current_time = (uint64_t) high << 32 | low;
+#elif defined _MSC_VER
+ uint64_t current_time = __rdtsc ();
+#else
+#error
+#endif
+
+ // Check whether certain time have elapsed since last command
+ // processing.
+ if (current_time - last_processing_time <= max_command_delay)
+ return !terminated;
+ last_processing_time = current_time;
+ }
+#endif
+
+ // Check whether there are any commands pending for this thread.
+ received = signaler.recv (&cmd, false);
+ }
+
+ // Process all the commands available at the moment.
+ while (received) {
+ cmd.destination->process_command (cmd);
+ received = signaler.recv (&cmd, false);
+ }
+
+ return !terminated;
+}
+
+zmq::socket_base_t *zmq::app_thread_t::create_socket (int type_)
+{
+ socket_base_t *s = NULL;
+ switch (type_) {
+ case ZMQ_PAIR:
+ s = new (std::nothrow) pair_t (this);
+ break;
+ case ZMQ_PUB:
+ s = new (std::nothrow) pub_t (this);
+ break;
+ case ZMQ_SUB:
+ s = new (std::nothrow) sub_t (this);
+ break;
+ case ZMQ_REQ:
+ s = new (std::nothrow) req_t (this);
+ break;
+ case ZMQ_REP:
+ s = new (std::nothrow) rep_t (this);
+ break;
+ case ZMQ_XREQ:
+ s = new (std::nothrow) xreq_t (this);
+ break;
+ case ZMQ_XREP:
+ s = new (std::nothrow) xrep_t (this);
+ break;
+ case ZMQ_UPSTREAM:
+ s = new (std::nothrow) upstream_t (this);
+ break;
+ case ZMQ_DOWNSTREAM:
+ s = new (std::nothrow) downstream_t (this);
+ break;
+ default:
+ if (sockets.empty ())
+ get_ctx ()->no_sockets (this);
+ errno = EINVAL;
+ return NULL;
+ }
+ zmq_assert (s);
+
+ sockets.push_back (s);
+
+ return s;
+}
+
+void zmq::app_thread_t::remove_socket (socket_base_t *socket_)
+{
+ sockets.erase (socket_);
+ if (sockets.empty ())
+ get_ctx ()->no_sockets (this);
+}
+
+void zmq::app_thread_t::process_stop ()
+{
+ terminated = true;
+}
+
+bool zmq::app_thread_t::is_terminated ()
+{
+ return terminated;
+}
+