StorageSizeWithTimeLimitLogUploadStrategy.hpp

Go to the documentation of this file.

#ifndef STORAGESIZEWITHTIMELIMITLOGUPLOADSTRATEGY_HPP_
#define STORAGESIZEWITHTIMELIMITLOGUPLOADSTRATEGY_HPP_

#include <cstdlib>
#include <chrono>

#include "kaa/logging/Log.hpp"
#include "kaa/log/ILogStorageStatus.hpp"
#include "kaa/log/DefaultLogUploadStrategy.hpp"

namespace kaa {

class StorageSizeWithTimeLimitLogUploadStrategy : public DefaultLogUploadStrategy {
public:
    StorageSizeWithTimeLimitLogUploadStrategy(std::size_t volumeThreshold, std::size_t logUploadCheckReriod, IKaaClientContext &context)
        : DefaultLogUploadStrategy(context), lastUploadTime_(Clock::now())
    {
        setVolumeThreshold(volumeThreshold);
        setLogUploadCheckPeriod(logUploadCheckReriod);
    }

    virtual LogUploadStrategyDecision isUploadNeeded(ILogStorageStatus& status) override
    {
        auto now = Clock::now();
        auto currentConsumedVolume =  status.getConsumedVolume();

        if ((currentConsumedVolume >= uploadVolumeThreshold_) || (now >= (lastUploadTime_ + std::chrono::seconds(logUploadCheckReriod_)))) {
            KAA_LOG_INFO(boost::format("Need to upload logs - current size: %llu, threshold: %llu, lastUploadedTime: %llu, timeLimit: %llu sec")
                        % currentConsumedVolume % uploadVolumeThreshold_ % lastUploadTime_.time_since_epoch().count() % logUploadCheckReriod_);
            lastUploadTime_ = now;
            return LogUploadStrategyDecision::UPLOAD;
        }

        return LogUploadStrategyDecision::NOOP;
    }

private:
    typedef std::chrono::system_clock Clock;
    std::chrono::time_point<Clock> lastUploadTime_;
};

}

#endif /* STORAGESIZEWITHTIMELIMITLOGUPLOADSTRATEGY_HPP_ */