summaryrefslogtreecommitdiffstats
path: root/src/librbd/PluginRegistry.cc
blob: 6ddf0a414f8a932aa365d1e419b24bffe4131a48 (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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
// vim: ts=8 sw=2 smarttab

#include "librbd/PluginRegistry.h"
#include "include/Context.h"
#include "common/dout.h"
#include "librbd/cache/ImageWriteback.h"
#include "librbd/ImageCtx.h"
#include "librbd/plugin/Api.h"
#include <boost/tokenizer.hpp>

#define dout_subsys ceph_subsys_rbd
#undef dout_prefix
#define dout_prefix *_dout << "librbd::PluginRegistry: " \
                           << this << " " << __func__ << ": "

namespace librbd {

template <typename I>
PluginRegistry<I>::PluginRegistry(I* image_ctx)
  : m_image_ctx(image_ctx), m_plugin_api(std::make_unique<plugin::Api<I>>()),
    m_image_writeback(std::make_unique<cache::ImageWriteback<I>>(*image_ctx)) {
}

template <typename I>
PluginRegistry<I>::~PluginRegistry() {
}

template <typename I>
void PluginRegistry<I>::init(const std::string& plugins, Context* on_finish) {
  auto cct = m_image_ctx->cct;
  auto plugin_registry = cct->get_plugin_registry();

  auto gather_ctx = new C_Gather(cct, on_finish);

  boost::tokenizer<boost::escaped_list_separator<char>> tokenizer(plugins);
  for (auto token : tokenizer) {
    ldout(cct, 5) << "attempting to load plugin: " << token << dendl;

    auto ctx = gather_ctx->new_sub();

    auto plugin = dynamic_cast<plugin::Interface<I>*>(
      plugin_registry->get_with_load("librbd", "librbd_" + token));
    if (plugin == nullptr) {
      lderr(cct) << "failed to load plugin: " << token << dendl;
      ctx->complete(-ENOSYS);
      break;
    }

    plugin->init(
	m_image_ctx, *m_plugin_api, *m_image_writeback, m_plugin_hook_points, ctx);
  }

  gather_ctx->activate();
}

template <typename I>
void PluginRegistry<I>::acquired_exclusive_lock(Context* on_finish) {
  auto cct = m_image_ctx->cct;
  ldout(cct, 20) << dendl;

  auto gather_ctx = new C_Gather(cct, on_finish);

  for (auto &hook : m_plugin_hook_points) {
    auto ctx = gather_ctx->new_sub();
    hook->acquired_exclusive_lock(ctx);
  }
  gather_ctx->activate();
}

template <typename I>
void PluginRegistry<I>::prerelease_exclusive_lock(Context* on_finish) {
  auto cct = m_image_ctx->cct;
  ldout(cct, 20) << dendl;

  auto gather_ctx = new C_Gather(cct, on_finish);

  for (auto &hook : m_plugin_hook_points) {
    auto ctx = gather_ctx->new_sub();
    hook->prerelease_exclusive_lock(ctx);
  }
  gather_ctx->activate();
}

template <typename I>
void PluginRegistry<I>::discard(Context* on_finish) {
  auto cct = m_image_ctx->cct;
  ldout(cct, 20) << dendl;

  auto gather_ctx = new C_Gather(cct, on_finish);

  for (auto &hook : m_plugin_hook_points) {
    auto ctx = gather_ctx->new_sub();
    hook->discard(ctx);
  }
  gather_ctx->activate();
}

} // namespace librbd

template class librbd::PluginRegistry<librbd::ImageCtx>;