MemoryAuditor.cpp

Go to the documentation of this file.

/***********************************************************************************\
* (c) Copyright 1998-2024 CERN for the benefit of the LHCb and ATLAS collaborations *
*                                                                                   *
* This software is distributed under the terms of the Apache version 2 licence,     *
* copied verbatim in the file "LICENSE".                                            *
*                                                                                   *
* In applying this licence, CERN does not waive the privileges and immunities       *
* granted to it by virtue of its status as an Intergovernmental Organization        *
* or submit itself to any jurisdiction.                                             *
\***********************************************************************************/
// MemoryAuditor:
//  An auditor that monitors memory usage
 
#include "CommonAuditor.h"
#include "ProcStats.h"
#include <GaudiKernel/MsgStream.h>
 
class MemoryAuditor : public CommonAuditor {
public:
  using CommonAuditor::CommonAuditor;
 
protected:
  void i_before( CustomEventTypeRef evt, std::string_view caller ) override;
 
  void i_after( CustomEventTypeRef evt, std::string_view caller, const StatusCode& sc ) override;
 
  virtual void i_printinfo( std::string_view msg, CustomEventTypeRef evt, std::string_view caller );
};
 
DECLARE_COMPONENT( MemoryAuditor )
 
void MemoryAuditor::i_before( CustomEventTypeRef evt, std::string_view caller ) {
  i_printinfo( "Memory usage before", evt, caller );
}
 
void MemoryAuditor::i_after( CustomEventTypeRef evt, std::string_view caller, const StatusCode& ) {
  i_printinfo( "Memory usage has changed after", evt, caller );
}
 
void MemoryAuditor::i_printinfo( std::string_view msg, CustomEventTypeRef evt, std::string_view caller ) {
  if ( procInfo pInfo; ProcStats::instance()->fetch( pInfo ) ) {
    if ( pInfo.vsize > 0 ) {
      info() << msg << " " << caller << " " << evt << " virtual size = " << pInfo.vsize << " MB"
             << " resident set size = " << pInfo.rss << " MB" << endmsg;
    }
  }
}