ExponentialBackOff.h

#pragma once

#include <Poco/Clock.h>
#include <Poco/Random.h>

#include "util/BackOff.h"

namespace BeeeOn {

class ExponentialBackOff : public BackOff {
public:
        typedef Poco::SharedPtr<ExponentialBackOff> Ptr;

        class Config {
        public:
                Config();

                void setInitialInterval(const Poco::Timespan &interval);
                Poco::Timespan initialInterval() const;

                void setMaxInterval(const Poco::Timespan &interval);
                Poco::Timespan maxInterval() const;

                void setMaxElapsedTime(const Poco::Timespan &time);
                Poco::Timespan maxElapsedTime() const;

                void setRandomizationFactor(double factor);
                double randomizationFactor() const;

                void setMultiplier(double multiplier);
                double multiplier() const;

        private:
                Poco::Timespan m_initialInterval;
                Poco::Timespan m_maxInterval;
                Poco::Timespan m_maxElapsedTime;
                double m_randomizationFactor;
                double m_multiplier;
        };

        ExponentialBackOff();

        void applyConfig(const Config &config);
        const Config &config() const;

        Poco::Timespan initialInterval() const;
        Poco::Timespan maxInterval() const;
        Poco::Timespan maxElapsedTime() const;
        double randomizationFactor() const;
        double multiplier() const;

        Poco::Timespan nextInterval() const;

        Poco::Timespan elapsed() const;

        Poco::Timespan next() override;
        void reset() override;

protected:
        Poco::Timespan nextRandom();

private:
        Config m_config;

        Poco::Timespan m_nextInterval;
        Poco::Clock m_resetStamp;
        Poco::Random m_random;
};

class ExponentialBackOffFactory : public BackOffFactory {
public:
        BackOff::Ptr create() override;

        void setInitialInterval(const Poco::Timespan &interval);
        void setMaxInterval(const Poco::Timespan &interval);
        void setMaxElapsedTime(const Poco::Timespan &time);
        void setRandomizationFactor(double factor);
        void setMultiplier(double multiplier);

private:
        ExponentialBackOff::Config m_config;
};

}