StreamHandler.h

Go to the documentation of this file.

// $Id: StreamHandler.h,v 1.16 2012/04/04 12:16:58 mommsen Exp $

#ifndef EventFilter_StorageManager_StreamHandler_h
#define EventFilter_StorageManager_StreamHandler_h

#include <stdint.h>

#include <boost/shared_ptr.hpp>

#include "EventFilter/StorageManager/interface/FilesMonitorCollection.h"
#include "EventFilter/StorageManager/interface/SharedResources.h"
#include "EventFilter/StorageManager/interface/StatisticsReporter.h"
#include "EventFilter/StorageManager/interface/StreamsMonitorCollection.h"


namespace stor {

  class FileHandler;
  class I2OChain;


  class StreamHandler
  {
  public:
    
    StreamHandler(const SharedResourcesPtr, const DbFileHandlerPtr);

    virtual ~StreamHandler() {};


    void closeAllFiles();

    void closeTimedOutFiles(utils::TimePoint_t currentTime =
                            utils::getCurrentTime());

    bool closeFilesForLumiSection(const uint32_t& lumiSection, std::string&);

    void writeEvent(const I2OChain& event);


  protected:

    typedef boost::shared_ptr<FileHandler> FileHandlerPtr;

    virtual std::string streamLabel() const = 0;

    virtual double fractionToDisk() const = 0;

    virtual FileHandlerPtr newFileHandler(const I2OChain& event) = 0;

    FilesMonitorCollection::FileRecordPtr getNewFileRecord(const I2OChain& event);

    virtual int getStreamMaxFileSize() const = 0;

    unsigned long long getMaxFileSize() const;


    virtual FileHandlerPtr getFileHandler(const I2OChain& event);

  private:

    bool fileTooLarge(const FileHandlerPtr, const unsigned long& dataSize) const;

    std::string getBaseFilePath(const uint32_t& runNumber, uint32_t fileCount) const;

    std::string getFileSystem(const uint32_t& runNumber, uint32_t fileCount) const;

    std::string getCoreFileName(const uint32_t& runNumber, const uint32_t& lumiSection) const;
    
    unsigned int getFileCounter(const std::string& coreFileName);


  protected:

    const SharedResourcesPtr sharedResources_;
    const StatisticsReporterPtr statReporter_;
    const StreamsMonitorCollection::StreamRecordPtr streamRecord_;
    const DiskWritingParams& diskWritingParams_;
    const DbFileHandlerPtr dbFileHandler_;

    typedef std::vector<FileHandlerPtr> FileHandlers;
    FileHandlers fileHandlers_;

    typedef std::map<std::string, unsigned int> CoreFileNamesMap;
    CoreFileNamesMap usedCoreFileNames_;
    

  private:

    //Prevent copying of the StreamHandler
    StreamHandler(StreamHandler const&);
    StreamHandler& operator=(StreamHandler const&);

  };
  
} // namespace stor

#endif // EventFilter_StorageManager_StreamHandler_h