CMSSW_4_4_3_patch1/src/EventFilter/StorageManager/interface/ConsumerMonitorCollection.h

Go to the documentation of this file.

// $Id: ConsumerMonitorCollection.h,v 1.12 2011/03/07 15:31:31 mommsen Exp $

#ifndef EventFilter_StorageManager_ConsumerMonitorCollection_h
#define EventFilter_StorageManager_ConsumerMonitorCollection_h

#include "EventFilter/StorageManager/interface/MonitorCollection.h"
#include "EventFilter/StorageManager/interface/QueueID.h"

#include <boost/thread/mutex.hpp>
#include <boost/shared_ptr.hpp>

#include <map>

namespace stor {

  class MonitoredQuantity;


  class ConsumerMonitorCollection: public MonitorCollection
  {

  public:

    struct TotalStats
    {
      MonitoredQuantity::Stats queuedStats;
      MonitoredQuantity::Stats droppedStats;
      MonitoredQuantity::Stats servedStats;
    };

    explicit ConsumerMonitorCollection
    (
      const utils::Duration_t& updateInterval,
      const utils::Duration_t& recentDuration
    );

    void addQueuedEventSample( const QueueID&, const unsigned int& data_size );

    void addDroppedEvents( const QueueID&, const size_t& count );

    void addServedEventSample( const QueueID&, const unsigned int& data_size );

    bool getQueued( const QueueID& qid, MonitoredQuantity::Stats& result ) const;

    bool getServed( const QueueID& qid, MonitoredQuantity::Stats& result ) const;

    bool getDropped( const QueueID& qid, MonitoredQuantity::Stats& result ) const;

    void getTotalStats( TotalStats& ) const;

    void resetCounters();

  private:

    // Prevent copying:
    ConsumerMonitorCollection( const ConsumerMonitorCollection& );
    ConsumerMonitorCollection& operator = ( const ConsumerMonitorCollection& );

    typedef std::map<QueueID, MonitoredQuantityPtr> ConsStatMap;

    void addEventSampleToMap( const QueueID&, const unsigned int& data_size, ConsStatMap& );
    bool getValueFromMap( const QueueID&, MonitoredQuantity::Stats&, const ConsStatMap& ) const;

    virtual void do_calculateStatistics();
    virtual void do_reset();

    const utils::Duration_t updateInterval_;
    const utils::Duration_t recentDuration_;
    MonitoredQuantity totalQueuedMQ_;
    MonitoredQuantity totalDroppedMQ_;
    MonitoredQuantity totalServedMQ_;

  protected:

    ConsStatMap qmap_; // queued
    ConsStatMap dmap_; // dropped
    ConsStatMap smap_; // served

    mutable boost::mutex mutex_;

  };

} // namespace stor

#endif // EventFilter_StorageManager_ConsumerMonitorCollection_h