KVZMQMessage.h

#ifndef KVZMQMESSAGE_H
#define KVZMQMESSAGE_H
#include "TMessage.h"
#include "TClass.h"
#include "zmq.hpp"
#include "KVConfig.h"
 
class KVZMQMessage : public TMessage {
   std::unique_ptr<TObject> fObject;
   Bool_t fReadNewObject;
   zmq::message_t fMessage;
 
public:
   KVZMQMessage()
      : TMessage(kMESS_OBJECT)
   {
      fReadNewObject = kFALSE;
   }
   KVZMQMessage(zmq::message_t& g)
      : TMessage(g.data(), g.size())
   {
      ResetBit(kIsOwner);
      fReadNewObject = kTRUE;
   }
   virtual ~KVZMQMessage() {}
#if ROOT_VERSION_CODE < ROOT_VERSION(6,12,0)
   virtual void WriteObject(const TObject* obj)
   {
      TMessage::WriteObject(obj);
#else
   virtual void WriteObject(const TObject* obj, Bool_t cacheReuse = kTRUE)
   {
      TMessage::WriteObject(obj, cacheReuse);
#endif
      SetLength();
   }
   zmq::message_t& GetMessage()
   {
      fMessage.rebuild(Buffer(), Length());
      return fMessage;
   }
   TObject* GetObject()
   {
      if (fReadNewObject) {
         fObject.reset(ReadObject(GetClass()));
         fReadNewObject = kFALSE;
      }
      return fObject.get();
   }
   template <class T>
   T* GetObject()
   {
      if (GetClass()->InheritsFrom(T::Class())) {
         return (T*)(GetObject());
      }
      return nullptr;
   }
};
 
#endif // KVMESSAGE_H