DimWriteStatistics.h

Go to the documentation of this file.

#ifndef FACT_DimWriteStatistics
#define FACT_DimWriteStatistics

#include <set>
#include <string>

// Keep these two together! Otheriwse it won't compile
#include <boost/bind.hpp>
#if BOOST_VERSION < 104400
#if (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ > 4))
#undef BOOST_HAS_RVALUE_REFS
#endif
#endif
#include <boost/thread.hpp>

#include "MessageImp.h"
#include "DimDescriptionService.h"

class DimWriteStatistics
{
public:
    struct Stats
    {
        uint64_t freeSpace;
        uint64_t sizeWritten;
        uint64_t rateWritten;
        uint64_t timeElapsed;

        Stats() : freeSpace(0), sizeWritten(0), rateWritten(0), timeElapsed(0) { }
    };

private:
    MessageImp &fLog;

    DimDescribedService   fDimService;

    std::string           fCurrentFolder;   
    uint16_t              fUpdateInterval;  
    size_t                fBaseSize;        
    std::set<std::string> fOpenedFiles;     

    bool fDebug;

    Stats fStats;

    boost::thread fThread;                  

    void UpdateService();

    int64_t GetFreeSpace();

    int64_t GetFileSizeOnDisk(const std::string& file) { return GetFileSizeOnDisk(file, fLog); }

    int Write(const Time &t, const std::string &txt, int qos);

public:
    DimWriteStatistics(const std::string& serverName, MessageImp &log);

    ~DimWriteStatistics();

    bool SetCurrentFolder(const std::string& folder);

    bool FileOpened(const std::string& fileName);

    void SetDebugMode(bool);
    void SetUpdateInterval(const int16_t millisec);

    const Stats &GetTotalSizeWritten() const { return fStats; }
    uint16_t GetUpdateInterval() const { return fUpdateInterval; }

    static int64_t GetFileSizeOnDisk(const std::string& file, MessageImp &imp);

    static bool DoesPathExist(std::string path, MessageImp &log);
};

#endif