summaryrefslogtreecommitdiffstats
path: root/src/librbd/AsioEngine.h
blob: 0f476d80b2a47f67dd54c8ef4516ee6430ade828 (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
// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
// vim: ts=8 sw=2 smarttab

#ifndef CEPH_LIBRBD_ASIO_ENGINE_H
#define CEPH_LIBRBD_ASIO_ENGINE_H

#include "include/common_fwd.h"
#include "include/rados/librados_fwd.hpp"
#include <memory>
#include <boost/asio/dispatch.hpp>
#include <boost/asio/io_context.hpp>
#include <boost/asio/io_context_strand.hpp>
#include <boost/asio/post.hpp>

struct Context;
namespace neorados { struct RADOS; }

namespace librbd {

namespace asio { struct ContextWQ; }

class AsioEngine {
public:
  explicit AsioEngine(std::shared_ptr<librados::Rados> rados);
  explicit AsioEngine(librados::IoCtx& io_ctx);
  ~AsioEngine();

  AsioEngine(AsioEngine&&) = delete;
  AsioEngine(const AsioEngine&) = delete;
  AsioEngine& operator=(const AsioEngine&) = delete;

  inline neorados::RADOS& get_rados_api() {
    return *m_rados_api;
  }

  inline boost::asio::io_context& get_io_context() {
    return m_io_context;
  }
  inline operator boost::asio::io_context&() {
    return m_io_context;
  }

  using executor_type = boost::asio::io_context::executor_type;
  inline executor_type get_executor() {
    return m_io_context.get_executor();
  }

  inline boost::asio::io_context::strand& get_api_strand() {
    // API client callbacks should never fire concurrently
    return *m_api_strand;
  }

  inline asio::ContextWQ* get_work_queue() {
    return m_context_wq.get();
  }

  template <typename T>
  void dispatch(T&& t) {
    boost::asio::dispatch(m_io_context, std::forward<T>(t));
  }
  void dispatch(Context* ctx, int r);

  template <typename T>
  void post(T&& t) {
    boost::asio::post(m_io_context, std::forward<T>(t));
  }
  void post(Context* ctx, int r);

private:
  std::shared_ptr<neorados::RADOS> m_rados_api;
  CephContext* m_cct;

  boost::asio::io_context& m_io_context;
  std::unique_ptr<boost::asio::io_context::strand> m_api_strand;
  std::unique_ptr<asio::ContextWQ> m_context_wq;
};

} // namespace librbd

#endif // CEPH_LIBRBD_ASIO_ENGINE_H