MRPT  2.0.1
zmq_serialization.h
Go to the documentation of this file.
1 /* +------------------------------------------------------------------------+
2  | Mobile Robot Programming Toolkit (MRPT) |
3  | https://www.mrpt.org/ |
4  | |
5  | Copyright (c) 2005-2020, Individual contributors, see AUTHORS file |
6  | See: https://www.mrpt.org/Authors - All rights reserved. |
7  | Released under BSD License. See: https://www.mrpt.org/License |
8  +------------------------------------------------------------------------+ */
9 #pragma once
10 
11 #include <mrpt/io/CMemoryStream.h>
13 #include <cmath> // ceil()
14 
15 namespace mrpt
16 {
17 namespace serialization
18 {
19 /** \addtogroup noncstream_serialization_zmq Serialization functions for ZMQ (v3
20  * or above) (in #include <mrpt/serialization/serialization_zmq.h>)
21  * \ingroup noncstream_serialization
22  * @{ */
23 
24 /** Send an MRPT object to a ZMQ socket.
25  * \param[in] obj The object to be serialized and sent to the socket.
26  * \param[in] zmq_socket The zmq socket object.
27  * \param[in] max_packet_len The object will be split into a series of ZMQ
28  * "message parts" of this maximum length (in bytes). Default=0, which means do
29  * not split in parts.
30  * \note Including `<mrpt/serialization/serialization_zmq.h>` requires libzmq to
31  * be
32  * available in your system and linked
33  * to your user code. This function can be used even if MRPT was built without
34  * ZMQ support, thanks to the use of templates.
35  * \exception std::exception If the object finds any critical error during
36  * serialization or on ZMQ errors.
37  * \note See examples of usage in
38  * https://github.com/MRPT/mrpt/tree/master/doc/mrpt-zeromq-example
39  */
40 template <typename ZMQ_SOCKET_TYPE>
42  ZMQ_SOCKET_TYPE zmq_socket, const mrpt::serialization::CSerializable& obj,
43  const size_t max_packet_len = 0)
44 {
46  if (!buf) throw std::bad_alloc();
47 
48  buf->WriteObject(&obj);
49  const size_t nBytes = buf->getTotalBytesCount();
50  if (!nBytes)
51  throw std::runtime_error(
52  "[mrpt_send_to_zmq] Serialized object has 0 bytes, which probably "
53  "means something went wrong...");
54  unsigned int nPkts =
55  (!max_packet_len)
56  ? 1U
57  : static_cast<unsigned int>(ceil(double(nBytes) / max_packet_len));
58  for (unsigned int iPkt = 0; iPkt < nPkts; ++iPkt)
59  {
60  // Prepare a msg part:
61  mrpt::serialization::internal::TFreeFnDataForZMQ* fd =
62  new mrpt::serialization::internal::TFreeFnDataForZMQ();
63  if (!fd) throw std::bad_alloc();
64  fd->buf = buf;
65  fd->do_free =
66  iPkt ==
67  (nPkts - 1); // Free buffer only after the last part is disposed.
68  void* pkt_data = reinterpret_cast<char*>(fd->buf->getRawBufferData()) +
69  max_packet_len * iPkt;
70  size_t nBytesThisPkt = nBytes - max_packet_len * iPkt;
71  if (max_packet_len != 0 && nBytesThisPkt > max_packet_len)
72  nBytesThisPkt = max_packet_len;
73  // Build ZMQ msg:
74  zmq_msg_t message;
75  if (0 != zmq_msg_init_data(
76  &message, pkt_data, nBytesThisPkt,
78  throw std::runtime_error(
79  "[mrpt_send_to_zmq] Error in zmq_msg_init_data()");
80  // Send:
81  const int sent_size =
82  zmq_msg_send(&message, zmq_socket, fd->do_free ? 0 : ZMQ_SNDMORE);
83  if (0 != zmq_msg_close(&message))
84  throw std::runtime_error(
85  "[mrpt_send_to_zmq] Error in zmq_msg_close()");
86  if (sent_size != static_cast<int>(nBytesThisPkt))
87  throw std::runtime_error(
88  "[mrpt_send_to_zmq] Error in zmq_msg_send()");
89  }
90 }
91 
92 /** Users may normally call mrpt_recv_from_zmq() and mrpt_recv_from_zmq_into().
93  * This function just stores the received data into a memory buffer without
94  * parsing it into an MRPT object.
95  * \return false on any error */
96 template <typename ZMQ_SOCKET_TYPE, typename VECTOR_MSG_T>
98  ZMQ_SOCKET_TYPE zmq_socket, VECTOR_MSG_T& out_lst_msgs,
99  mrpt::io::CMemoryStream& target_buf, bool dont_wait,
100  size_t* rx_obj_length_in_bytes)
101 {
102  if (rx_obj_length_in_bytes) *rx_obj_length_in_bytes = 0;
103  out_lst_msgs.clear();
104  target_buf.clear();
105  int64_t more;
106  size_t more_size = sizeof(more);
107  do
108  {
109  // Init rx msg:
110  zmq_msg_t* msg = new zmq_msg_t();
111  if (0 != zmq_msg_init(msg)) return false;
112  out_lst_msgs.push_back(msg);
113  // Recv:
114  int rc = zmq_msg_recv(msg, zmq_socket, dont_wait ? ZMQ_DONTWAIT : 0);
115  if (rc == -1) return false;
116  // Determine if more message parts are to follow
117  rc = zmq_getsockopt(zmq_socket, ZMQ_RCVMORE, &more, &more_size);
118  if (rc != 0) return false;
119  // Only one part?
120  if (out_lst_msgs.size() == 1 && !more)
121  {
122  target_buf.assignMemoryNotOwn(zmq_msg_data(msg), zmq_msg_size(msg));
123  if (rx_obj_length_in_bytes)
124  *rx_obj_length_in_bytes = zmq_msg_size(msg);
125  }
126  } while (more);
127  // More than 1 part?
128  if (out_lst_msgs.size() > 1)
129  {
130  for (size_t i = 0; i < out_lst_msgs.size(); i++)
131  {
132  target_buf.WriteBuffer(
133  zmq_msg_data(out_lst_msgs[i]), zmq_msg_size(out_lst_msgs[i]));
134  }
135  if (rx_obj_length_in_bytes)
136  *rx_obj_length_in_bytes = target_buf.getTotalBytesCount();
137  target_buf.Seek(0);
138  }
139  return true;
140 }
141 
142 namespace internal
143 {
144 template <typename VECTOR_MSG_T>
145 void free_zmq_msg_lst(VECTOR_MSG_T& lst_msgs)
146 {
147  for (size_t i = 0; i < lst_msgs.size(); ++i)
148  {
149  zmq_msg_close(lst_msgs[i]);
150  delete lst_msgs[i];
151  }
152 }
153 } // namespace internal
154 
155 /** Receives an MRPT object from a ZMQ socket, determining the type of the
156  * object on-the-fly.
157  * \param[in] zmq_socket The zmq socket object.
158  * \param[in] dont_wait If true, will fail if there is no data ready to
159  * be read. If false (default) this function will block until data arrives.
160  * \param[out] rx_obj_length_in_bytes If non-nullptr, the object length will be
161  * stored here.
162  * \return An empty smart pointer if there was any error. The received
163  * object if all went OK.
164  * \note Including `<mrpt/serialization/serialization_zmq.h>` requires libzmq to
165  * be
166  * available in your system and linked to your user code. This function
167  * can be used even if MRPT was built without ZMQ support, thanks to the
168  * use of templates.
169  * \exception std::exception If the object finds any critical error during
170  * de-serialization.
171  * \sa mrpt_recv_from_zmq_into
172  * \note See examples of usage in
173  * https://github.com/MRPT/mrpt/tree/master/doc/mrpt-zeromq-example
174  */
175 template <typename ZMQ_SOCKET_TYPE>
177  ZMQ_SOCKET_TYPE zmq_socket, bool dont_wait = false,
178  size_t* rx_obj_length_in_bytes = nullptr)
179 {
180  CMemoryStream target_buf;
182  std::vector<zmq_msg_t*> lst_msgs_to_close;
184  zmq_socket, lst_msgs_to_close, target_buf, dont_wait,
185  rx_obj_length_in_bytes))
186  return obj;
187  // De-serialize:
188  obj = target_buf.ReadObject();
189  internal::free_zmq_msg_lst(lst_msgs_to_close); // Free msgs mem
190  return obj;
191 }
192 /** Like mrpt_recv_from_zmq() but without dynamically allocating the received
193  * object,
194  * more efficient to use if the type of the received object is known in
195  * advance.
196  * \param[in] target_object The received object will be stored here. An
197  * exception will be raised upon type mismatch.
198  * \return true if all was OK, false on any ZMQ error.
199  * \sa mrpt_recv_from_zmq() for details on the rest of parameters.
200  * \note See examples of usage in
201  * https://github.com/MRPT/mrpt/tree/master/doc/mrpt-zeromq-example
202  */
203 template <typename ZMQ_SOCKET_TYPE>
205  ZMQ_SOCKET_TYPE zmq_socket,
206  mrpt::serialization::CSerializable& target_object, bool dont_wait = false,
207  size_t* rx_obj_length_in_bytes = nullptr)
208 {
209  CMemoryStream target_buf;
210  std::vector<zmq_msg_t*> lst_msgs_to_close;
212  zmq_socket, lst_msgs_to_close, target_buf, dont_wait,
213  rx_obj_length_in_bytes))
214  return false;
215  // De-serialize:
216  target_buf.ReadObject(&target_object);
217  internal::free_zmq_msg_lst(lst_msgs_to_close); // Free msgs mem
218  return true;
219 }
220 
221 /** @} */
222 } // namespace serialization
223 } // namespace mrpt
void free_fn_for_zmq(void *data, void *hint)
Used in mrpt_send_to_zmq().
mrpt::serialization::CSerializable::Ptr mrpt_recv_from_zmq(ZMQ_SOCKET_TYPE zmq_socket, bool dont_wait=false, size_t *rx_obj_length_in_bytes=nullptr)
Receives an MRPT object from a ZMQ socket, determining the type of the object on-the-fly.
mrpt::io::CMemoryStream CMemoryStream
void mrpt_send_to_zmq(ZMQ_SOCKET_TYPE zmq_socket, const mrpt::serialization::CSerializable &obj, const size_t max_packet_len=0)
Send an MRPT object to a ZMQ socket.
uint64_t getTotalBytesCount() const override
Returns the total size of the internal buffer.
This CStream derived class allow using a memory buffer as a CStream.
void assignMemoryNotOwn(const void *data, const uint64_t nBytesInData)
Initilize the data in the stream from a block of memory which is NEITHER OWNED NOR COPIED by the obje...
uint64_t Seek(int64_t Offset, CStream::TSeekOrigin Origin=sFromBeginning) override
Introduces a pure virtual method for moving to a specified position in the streamed resource...
This is the global namespace for all Mobile Robot Programming Toolkit (MRPT) libraries.
The virtual base class which provides a unified interface for all persistent objects in MRPT...
Definition: CSerializable.h:30
void free_zmq_msg_lst(VECTOR_MSG_T &lst_msgs)
bool mrpt_recv_from_zmq_buf(ZMQ_SOCKET_TYPE zmq_socket, VECTOR_MSG_T &out_lst_msgs, mrpt::io::CMemoryStream &target_buf, bool dont_wait, size_t *rx_obj_length_in_bytes)
Users may normally call mrpt_recv_from_zmq() and mrpt_recv_from_zmq_into().
void clear()
Clears the memory buffer.
bool mrpt_recv_from_zmq_into(ZMQ_SOCKET_TYPE zmq_socket, mrpt::serialization::CSerializable &target_object, bool dont_wait=false, size_t *rx_obj_length_in_bytes=nullptr)
Like mrpt_recv_from_zmq() but without dynamically allocating the received object, more efficient to u...



Page generated by Doxygen 1.8.14 for MRPT 2.0.1 Git: 0fef1a6d7 Fri Apr 3 23:00:21 2020 +0200 at vie abr 3 23:20:28 CEST 2020