RateLimiter.h

Go to the documentation of this file.

// Copyright (c) Qualcomm Technologies, Inc. and/or its subsidiaries.
// SPDX-License-Identifier: BSD-3-Clause-Clear
 
#ifndef RATE_LIMITER_H
#define RATE_LIMITER_H
 
#include <mutex>
#include <shared_mutex>
#include <memory>
 
#include "RequestManager.h"
#include "ClientDataManager.h"
#include "ResourceTunerSettings.h"
 
class RateLimiter {
private:
    static std::shared_ptr<RateLimiter> mRateLimiterInstance;
    static std::mutex instanceProtectionLock;
    std::shared_timed_mutex mRateLimiterMutex;
 
    uint32_t mDelta;
    double mPenaltyFactor;
    double mRewardFactor;
    int8_t shouldBeProcessed(int32_t clientPID);
 
    RateLimiter();
 
public:
    int8_t isRateLimitHonored(int32_t clientTID);
 
    int8_t isGlobalRateLimitHonored();
 
    static std::shared_ptr<RateLimiter> getInstance() {
        if(mRateLimiterInstance == nullptr) {
            instanceProtectionLock.lock();
            if(mRateLimiterInstance == nullptr) {
                try {
                    mRateLimiterInstance = std::shared_ptr<RateLimiter> (new RateLimiter());
                } catch(const std::bad_alloc& e) {
                    instanceProtectionLock.unlock();
                    return nullptr;
                }
            }
            instanceProtectionLock.unlock();
        }
        return mRateLimiterInstance;
    }
};
 
#endif