ratelimiter.h

Go to the documentation of this file.

#ifndef RATELIMITER_H
#define RATELIMITER_H
 
#include <QList>
#include <QMap>
#include <QMutex>
#include <QString>

class RateLimiter
{
public:
    bool recordAttempt(const QString &key, int maxAttempts, int windowSeconds);

    bool recordAttemptAt(const QString &key, int maxAttempts, int windowSeconds, qint64 now);

    bool isBlocked(const QString &key, int maxAttempts, int windowSeconds) const;

    bool isBlockedAt(const QString &key, int maxAttempts, int windowSeconds, qint64 now) const;

    void clearAttempts(const QString &key);

    int retryAfterSeconds(const QString &key, int maxAttempts, int windowSeconds) const;

    int retryAfterSecondsAt(const QString &key, int maxAttempts, int windowSeconds, qint64 now) const;
 
private:
    struct AttemptHistory
    {
        QList<qint64> timestamps; // epoch seconds, oldest first
        int windowSeconds;
    };

    void pruneLocked(qint64 now);
 
    mutable QMutex mutex;
    QMap<QString, AttemptHistory> attempts; // key -> attempt history
    qint64 lastPruneSecs = 0;
    static constexpr int PruneIntervalSeconds = 60;
};
 
#endif