mirror of
https://github.com/scylladb/scylladb.git
synced 2026-06-05 22:43:15 +00:00
f3eade2f62
Drop the AGPL license in favor of a source-available license. See the blog post [1] for details. [1] https://www.scylladb.com/2024/12/18/why-were-moving-to-a-source-available-license/
36 lines
710 B
C++
36 lines
710 B
C++
/*
|
|
* Copyright (C) 2015-present ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: LicenseRef-ScyllaDB-Source-Available-1.0
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <seastar/core/timer.hh>
|
|
#include <seastar/core/semaphore.hh>
|
|
#include <seastar/core/seastar.hh>
|
|
#include "seastarx.hh"
|
|
|
|
namespace utils {
|
|
|
|
/**
|
|
* 100% naive rate limiter. Consider it a placeholder
|
|
* Will let you process X "units" per second, then reset this every s.
|
|
* Obviously, accuracy is virtually non-existent and steady rate will fluctuate.
|
|
*/
|
|
class rate_limiter {
|
|
private:
|
|
timer<lowres_clock> _timer;
|
|
size_t _units_per_s;
|
|
semaphore _sem {0};
|
|
|
|
void on_timer();
|
|
public:
|
|
rate_limiter(size_t rate);
|
|
future<> reserve(size_t u);
|
|
};
|
|
|
|
}
|