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