blob: ae422e1dada275b3609dfc9e981b808c58149407 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
#include "rgw_ratelimit.h"
#include "rgw_common.h"
#include "random"
#include <cstdlib>
#include <string>
#include <chrono>
#include <boost/program_options.hpp>
int main(int argc, char **argv)
{
int num_qos_classes = 1;
try
{
using namespace boost::program_options;
options_description desc{"Options"};
desc.add_options()
("help,h", "Help screen")
("num_qos_classes", value<int>()->default_value(1), "how many qos tenants");
variables_map vm;
store(parse_command_line(argc, argv, desc), vm);
if (vm.count("help")) {
std::cout << desc << std::endl;
return EXIT_SUCCESS;
}
num_qos_classes = vm["num_qos_classes"].as<int>();
}
catch (const boost::program_options::error &ex)
{
std::cerr << ex.what() << std::endl;
return EXIT_FAILURE;
}
RGWRateLimitInfo info;
info.enabled = true;
info.max_read_bytes = 0;
info.max_write_bytes = 0;
info.max_read_ops = 0;
info.max_write_ops = 0;
std::unique_ptr<CephContext> cct = std::make_unique<CephContext>(CEPH_ENTITY_TYPE_ANY);
if (!g_ceph_context)
{
g_ceph_context = cct.get();
}
std::shared_ptr<ActiveRateLimiter> ratelimit(new ActiveRateLimiter(g_ceph_context));
ratelimit->start();
auto dout = DoutPrefix(g_ceph_context, ceph_subsys_rgw, "rate limiter: ");
for(int i = 0; i < num_qos_classes; i++)
{
std::string tenant = "uuser" + std::to_string(i);
auto time = ceph::coarse_real_clock::now();
ratelimit->get_active()->should_rate_limit("PUT", tenant, time, &info);
}
}
|
