initial code repo

[stor4nfv.git] / src / ceph / src / tools / rbd_mirror / ImageSyncThrottler.h

diff --git a/src/ceph/src/tools/rbd_mirror/ImageSyncThrottler.h b/src/ceph/src/tools/rbd_mirror/ImageSyncThrottler.h
new file mode 100644 (file)
index 0000000..e0c3f0b
--- /dev/null
+++ b/src/ceph/src/tools/rbd_mirror/ImageSyncThrottler.h
@@ -0,0 +1,61 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+
+#ifndef RBD_MIRROR_IMAGE_SYNC_THROTTLER_H
+#define RBD_MIRROR_IMAGE_SYNC_THROTTLER_H
+
+#include <list>
+#include <set>
+#include <sstream>
+#include <string>
+#include <utility>
+
+#include "common/Mutex.h"
+#include "common/config_obs.h"
+
+class Context;
+
+namespace ceph { class Formatter; }
+namespace librbd { class ImageCtx; }
+
+namespace rbd {
+namespace mirror {
+
+template <typename ImageCtxT = librbd::ImageCtx>
+class ImageSyncThrottler : public md_config_obs_t {
+public:
+  static ImageSyncThrottler *create() {
+    return new ImageSyncThrottler();
+  }
+  void destroy() {
+    delete this;
+  }
+
+  ImageSyncThrottler();
+  ~ImageSyncThrottler() override;
+
+  void set_max_concurrent_syncs(uint32_t max);
+  void start_op(const std::string &id, Context *on_start);
+  bool cancel_op(const std::string &id);
+  void finish_op(const std::string &id);
+  void drain(int r);
+
+  void print_status(Formatter *f, std::stringstream *ss);
+
+private:
+  Mutex m_lock;
+  uint32_t m_max_concurrent_syncs;
+  std::list<std::pair<std::string, Context *>> m_queue;
+  std::set<std::string> m_inflight_ops;
+
+  const char **get_tracked_conf_keys() const override;
+  void handle_conf_change(const struct md_config_t *conf,
+                          const std::set<std::string> &changed) override;
+};
+
+} // namespace mirror
+} // namespace rbd
+
+extern template class rbd::mirror::ImageSyncThrottler<librbd::ImageCtx>;
+
+#endif // RBD_MIRROR_IMAGE_SYNC_THROTTLER_H