TBBMessageSvc.h

Go to the documentation of this file.

#ifndef MESSAGESVC_TBBMESSAGESVC_H
#define MESSAGESVC_TBBMESSAGESVC_H 1
// Include files
#include "MessageSvc.h"

#include "GaudiKernel/SerialTaskQueue.h"

// just needed for the thread id
#include <thread>

#include <thread>

class TBBMessageSvc: public MessageSvc {
public:
  TBBMessageSvc(const std::string& name, ISvcLocator* pSvcLocator);

  ~TBBMessageSvc() override; 

  StatusCode initialize() override;

  StatusCode finalize() override;

  using MessageSvc::reportMessage;

  void reportMessage(const Message& msg) override;

  void reportMessage(const Message& msg, int outputLevel) override;

  void reportMessage(const StatusCode& code, const std::string& source = "") override;

protected:
private:

  // ============================================================================
  // Helper tasks for message reporting.
  // ============================================================================
  class MessageTaskCommon: public Gaudi::SerialTaskQueue::WorkItem {
  public:
    MessageTaskCommon(TBBMessageSvc& svc):
      m_svc(svc),
      m_sender(std::this_thread::get_id()) {}
  protected:
    TBBMessageSvc& m_svc;
    std::thread::id m_sender;
  };

  class MessageWithLevel: public MessageTaskCommon {
  public:
    MessageWithLevel(TBBMessageSvc& svc, Message msg, int level):
      MessageTaskCommon(svc),
      m_msg(msg), m_level(level) {}
    virtual void run() override {
      m_svc.i_reportMessage(m_msg, m_level);
    }
  private:
    Message m_msg;
    int m_level;
  };

  class MessageWithoutLevel: public MessageTaskCommon {
  public:
    MessageWithoutLevel(TBBMessageSvc& svc, Message msg):
      MessageTaskCommon(svc),
      m_msg(msg) {}
    virtual void run() override {
      const int level = m_svc.outputLevel(m_msg.getSource());
      m_svc.i_reportMessage(m_msg, level);
    }
  private:
    Message m_msg;
  };

  class StatusCodeMessage: public MessageTaskCommon {
  public:
    StatusCodeMessage(TBBMessageSvc& svc, const StatusCode &sc, const std::string& source):
      MessageTaskCommon(svc),
      m_sc(sc),
      m_source(source) {}
    virtual void run() override {
      m_svc.i_reportMessage(m_sc, m_source);
    }
  private:
    StatusCode m_sc;
    std::string m_source;
  };

  Gaudi::SerialTaskQueue m_messageQueue;
};

#endif // MESSAGESVC_TBBMESSAGESVC_H