ratetracker.h

Go to the documentation of this file.

 
#pragma once
 
 
#include <promeki/config.h>
#if PROMEKI_ENABLE_CORE
#include <promeki/namespace.h>
#include <promeki/atomic.h>
#include <promeki/mutex.h>
#include <promeki/timestamp.h>
 
PROMEKI_NAMESPACE_BEGIN
 
class RateTracker {
        public:
                static constexpr int64_t kDefaultWindowMs = 5000;
 
                explicit RateTracker(int64_t windowMs = kDefaultWindowMs);
 
                RateTracker(const RateTracker &) = delete;
                RateTracker &operator=(const RateTracker &) = delete;
                RateTracker(RateTracker &&) = delete;
                RateTracker &operator=(RateTracker &&) = delete;
 
                void record(int64_t bytes);
 
                double bytesPerSecond() const;
 
                double framesPerSecond() const;
 
                void reset();
 
                int64_t windowMs() const { return _windowMs; }
 
        private:
                void rotateIfStale(int64_t nowNs) const;
 
                int64_t                 _windowMs;
                mutable Atomic<int64_t> _bytes;
                mutable Atomic<int64_t> _frames;
                mutable Mutex           _mutex;
                mutable int64_t         _windowStartNs = 0;
                mutable int64_t         _lastWindowBytes = 0;
                mutable int64_t         _lastWindowFrames = 0;
                mutable int64_t         _lastWindowElapsedNs = 0;
                mutable bool            _haveLastWindow = false;
};
 
PROMEKI_NAMESPACE_END
 
#endif // PROMEKI_ENABLE_CORE