summaryrefslogtreecommitdiffstats
path: root/src/tools/rbd_mirror/image_replayer/PrepareLocalImageRequest.h
blob: 6372169ff05507ade5c7e23391e76143dc47454c (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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
// vim: ts=8 sw=2 smarttab

#ifndef RBD_MIRROR_IMAGE_REPLAYER_PREPARE_LOCAL_IMAGE_REQUEST_H
#define RBD_MIRROR_IMAGE_REPLAYER_PREPARE_LOCAL_IMAGE_REQUEST_H

#include "include/buffer.h"
#include "include/rados/librados_fwd.hpp"
#include "cls/rbd/cls_rbd_types.h"
#include "librbd/mirror/Types.h"
#include <string>

struct Context;

namespace librbd {
struct ImageCtx;
namespace asio { struct ContextWQ; }
} // namespace librbd

namespace rbd {
namespace mirror {
namespace image_replayer {

template <typename> class StateBuilder;

template <typename ImageCtxT = librbd::ImageCtx>
class PrepareLocalImageRequest {
public:
  static PrepareLocalImageRequest *create(
      librados::IoCtx &io_ctx,
      const std::string &global_image_id,
      std::string *local_image_name,
      StateBuilder<ImageCtxT>** state_builder,
      librbd::asio::ContextWQ *work_queue,
      Context *on_finish) {
    return new PrepareLocalImageRequest(io_ctx, global_image_id,
                                        local_image_name, state_builder,
                                        work_queue, on_finish);
  }

  PrepareLocalImageRequest(
      librados::IoCtx &io_ctx,
      const std::string &global_image_id,
      std::string *local_image_name,
      StateBuilder<ImageCtxT>** state_builder,
      librbd::asio::ContextWQ *work_queue,
      Context *on_finish)
    : m_io_ctx(io_ctx), m_global_image_id(global_image_id),
      m_local_image_name(local_image_name), m_state_builder(state_builder),
      m_work_queue(work_queue), m_on_finish(on_finish) {
  }

  void send();

private:
  /**
   * @verbatim
   *
   * <start>
   *    |
   *    v
   * GET_LOCAL_IMAGE_ID
   *    |
   *    v
   * GET_LOCAL_IMAGE_NAME
   *    |
   *    v
   * GET_MIRROR_INFO
   *    |
   *    | (if the image mirror state is CREATING)
   *    v
   * TRASH_MOVE
   *    |
   *    v
   * <finish>
   *
   * @endverbatim
   */

  librados::IoCtx &m_io_ctx;
  std::string m_global_image_id;
  std::string *m_local_image_name;
  StateBuilder<ImageCtxT>** m_state_builder;
  librbd::asio::ContextWQ *m_work_queue;
  Context *m_on_finish;

  bufferlist m_out_bl;
  std::string m_local_image_id;
  cls::rbd::MirrorImage m_mirror_image;
  librbd::mirror::PromotionState m_promotion_state;
  std::string m_primary_mirror_uuid;

  void get_local_image_id();
  void handle_get_local_image_id(int r);

  void get_local_image_name();
  void handle_get_local_image_name(int r);

  void get_mirror_info();
  void handle_get_mirror_info(int r);

  void move_to_trash();
  void handle_move_to_trash(int r);

  void finish(int r);

};

} // namespace image_replayer
} // namespace mirror
} // namespace rbd

extern template class rbd::mirror::image_replayer::PrepareLocalImageRequest<librbd::ImageCtx>;

#endif // RBD_MIRROR_IMAGE_REPLAYER_PREPARE_LOCAL_IMAGE_REQUEST_H