blob: 32080238aea22a45459750c97ed9ef9c9f415b83 (
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
// vim: ts=8 sw=2 smarttab
#ifndef RBD_MIRROR_THROTTLER_H
#define RBD_MIRROR_THROTTLER_H
#include <list>
#include <map>
#include <set>
#include <sstream>
#include <string>
#include <utility>
#include "common/ceph_mutex.h"
#include "common/config_obs.h"
#include "include/common_fwd.h"
class Context;
namespace ceph { class Formatter; }
namespace librbd { class ImageCtx; }
namespace rbd {
namespace mirror {
template <typename ImageCtxT = librbd::ImageCtx>
class Throttler : public md_config_obs_t {
public:
static Throttler *create(
CephContext *cct,
const std::string &config_key) {
return new Throttler(cct, config_key);
}
void destroy() {
delete this;
}
Throttler(CephContext *cct,
const std::string &config_key);
~Throttler() override;
void set_max_concurrent_ops(uint32_t max);
void start_op(const std::string &ns, const std::string &id,
Context *on_start);
bool cancel_op(const std::string &ns, const std::string &id);
void finish_op(const std::string &ns, const std::string &id);
void drain(const std::string &ns, int r);
void print_status(ceph::Formatter *f);
private:
typedef std::pair<std::string, std::string> Id;
CephContext *m_cct;
const std::string m_config_key;
mutable const char* m_config_keys[2];
ceph::mutex m_lock;
uint32_t m_max_concurrent_ops;
std::list<Id> m_queue;
std::map<Id, Context *> m_queued_ops;
std::set<Id> m_inflight_ops;
const char **get_tracked_conf_keys() const override;
void handle_conf_change(const ConfigProxy& conf,
const std::set<std::string> &changed) override;
};
} // namespace mirror
} // namespace rbd
extern template class rbd::mirror::Throttler<librbd::ImageCtx>;
#endif // RBD_MIRROR_THROTTLER_H
|
