/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_6_2_7/src/EventFilter/StorageManager/interface/RegistrationCollection.h

Go to the documentation of this file.

// $Id: RegistrationCollection.h,v 1.8 2011/03/07 15:31:32 mommsen Exp $

#ifndef EventFilter_StorageManager_RegistrationCollection_h
#define EventFilter_StorageManager_RegistrationCollection_h

#include "EventFilter/StorageManager/interface/ConsumerID.h"
#include "EventFilter/StorageManager/interface/RegistrationInfoBase.h"
#include "EventFilter/StorageManager/interface/EventConsumerRegistrationInfo.h"
#include "EventFilter/StorageManager/interface/DQMEventConsumerRegistrationInfo.h"

#include <vector>
#include <map>

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

namespace stor
{

  class RegistrationCollection
  {

  public:

    RegistrationCollection();

    ~RegistrationCollection();

    ConsumerID getConsumerId();

    bool addRegistrationInfo( RegPtr );

    RegPtr getRegistrationInfo( const ConsumerID ) const;

    typedef std::vector<EventConsRegPtr> ConsumerRegistrations;
    void getEventConsumers( ConsumerRegistrations& ) const;

    typedef std::vector<DQMEventConsRegPtr> DQMConsumerRegistrations;
    void getDQMEventConsumers( DQMConsumerRegistrations& ) const;

    void enableConsumerRegistration();

    void disableConsumerRegistration();

    void clearRegistrations();

    bool registrationIsAllowed( const ConsumerID ) const;

  private:

    mutable boost::mutex lock_;

    ConsumerID nextConsumerId_;

    bool registrationAllowed_;
      
    typedef std::map<ConsumerID, RegPtr> RegistrationMap;
    RegistrationMap consumers_;

  };

  typedef boost::shared_ptr<RegistrationCollection> RegistrationCollectionPtr;

} // namespace stor

#endif // EventFilter_StorageManager_RegistrationCollection_h