--- /dev/null
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+
+#include "include/stringify.h"
+#include "common/Timer.h"
+#include "common/WorkQueue.h"
+#include "common/debug.h"
+#include "common/errno.h"
+#include "librbd/Utils.h"
+#include "InstanceWatcher.h"
+#include "Instances.h"
+#include "Threads.h"
+
+#define dout_context g_ceph_context
+#define dout_subsys ceph_subsys_rbd_mirror
+#undef dout_prefix
+#define dout_prefix *_dout << "rbd::mirror::Instances: " \
+ << this << " " << __func__ << ": "
+
+namespace rbd {
+namespace mirror {
+
+using librbd::util::create_async_context_callback;
+using librbd::util::create_context_callback;
+using librbd::util::create_rados_callback;
+
+template <typename I>
+Instances<I>::Instances(Threads<I> *threads, librados::IoCtx &ioctx) :
+ m_threads(threads), m_ioctx(ioctx),
+ m_cct(reinterpret_cast<CephContext *>(ioctx.cct())),
+ m_lock("rbd::mirror::Instances " + ioctx.get_pool_name()) {
+}
+
+template <typename I>
+Instances<I>::~Instances() {
+}
+
+template <typename I>
+void Instances<I>::init(Context *on_finish) {
+ dout(20) << dendl;
+
+ Mutex::Locker locker(m_lock);
+ assert(m_on_finish == nullptr);
+ m_on_finish = on_finish;
+ get_instances();
+}
+
+template <typename I>
+void Instances<I>::shut_down(Context *on_finish) {
+ dout(20) << dendl;
+
+ Mutex::Locker locker(m_lock);
+ assert(m_on_finish == nullptr);
+ m_on_finish = on_finish;
+
+ Context *ctx = new FunctionContext(
+ [this](int r) {
+ Mutex::Locker timer_locker(m_threads->timer_lock);
+ Mutex::Locker locker(m_lock);
+
+ for (auto it : m_instances) {
+ cancel_remove_task(it.second);
+ }
+ wait_for_ops();
+ });
+
+ m_threads->work_queue->queue(ctx, 0);
+}
+
+template <typename I>
+void Instances<I>::notify(const std::string &instance_id) {
+ dout(20) << instance_id << dendl;
+
+ Mutex::Locker locker(m_lock);
+
+ if (m_on_finish != nullptr) {
+ dout(20) << "received on shut down, ignoring" << dendl;
+ return;
+ }
+
+ Context *ctx = new C_Notify(this, instance_id);
+
+ m_threads->work_queue->queue(ctx, 0);
+}
+
+template <typename I>
+void Instances<I>::handle_notify(const std::string &instance_id) {
+ dout(20) << instance_id << dendl;
+
+ Mutex::Locker timer_locker(m_threads->timer_lock);
+ Mutex::Locker locker(m_lock);
+
+ if (m_on_finish != nullptr) {
+ dout(20) << "handled on shut down, ignoring" << dendl;
+ return;
+ }
+
+ auto &instance = m_instances.insert(
+ std::make_pair(instance_id, Instance(instance_id))).first->second;
+
+ schedule_remove_task(instance);
+}
+
+template <typename I>
+void Instances<I>::list(std::vector<std::string> *instance_ids) {
+ dout(20) << dendl;
+
+ Mutex::Locker locker(m_lock);
+
+ for (auto it : m_instances) {
+ instance_ids->push_back(it.first);
+ }
+}
+
+
+template <typename I>
+void Instances<I>::get_instances() {
+ dout(20) << dendl;
+
+ assert(m_lock.is_locked());
+
+ Context *ctx = create_context_callback<
+ Instances, &Instances<I>::handle_get_instances>(this);
+
+ InstanceWatcher<I>::get_instances(m_ioctx, &m_instance_ids, ctx);
+}
+
+template <typename I>
+void Instances<I>::handle_get_instances(int r) {
+ dout(20) << "r=" << r << dendl;
+
+ Context *on_finish = nullptr;
+ {
+ Mutex::Locker timer_locker(m_threads->timer_lock);
+ Mutex::Locker locker(m_lock);
+
+ if (r < 0) {
+ derr << "error retrieving instances: " << cpp_strerror(r) << dendl;
+ } else {
+ auto my_instance_id = stringify(m_ioctx.get_instance_id());
+ for (auto &instance_id : m_instance_ids) {
+ if (instance_id == my_instance_id) {
+ continue;
+ }
+ auto &instance = m_instances.insert(
+ std::make_pair(instance_id, Instance(instance_id))).first->second;
+ schedule_remove_task(instance);
+ }
+ }
+ std::swap(on_finish, m_on_finish);
+ }
+ on_finish->complete(r);
+}
+
+template <typename I>
+void Instances<I>::wait_for_ops() {
+ dout(20) << dendl;
+
+ assert(m_lock.is_locked());
+
+ Context *ctx = create_async_context_callback(
+ m_threads->work_queue, create_context_callback<
+ Instances, &Instances<I>::handle_wait_for_ops>(this));
+
+ m_async_op_tracker.wait_for_ops(ctx);
+}
+
+template <typename I>
+void Instances<I>::handle_wait_for_ops(int r) {
+ dout(20) << "r=" << r << dendl;
+
+ assert(r == 0);
+
+ Context *on_finish = nullptr;
+ {
+ Mutex::Locker locker(m_lock);
+ std::swap(on_finish, m_on_finish);
+ }
+ on_finish->complete(r);
+}
+
+template <typename I>
+void Instances<I>::remove_instance(Instance &instance) {
+ assert(m_lock.is_locked());
+
+ dout(20) << instance.id << dendl;
+
+ Context *ctx = create_async_context_callback(
+ m_threads->work_queue, create_context_callback<
+ Instances, &Instances<I>::handle_remove_instance>(this));
+
+ m_async_op_tracker.start_op();
+ InstanceWatcher<I>::remove_instance(m_ioctx, m_threads->work_queue,
+ instance.id, ctx);
+ m_instances.erase(instance.id);
+}
+
+template <typename I>
+void Instances<I>::handle_remove_instance(int r) {
+ Mutex::Locker locker(m_lock);
+
+ dout(20) << " r=" << r << dendl;
+
+ assert(r == 0);
+
+ m_async_op_tracker.finish_op();
+}
+
+template <typename I>
+void Instances<I>::cancel_remove_task(Instance &instance) {
+ assert(m_threads->timer_lock.is_locked());
+ assert(m_lock.is_locked());
+
+ if (instance.timer_task == nullptr) {
+ return;
+ }
+
+ dout(20) << instance.timer_task << dendl;
+
+ bool canceled = m_threads->timer->cancel_event(instance.timer_task);
+ assert(canceled);
+ instance.timer_task = nullptr;
+}
+
+template <typename I>
+void Instances<I>::schedule_remove_task(Instance &instance) {
+ dout(20) << dendl;
+
+ cancel_remove_task(instance);
+
+ int after = m_cct->_conf->get_val<int64_t>("rbd_mirror_leader_heartbeat_interval") *
+ (1 + m_cct->_conf->get_val<int64_t>("rbd_mirror_leader_max_missed_heartbeats") +
+ m_cct->_conf->get_val<int64_t>("rbd_mirror_leader_max_acquire_attempts_before_break"));
+
+ instance.timer_task = new FunctionContext(
+ [this, &instance](int r) {
+ assert(m_threads->timer_lock.is_locked());
+ Mutex::Locker locker(m_lock);
+ instance.timer_task = nullptr;
+ remove_instance(instance);
+ });
+
+ dout(20) << "scheduling instance " << instance.id << " remove after " << after
+ << " sec (task " << instance.timer_task << ")" << dendl;
+
+ m_threads->timer->add_event_after(after, instance.timer_task);
+}
+
+} // namespace mirror
+} // namespace rbd
+
+template class rbd::mirror::Instances<librbd::ImageCtx>;