+++ /dev/null
-// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
-// vim: ts=8 sw=2 smarttab
-
-#ifndef CEPH_RBD_MIRROR_POOL_WATCHER_H
-#define CEPH_RBD_MIRROR_POOL_WATCHER_H
-
-#include <map>
-#include <memory>
-#include <set>
-#include <string>
-
-#include "common/AsyncOpTracker.h"
-#include "common/ceph_context.h"
-#include "common/Mutex.h"
-#include "include/rados/librados.hpp"
-#include "types.h"
-#include <boost/functional/hash.hpp>
-#include <boost/optional.hpp>
-#include "include/assert.h"
-
-namespace librbd { struct ImageCtx; }
-
-namespace rbd {
-namespace mirror {
-
-template <typename> struct Threads;
-
-/**
- * Keeps track of images that have mirroring enabled within all
- * pools.
- */
-template <typename ImageCtxT = librbd::ImageCtx>
-class PoolWatcher {
-public:
- struct Listener {
- virtual ~Listener() {
- }
-
- virtual void handle_update(const std::string &mirror_uuid,
- ImageIds &&added_image_ids,
- ImageIds &&removed_image_ids) = 0;
- };
-
- PoolWatcher(Threads<ImageCtxT> *threads, librados::IoCtx &remote_io_ctx,
- Listener &listener);
- ~PoolWatcher();
- PoolWatcher(const PoolWatcher&) = delete;
- PoolWatcher& operator=(const PoolWatcher&) = delete;
-
- bool is_blacklisted() const;
-
- void init(Context *on_finish = nullptr);
- void shut_down(Context *on_finish);
-
- inline uint64_t get_image_count() const {
- Mutex::Locker locker(m_lock);
- return m_image_ids.size();
- }
-
-private:
- /**
- * @verbatim
- *
- * <start>
- * |
- * v
- * INIT
- * |
- * v
- * REGISTER_WATCHER
- * |
- * |/--------------------------------\
- * | |
- * v |
- * REFRESH_IMAGES |
- * | |
- * |/----------------------------\ |
- * | | |
- * v | |
- * GET_MIRROR_UUID | |
- * | | |
- * v | |
- * NOTIFY_LISTENER | |
- * | | |
- * v | |
- * IDLE ---\ | |
- * | | | |
- * | |\---> IMAGE_UPDATED | |
- * | | | | |
- * | | v | |
- * | | GET_IMAGE_NAME --/ |
- * | | |
- * | \----> WATCH_ERROR ---------/
- * v
- * SHUT_DOWN
- * |
- * v
- * UNREGISTER_WATCHER
- * |
- * v
- * <finish>
- *
- * @endverbatim
- */
- class MirroringWatcher;
-
- Threads<ImageCtxT> *m_threads;
- librados::IoCtx m_remote_io_ctx;
- Listener &m_listener;
-
- ImageIds m_refresh_image_ids;
- bufferlist m_out_bl;
-
- mutable Mutex m_lock;
-
- Context *m_on_init_finish = nullptr;
-
- ImageIds m_image_ids;
- std::string m_mirror_uuid;
-
- bool m_pending_updates = false;
- bool m_notify_listener_in_progress = false;
- ImageIds m_pending_image_ids;
- ImageIds m_pending_added_image_ids;
- ImageIds m_pending_removed_image_ids;
-
- std::string m_pending_mirror_uuid;
-
- MirroringWatcher *m_mirroring_watcher;
-
- Context *m_timer_ctx = nullptr;
-
- AsyncOpTracker m_async_op_tracker;
- bool m_blacklisted = false;
- bool m_shutting_down = false;
- bool m_image_ids_invalid = true;
- bool m_refresh_in_progress = false;
- bool m_deferred_refresh = false;
-
- void register_watcher();
- void handle_register_watcher(int r);
- void unregister_watcher();
-
- void refresh_images();
- void handle_refresh_images(int r);
-
- void schedule_refresh_images(double interval);
- void process_refresh_images();
-
- void get_mirror_uuid();
- void handle_get_mirror_uuid(int r);
-
- void handle_rewatch_complete(int r);
- void handle_image_updated(const std::string &remote_image_id,
- const std::string &global_image_id,
- bool enabled);
-
- void schedule_listener();
- void notify_listener();
-
-};
-
-} // namespace mirror
-} // namespace rbd
-
-extern template class rbd::mirror::PoolWatcher<librbd::ImageCtx>;
-
-#endif // CEPH_RBD_MIRROR_POOL_WATCHER_H