HttpRateLimitHandle.hpp

Go to the documentation of this file.

#pragma once
#ifndef _KURLYK_HTTP_RATE_LIMIT_HANDLE_HPP_INCLUDED
#define _KURLYK_HTTP_RATE_LIMIT_HANDLE_HPP_INCLUDED

 
namespace kurlyk {

    class HttpRateLimitHandle {
    public:
        HttpRateLimitHandle(const HttpRateLimitHandle&) = delete;
        HttpRateLimitHandle& operator=(const HttpRateLimitHandle&) = delete;

        ~HttpRateLimitHandle() {
            reset();
        }

        long id() const {
            return m_id;
        }

        explicit operator bool() const {
            return m_id != 0;
        }
 
    private:
        friend class HttpRateLimiter;
 
        using remove_fn_t = std::function<void(long)>;

        HttpRateLimitHandle(long id, remove_fn_t remove_fn)
            : m_id(id),
              m_remove_fn(std::move(remove_fn)) {}

        void reset() {
            const long id = m_id;
            m_id = 0;
 
            if (id != 0 && m_remove_fn) {
                m_remove_fn(id);
            }
        }
 
    private:
        long        m_id = 0;    
        remove_fn_t m_remove_fn; 
    };

    using HttpRateLimitHandlePtr = std::shared_ptr<HttpRateLimitHandle>;
 
} // namespace kurlyk
 
#endif // _KURLYK_HTTP_RATE_LIMIT_HANDLE_HPP_INCLUDED