mirror of
https://github.com/scylladb/scylladb.git
synced 2026-04-20 00:20:47 +00:00
35 lines
675 B
C++
35 lines
675 B
C++
/*
|
|
* Copyright (C) 2015-present ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: LicenseRef-ScyllaDB-Source-Available-1.1
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <seastar/core/timer.hh>
|
|
#include <seastar/core/semaphore.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);
|
|
};
|
|
|
|
}
|