Zero MQ implimentation and example (Tutorial3)

git-svn-id: https://subversion.gsi.de/fairroot/fairbase/trunk@20162 0381ead4-6506-0410-b988-94b70fbc4730

fairmq/FairMQSocket.cxx

@@ -0,0 +1,169 @@
+/*
+ * FairMQSocket.cxx
+ *
+ *  Created on: Dec 5, 2012
+ *      Author: dklein
+ */
+
+#include "FairMQSocket.h"
+#include <sstream>
+#include "FairMQLogger.h"
+
+
+const TString FairMQSocket::TCP = "tcp://";
+const TString FairMQSocket::IPC = "ipc://";
+const TString FairMQSocket::INPROC = "inproc://";
+
+FairMQSocket::FairMQSocket(FairMQContext* context, int type, int num) :
+  fBytesTx(0),
+  fBytesRx(0),
+  fMessagesTx(0),
+  fMessagesRx(0)
+{
+  std::stringstream id;
+  id << context->GetId() << "." << GetTypeString(type) << "." << num;
+  fId = id.str();
+
+  fSocket = new zmq::socket_t(*context->GetContext(), type);
+  fSocket->setsockopt(ZMQ_IDENTITY, &fId, fId.Length());
+  if (type == ZMQ_SUB) {
+    fSocket->setsockopt(ZMQ_SUBSCRIBE, NULL, 0);
+  }
+
+  std::stringstream logmsg;
+  logmsg << "created socket #" << fId;
+  FairMQLogger::GetInstance()->Log(FairMQLogger::INFO, logmsg.str());
+}
+
+FairMQSocket::~FairMQSocket()
+{
+}
+
+TString FairMQSocket::GetId()
+{
+  return fId;
+}
+
+TString FairMQSocket::GetTypeString(Int_t type)
+{
+  switch (type) {
+  case ZMQ_SUB:
+    return "sub";
+  case ZMQ_PUB:
+    return "pub";
+  case ZMQ_PUSH:
+    return "push";
+  case ZMQ_PULL:
+    return "pull";
+  default:
+    return "";
+  }
+}
+
+Bool_t FairMQSocket::Bind(TString address)
+{
+  Bool_t result = true;
+
+  try {
+    if ( address.Length() > 0 /*!address.empty()*/) {
+      std::stringstream logmsg;
+      logmsg << "bind socket #" << fId << " on " << address;
+      FairMQLogger::GetInstance()->Log(FairMQLogger::INFO, logmsg.str());
+      fSocket->bind(address.Data());
+    }
+  } catch (zmq::error_t& e) {
+    std::stringstream logmsg2;
+    logmsg2 << "failed binding socket #" << fId << ", reason: " << e.what();
+    FairMQLogger::GetInstance()->Log(FairMQLogger::ERROR, logmsg2.str());
+    result = false;
+  }
+
+  return result;
+}
+
+Bool_t FairMQSocket::Connect(TString address)
+{
+  Bool_t result = true;
+
+  try {
+    if ( address.Length() > 0 /*!address.empty()*/) {
+      std::stringstream logmsg;
+      logmsg << "connect socket #" << fId << " to " << address;
+      FairMQLogger::GetInstance()->Log(FairMQLogger::INFO, logmsg.str());
+      fSocket->connect(address.Data());
+    }
+  } catch (zmq::error_t& e) {
+    std::stringstream logmsg2;
+    logmsg2 << "failed connecting socket #" << fId << ", reason: " << e.what();
+    FairMQLogger::GetInstance()->Log(FairMQLogger::ERROR, logmsg2.str());
+    result = false;
+  }
+
+  return result;
+}
+
+Bool_t FairMQSocket::Send(FairMQMessage* msg)
+{
+  Bool_t result = false;
+
+  try {
+    fBytesTx += msg->Size();
+    ++fMessagesTx;
+    result = fSocket->send(*msg->GetMessage());
+  } catch (zmq::error_t& e) {
+    std::stringstream logmsg;
+    logmsg << "failed sending on socket #" << fId << ", reason: " << e.what();
+    FairMQLogger::GetInstance()->Log(FairMQLogger::ERROR, logmsg.str());
+    result = false;
+  }
+
+  return result;
+}
+
+Bool_t FairMQSocket::Receive(FairMQMessage* msg)
+{
+  Bool_t result = false;
+
+  try {
+    result = fSocket->recv(msg->GetMessage());
+    fBytesRx += msg->Size();
+    ++fMessagesRx;
+  } catch (zmq::error_t& e) {
+    std::stringstream logmsg;
+    logmsg << "failed receiving on socket #" << fId << ", reason: " << e.what();
+    FairMQLogger::GetInstance()->Log(FairMQLogger::ERROR, logmsg.str());
+    result = false;
+  }
+
+  return result;
+}
+
+void FairMQSocket::Close()
+{
+  fSocket->close();
+}
+
+zmq::socket_t* FairMQSocket::GetSocket()
+{
+  return fSocket;
+}
+
+ULong_t FairMQSocket::GetBytesTx()
+{
+  return fBytesTx;
+}
+
+ULong_t FairMQSocket::GetBytesRx()
+{
+  return fBytesRx;
+}
+
+ULong_t FairMQSocket::GetMessagesTx()
+{
+  return fMessagesTx;
+}
+
+ULong_t FairMQSocket::GetMessagesRx()
+{
+  return fMessagesRx;
+}