From 4ed70a930202b103e7e80b8dc925e0aaa4622595 Mon Sep 17 00:00:00 2001 From: Martin Sustrik Date: Wed, 29 Jul 2009 12:07:54 +0200 Subject: initial commit --- include/zs.hpp | 231 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 231 insertions(+) create mode 100644 include/zs.hpp (limited to 'include/zs.hpp') diff --git a/include/zs.hpp b/include/zs.hpp new file mode 100644 index 0000000..d0f607f --- /dev/null +++ b/include/zs.hpp @@ -0,0 +1,231 @@ +/* + Copyright (c) 2007-2009 FastMQ Inc. + + 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 . +*/ + +#ifndef __ZSOCKETS_HPP_INCLUDED__ +#define __ZSOCKETS_HPP_INCLUDED__ + +#include "zs.h" + +#include + +namespace zs +{ + + typedef zs_free_fn free_fn; + + enum message_type_t + { + message_data = 1 << 0, + message_gap = 1 << ZS_GAP, + message_delimiter = 1 << ZS_DELIMITER + }; + + // A message. Caution: Don't change the body of the message once you've + // copied it - the behaviour is undefined. Don't change the body of the + // received message either - other threads may be accessing it in parallel. + + class message_t : private zs_msg + { + friend class socket_t; + + public: + + // Creates message size_ bytes long. + inline message_t (size_t size_ = 0) + { + int rc = zs_msg_init_size (this, size_); + assert (rc == 0); + } + + // Creates message from the supplied buffer. 0MQ takes care of + // deallocating the buffer once it is not needed. The deallocation + // function is supplied in ffn_ parameter. If ffn_ is NULL, no + // deallocation happens - this is useful for sending static buffers. + inline message_t (void *data_, size_t size_, + free_fn *ffn_) + { + int rc = zs_msg_init_data (this, data_, size_, ffn_); + assert (rc == 0); + } + + // Destroys the message. + inline ~message_t () + { + int rc = zs_msg_close (this); + assert (rc == 0); + } + + // Destroys old content of the message and allocates buffer for the + // new message body. Having this as a separate function allows user + // to reuse once-allocated message for multiple times. + inline void rebuild (size_t size_) + { + int rc = zs_msg_close (this); + assert (rc == 0); + rc = zs_msg_init_size (this, size_); + assert (rc == 0); + } + + // Same as above, however, the message is rebuilt from the supplied + // buffer. See appropriate constructor for discussion of buffer + // deallocation mechanism. + inline void rebuild (void *data_, size_t size_, free_fn *ffn_) + { + int rc = zs_msg_close (this); + assert (rc == 0); + rc = zs_msg_init_data (this, data_, size_, ffn_); + assert (rc == 0); + } + + // Moves the message content from one message to the another. If the + // destination message have contained data prior to the operation + // these get deallocated. The source message will contain 0 bytes + // of data after the operation. + inline void move_to (message_t *msg_) + { + int rc = zs_msg_move (this, (zs_msg*) msg_); + assert (rc == 0); + } + + // Copies the message content from one message to the another. If the + // destination message have contained data prior to the operation + // these get deallocated. + inline void copy_to (message_t *msg_) + { + int rc = zs_msg_copy (this, (zs_msg*) msg_); + assert (rc == 0); + } + + // Returns message type. + inline message_type_t type () + { + return (message_type_t) (1 << zs_msg_type (this)); + } + + // Returns pointer to message's data buffer. + inline void *data () + { + return zs_msg_data (this); + } + + // Returns the size of message data buffer. + inline size_t size () + { + return zs_msg_size (this); + } + + private: + + // Disable implicit message copying, so that users won't use shared + // messages (less efficient) without being aware of the fact. + message_t (const message_t&); + void operator = (const message_t&); + }; + + class context_t + { + friend class socket_t; + + public: + + inline context_t (int app_threads_, int io_threads_) + { + ptr = zs_init (app_threads_, io_threads_); + assert (ptr); + } + + inline ~context_t () + { + int rc = zs_term (ptr); + assert (rc == 0); + } + + private: + + void *ptr; + + // Disable copying. + context_t (const context_t&); + void operator = (const context_t&); + }; + + class socket_t + { + public: + + inline socket_t (context_t &context_, int type_ = 0) + { + ptr = zs_socket (context_.ptr, type_); + assert (ptr); + } + + inline ~socket_t () + { + int rc = zs_close (ptr); + assert (rc == 0); + } + + inline void bind (const char *addr_, zs_opts *opts_ = NULL) + { + int rc = zs_bind (ptr, addr_, opts_); + assert (rc == 0); + } + + inline void connect (const char *addr_, zs_opts *opts_ = NULL) + { + int rc = zs_connect (ptr, addr_, opts_); + assert (rc == 0); + } + + inline void subscribe (const char *criteria_) + { + int rc = zs_subscribe (ptr, criteria_); + assert (rc == 0); + } + + inline void send (message_t &msg_, int flags_ = 0) + { + int rc = zs_send (ptr, &msg_, flags_); + assert (rc == 0); + } + + inline void flush () + { + int rc = zs_flush (ptr); + assert (rc == 0); + } + + inline void recv (message_t *msg_, int flags_ = 0) + { + int rc = zs_recv (ptr, msg_, flags_); + assert (rc == 0); + } + + private: + + void *ptr; + + // Disable copying. + socket_t (const socket_t&); + void operator = (const socket_t&); + }; + +} + +#endif -- cgit v1.2.3