]>
Commit | Line | Data |
---|---|---|
1 | // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- | |
2 | // vim: ts=8 sw=2 smarttab | |
3 | ||
4 | #include "librbd/object_map/UpdateRequest.h" | |
5 | #include "include/rbd/object_map_types.h" | |
6 | #include "include/stringify.h" | |
7 | #include "common/dout.h" | |
8 | #include "librbd/ImageCtx.h" | |
9 | #include "librbd/ObjectMap.h" | |
10 | #include "librbd/Utils.h" | |
11 | #include "cls/lock/cls_lock_client.h" | |
12 | #include <string> | |
13 | ||
14 | #define dout_subsys ceph_subsys_rbd | |
15 | #undef dout_prefix | |
16 | #define dout_prefix *_dout << "librbd::object_map::UpdateRequest: " << this \ | |
17 | << " " << __func__ << ": " | |
18 | ||
19 | namespace librbd { | |
20 | namespace object_map { | |
21 | ||
22 | namespace { | |
23 | ||
24 | // keep aligned to bit_vector 4K block sizes | |
25 | const uint64_t MAX_OBJECTS_PER_UPDATE = 256 * (1 << 10); | |
26 | ||
27 | } | |
28 | ||
29 | template <typename I> | |
30 | void UpdateRequest<I>::send() { | |
31 | update_object_map(); | |
32 | } | |
33 | ||
34 | template <typename I> | |
35 | void UpdateRequest<I>::update_object_map() { | |
36 | ceph_assert(ceph_mutex_is_locked(m_image_ctx.image_lock)); | |
37 | ceph_assert(ceph_mutex_is_locked(*m_object_map_lock)); | |
38 | CephContext *cct = m_image_ctx.cct; | |
39 | ||
40 | // break very large requests into manageable batches | |
41 | m_update_end_object_no = std::min( | |
42 | m_end_object_no, m_update_start_object_no + MAX_OBJECTS_PER_UPDATE); | |
43 | ||
44 | std::string oid(ObjectMap<>::object_map_name(m_image_ctx.id, m_snap_id)); | |
45 | ldout(cct, 20) << "ictx=" << &m_image_ctx << ", oid=" << oid << ", " | |
46 | << "[" << m_update_start_object_no << "," | |
47 | << m_update_end_object_no << ") = " | |
48 | << (m_current_state ? | |
49 | stringify(static_cast<uint32_t>(*m_current_state)) : "") | |
50 | << "->" << static_cast<uint32_t>(m_new_state) | |
51 | << dendl; | |
52 | ||
53 | librados::ObjectWriteOperation op; | |
54 | if (m_snap_id == CEPH_NOSNAP) { | |
55 | rados::cls::lock::assert_locked(&op, RBD_LOCK_NAME, ClsLockType::EXCLUSIVE, "", ""); | |
56 | } | |
57 | cls_client::object_map_update(&op, m_update_start_object_no, | |
58 | m_update_end_object_no, m_new_state, | |
59 | m_current_state); | |
60 | ||
61 | auto rados_completion = librbd::util::create_rados_callback< | |
62 | UpdateRequest<I>, &UpdateRequest<I>::handle_update_object_map>(this); | |
63 | std::vector<librados::snap_t> snaps; | |
64 | int r = m_image_ctx.md_ctx.aio_operate( | |
65 | oid, rados_completion, &op, 0, snaps, | |
66 | (m_trace.valid() ? m_trace.get_info() : nullptr)); | |
67 | ceph_assert(r == 0); | |
68 | rados_completion->release(); | |
69 | } | |
70 | ||
71 | template <typename I> | |
72 | void UpdateRequest<I>::handle_update_object_map(int r) { | |
73 | ldout(m_image_ctx.cct, 20) << "r=" << r << dendl; | |
74 | ||
75 | if (r == -ENOENT && m_ignore_enoent) { | |
76 | r = 0; | |
77 | } | |
78 | if (r < 0 && m_ret_val == 0) { | |
79 | m_ret_val = r; | |
80 | } | |
81 | ||
82 | { | |
83 | std::shared_lock image_locker{m_image_ctx.image_lock}; | |
84 | std::unique_lock object_map_locker{*m_object_map_lock}; | |
85 | update_in_memory_object_map(); | |
86 | ||
87 | if (m_update_end_object_no < m_end_object_no) { | |
88 | m_update_start_object_no = m_update_end_object_no; | |
89 | update_object_map(); | |
90 | return; | |
91 | } | |
92 | } | |
93 | ||
94 | // no more batch updates to send | |
95 | complete(m_ret_val); | |
96 | } | |
97 | ||
98 | template <typename I> | |
99 | void UpdateRequest<I>::update_in_memory_object_map() { | |
100 | ceph_assert(ceph_mutex_is_locked(m_image_ctx.image_lock)); | |
101 | ceph_assert(ceph_mutex_is_locked(*m_object_map_lock)); | |
102 | ||
103 | // rebuilding the object map might update on-disk only | |
104 | if (m_snap_id == m_image_ctx.snap_id) { | |
105 | ldout(m_image_ctx.cct, 20) << dendl; | |
106 | ||
107 | auto it = m_object_map.begin() + | |
108 | std::min(m_update_start_object_no, m_object_map.size()); | |
109 | auto end_it = m_object_map.begin() + | |
110 | std::min(m_update_end_object_no, m_object_map.size()); | |
111 | for (; it != end_it; ++it) { | |
112 | auto state_ref = *it; | |
113 | uint8_t state = state_ref; | |
114 | if (!m_current_state || state == *m_current_state || | |
115 | (*m_current_state == OBJECT_EXISTS && state == OBJECT_EXISTS_CLEAN)) { | |
116 | state_ref = m_new_state; | |
117 | } | |
118 | } | |
119 | } | |
120 | } | |
121 | ||
122 | template <typename I> | |
123 | void UpdateRequest<I>::finish_request() { | |
124 | } | |
125 | ||
126 | } // namespace object_map | |
127 | } // namespace librbd | |
128 | ||
129 | template class librbd::object_map::UpdateRequest<librbd::ImageCtx>; |