summaryrefslogtreecommitdiff
path: root/src/pipe_writer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/pipe_writer.cpp')
-rw-r--r--src/pipe_writer.cpp120
1 files changed, 120 insertions, 0 deletions
diff --git a/src/pipe_writer.cpp b/src/pipe_writer.cpp
new file mode 100644
index 0000000..173cf97
--- /dev/null
+++ b/src/pipe_writer.cpp
@@ -0,0 +1,120 @@
+/*
+ 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 <http://www.gnu.org/licenses/>.
+*/
+
+#include "../include/zs.h"
+
+#include "pipe_writer.hpp"
+#include "pipe.hpp"
+#include "i_demux.hpp"
+
+zs::pipe_writer_t::pipe_writer_t (object_t *parent_, pipe_t *pipe_,
+ object_t *peer_, uint64_t hwm_, uint64_t lwm_) :
+ object_t (parent_),
+ pipe (pipe_),
+ peer (peer_),
+ demux (NULL),
+ index (-1),
+ hwm (hwm_),
+ lwm (lwm_),
+ head (0),
+ tail (0)
+{
+}
+
+zs::pipe_writer_t::~pipe_writer_t ()
+{
+}
+
+void zs::pipe_writer_t::set_demux (i_demux *demux_)
+{
+ demux = demux_;
+}
+
+void zs::pipe_writer_t::set_index (int index_)
+{
+ index = index_;
+}
+
+int zs::pipe_writer_t::get_index ()
+{
+ return index;
+}
+
+bool zs::pipe_writer_t::write (zs_msg *msg_)
+{
+ size_t msg_size = zs_msg_size (msg_);
+
+ // If message won't fit into the in-memory pipe, there's no way
+ // to pass it further.
+ // TODO: It should be discarded and 'oversized' notification should be
+ // placed into the pipe.
+ zs_assert (!hwm || msg_size <= hwm);
+
+ // If there's not enough space in the pipe at the moment, return false.
+ if (hwm && tail + msg_size - head > hwm)
+ return false;
+
+ // Write the message to the pipe and adjust tail position.
+ pipe->write (*msg_);
+ flush ();
+ tail += msg_size;
+
+ return true;
+}
+
+void zs::pipe_writer_t::flush ()
+{
+ if (!pipe->flush ())
+ send_tail (peer, tail);
+}
+
+void zs::pipe_writer_t::process_head (uint64_t bytes_)
+{
+ head = bytes_;
+}
+
+void zs::pipe_writer_t::terminate ()
+{
+ // Disconnect from the associated demux.
+ if (demux) {
+ demux->detach_pipe (this);
+ demux = NULL;
+ }
+
+ // Push the delimiter to the pipe. Delimiter is a notification for pipe
+ // reader that there will be no more messages in the pipe.
+ zs_msg delimiter;
+ delimiter.content = (zs_msg_content*) ZS_DELIMITER;
+ delimiter.shared = false;
+ delimiter.vsm_size = 0;
+ pipe->write (delimiter);
+ flush ();
+}
+
+void zs::pipe_writer_t::process_terminate ()
+{
+ // Disconnect from the associated demux.
+ if (demux) {
+ demux->detach_pipe (this);
+ demux = NULL;
+ }
+
+ // Send termination acknowledgement to the pipe reader.
+ send_terminate_ack (peer);
+}