couchbase 3.4.3 → 3.4.4

Sign up to get free protection for your applications and to get access to all the features.
Files changed (148) hide show
  1. checksums.yaml +4 -4
  2. data/README.md +2 -2
  3. data/ext/couchbase/CMakeLists.txt +15 -1
  4. data/ext/couchbase/core/bucket.cxx +183 -152
  5. data/ext/couchbase/core/bucket.hxx +17 -4
  6. data/ext/couchbase/core/cluster.hxx +34 -13
  7. data/ext/couchbase/core/cluster_options.hxx +3 -0
  8. data/ext/couchbase/core/crud_component.cxx +51 -22
  9. data/ext/couchbase/core/error_context/key_value.cxx +2 -1
  10. data/ext/couchbase/core/error_context/key_value.hxx +10 -12
  11. data/ext/couchbase/core/impl/build_deferred_query_indexes.cxx +115 -50
  12. data/ext/couchbase/core/impl/cluster.cxx +6 -0
  13. data/ext/couchbase/core/impl/create_bucket.cxx +155 -0
  14. data/ext/couchbase/core/impl/create_query_index.cxx +172 -59
  15. data/ext/couchbase/core/impl/dns_srv_tracker.cxx +2 -1
  16. data/ext/couchbase/core/impl/drop_bucket.cxx +66 -0
  17. data/ext/couchbase/core/impl/drop_query_index.cxx +138 -59
  18. data/ext/couchbase/core/impl/flush_bucket.cxx +66 -0
  19. data/ext/couchbase/core/impl/get_all_buckets.cxx +163 -0
  20. data/ext/couchbase/core/impl/get_all_query_indexes.cxx +67 -37
  21. data/ext/couchbase/core/impl/get_bucket.cxx +153 -0
  22. data/ext/couchbase/core/impl/internal_manager_error_context.cxx +113 -0
  23. data/ext/couchbase/core/impl/internal_manager_error_context.hxx +60 -0
  24. data/ext/couchbase/core/impl/key_value_error_category.cxx +2 -4
  25. data/ext/couchbase/core/impl/key_value_error_context.cxx +98 -0
  26. data/ext/couchbase/core/impl/lookup_in.cxx +1 -0
  27. data/ext/couchbase/core/impl/lookup_in_all_replicas.cxx +176 -0
  28. data/ext/couchbase/core/impl/lookup_in_all_replicas.hxx +80 -0
  29. data/ext/couchbase/core/impl/lookup_in_any_replica.cxx +167 -0
  30. data/ext/couchbase/core/impl/lookup_in_any_replica.hxx +75 -0
  31. data/ext/couchbase/core/impl/lookup_in_replica.cxx +97 -0
  32. data/ext/couchbase/core/impl/lookup_in_replica.hxx +67 -0
  33. data/ext/couchbase/core/impl/manager_error_context.cxx +100 -0
  34. data/ext/couchbase/core/impl/query.cxx +1 -0
  35. data/ext/couchbase/core/impl/query_error_context.cxx +75 -0
  36. data/ext/couchbase/core/impl/update_bucket.cxx +130 -0
  37. data/ext/couchbase/core/impl/watch_query_indexes.cxx +53 -29
  38. data/ext/couchbase/core/io/dns_client.cxx +111 -40
  39. data/ext/couchbase/core/io/dns_config.cxx +5 -4
  40. data/ext/couchbase/core/io/http_session.hxx +24 -1
  41. data/ext/couchbase/core/io/mcbp_command.hxx +9 -2
  42. data/ext/couchbase/core/io/mcbp_session.cxx +80 -43
  43. data/ext/couchbase/core/io/mcbp_session.hxx +4 -3
  44. data/ext/couchbase/core/logger/custom_rotating_file_sink.cxx +1 -1
  45. data/ext/couchbase/core/logger/logger.cxx +80 -20
  46. data/ext/couchbase/core/logger/logger.hxx +31 -0
  47. data/ext/couchbase/core/meta/features.hxx +25 -0
  48. data/ext/couchbase/core/operations/document_lookup_in_all_replicas.hxx +192 -0
  49. data/ext/couchbase/core/operations/document_lookup_in_any_replica.hxx +188 -0
  50. data/ext/couchbase/core/operations/document_query.cxx +11 -0
  51. data/ext/couchbase/core/operations/document_query.hxx +1 -0
  52. data/ext/couchbase/core/operations.hxx +2 -0
  53. data/ext/couchbase/core/origin.cxx +270 -0
  54. data/ext/couchbase/core/origin.hxx +2 -0
  55. data/ext/couchbase/core/protocol/client_response.hxx +1 -0
  56. data/ext/couchbase/core/protocol/cmd_hello.hxx +1 -0
  57. data/ext/couchbase/core/protocol/cmd_lookup_in_replica.cxx +107 -0
  58. data/ext/couchbase/core/protocol/cmd_lookup_in_replica.hxx +137 -0
  59. data/ext/couchbase/core/protocol/hello_feature.hxx +6 -0
  60. data/ext/couchbase/core/protocol/hello_feature_fmt.hxx +3 -0
  61. data/ext/couchbase/core/protocol/status.cxx +2 -2
  62. data/ext/couchbase/core/range_scan_options.cxx +3 -27
  63. data/ext/couchbase/core/range_scan_options.hxx +13 -17
  64. data/ext/couchbase/core/range_scan_orchestrator.cxx +388 -170
  65. data/ext/couchbase/core/range_scan_orchestrator.hxx +13 -2
  66. data/ext/couchbase/core/range_scan_orchestrator_options.hxx +5 -3
  67. data/ext/couchbase/core/scan_options.hxx +0 -19
  68. data/ext/couchbase/core/scan_result.cxx +19 -5
  69. data/ext/couchbase/core/scan_result.hxx +5 -2
  70. data/ext/couchbase/core/timeout_defaults.hxx +2 -3
  71. data/ext/couchbase/core/topology/capabilities.hxx +3 -0
  72. data/ext/couchbase/core/topology/capabilities_fmt.hxx +8 -0
  73. data/ext/couchbase/core/topology/collections_manifest_fmt.hxx +1 -1
  74. data/ext/couchbase/core/topology/configuration.hxx +15 -0
  75. data/ext/couchbase/core/topology/configuration_json.hxx +6 -1
  76. data/ext/couchbase/core/utils/connection_string.cxx +62 -47
  77. data/ext/couchbase/core/utils/connection_string.hxx +1 -0
  78. data/ext/couchbase/couchbase/analytics_error_context.hxx +1 -1
  79. data/ext/couchbase/couchbase/behavior_options.hxx +19 -2
  80. data/ext/couchbase/couchbase/bucket_manager.hxx +135 -0
  81. data/ext/couchbase/couchbase/build_query_index_options.hxx +0 -30
  82. data/ext/couchbase/couchbase/cluster.hxx +14 -0
  83. data/ext/couchbase/couchbase/collection.hxx +111 -0
  84. data/ext/couchbase/couchbase/collection_query_index_manager.hxx +7 -48
  85. data/ext/couchbase/couchbase/create_bucket_options.hxx +41 -0
  86. data/ext/couchbase/couchbase/create_primary_query_index_options.hxx +0 -29
  87. data/ext/couchbase/couchbase/create_query_index_options.hxx +0 -33
  88. data/ext/couchbase/couchbase/drop_bucket_options.hxx +41 -0
  89. data/ext/couchbase/couchbase/drop_primary_query_index_options.hxx +0 -30
  90. data/ext/couchbase/couchbase/drop_query_index_options.hxx +0 -31
  91. data/ext/couchbase/couchbase/error_codes.hxx +1 -2
  92. data/ext/couchbase/couchbase/error_context.hxx +10 -2
  93. data/ext/couchbase/couchbase/flush_bucket_options.hxx +41 -0
  94. data/ext/couchbase/{core/topology/error_map_fmt.hxx → couchbase/fmt/key_value_error_map_attribute.hxx} +21 -21
  95. data/ext/couchbase/couchbase/get_all_buckets_options.hxx +44 -0
  96. data/ext/couchbase/couchbase/get_all_query_indexes_options.hxx +0 -30
  97. data/ext/couchbase/couchbase/get_and_lock_options.hxx +2 -2
  98. data/ext/couchbase/couchbase/get_and_touch_options.hxx +2 -2
  99. data/ext/couchbase/couchbase/get_bucket_options.hxx +43 -0
  100. data/ext/couchbase/couchbase/get_options.hxx +2 -2
  101. data/ext/couchbase/couchbase/insert_options.hxx +3 -3
  102. data/ext/couchbase/couchbase/key_value_error_context.hxx +7 -2
  103. data/ext/couchbase/couchbase/lookup_in_all_replicas_options.hxx +109 -0
  104. data/ext/couchbase/couchbase/lookup_in_any_replica_options.hxx +101 -0
  105. data/ext/couchbase/couchbase/lookup_in_options.hxx +2 -2
  106. data/ext/couchbase/couchbase/lookup_in_replica_result.hxx +74 -0
  107. data/ext/couchbase/couchbase/lookup_in_result.hxx +26 -0
  108. data/ext/couchbase/couchbase/management/bucket_settings.hxx +116 -0
  109. data/ext/couchbase/couchbase/manager_error_context.hxx +29 -53
  110. data/ext/couchbase/couchbase/mutate_in_options.hxx +2 -2
  111. data/ext/couchbase/couchbase/query_error_context.hxx +3 -1
  112. data/ext/couchbase/couchbase/query_index_manager.hxx +16 -83
  113. data/ext/couchbase/couchbase/query_options.hxx +18 -0
  114. data/ext/couchbase/couchbase/remove_options.hxx +2 -2
  115. data/ext/couchbase/couchbase/replace_options.hxx +3 -3
  116. data/ext/couchbase/couchbase/security_options.hxx +15 -0
  117. data/ext/couchbase/couchbase/subdocument_error_context.hxx +4 -2
  118. data/ext/couchbase/couchbase/touch_options.hxx +2 -2
  119. data/ext/couchbase/couchbase/unlock_options.hxx +2 -2
  120. data/ext/couchbase/couchbase/update_bucket_options.hxx +41 -0
  121. data/ext/couchbase/couchbase/upsert_options.hxx +3 -3
  122. data/ext/couchbase/couchbase/watch_query_indexes_options.hxx +0 -31
  123. data/ext/couchbase/test/CMakeLists.txt +1 -0
  124. data/ext/couchbase/test/test_integration_collections.cxx +6 -0
  125. data/ext/couchbase/test/test_integration_crud.cxx +5 -0
  126. data/ext/couchbase/test/test_integration_examples.cxx +137 -1
  127. data/ext/couchbase/test/test_integration_management.cxx +709 -266
  128. data/ext/couchbase/test/test_integration_query.cxx +19 -7
  129. data/ext/couchbase/test/test_integration_range_scan.cxx +351 -112
  130. data/ext/couchbase/test/test_integration_search.cxx +10 -1
  131. data/ext/couchbase/test/test_integration_subdoc.cxx +655 -0
  132. data/ext/couchbase/test/test_transaction_public_async_api.cxx +13 -12
  133. data/ext/couchbase/test/test_transaction_public_blocking_api.cxx +27 -21
  134. data/ext/couchbase/test/test_unit_connection_string.cxx +29 -0
  135. data/ext/couchbase/test/test_unit_query.cxx +75 -0
  136. data/ext/couchbase.cxx +583 -29
  137. data/ext/revisions.rb +3 -3
  138. data/lib/couchbase/cluster.rb +1 -1
  139. data/lib/couchbase/collection.rb +108 -0
  140. data/lib/couchbase/collection_options.rb +100 -0
  141. data/lib/couchbase/errors.rb +5 -0
  142. data/lib/couchbase/key_value_scan.rb +125 -0
  143. data/lib/couchbase/options.rb +151 -0
  144. data/lib/couchbase/scope.rb +1 -1
  145. data/lib/couchbase/utils/time.rb +14 -1
  146. data/lib/couchbase/version.rb +1 -1
  147. metadata +41 -7
  148. data/ext/couchbase/core/impl/collection_query_index_manager.cxx +0 -93
@@ -0,0 +1,80 @@
1
+ /* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
2
+ /*
3
+ * Copyright 2020-Present Couchbase, Inc.
4
+ *
5
+ * Licensed under the Apache License, Version 2.0 (the "License");
6
+ * you may not use this file except in compliance with the License.
7
+ * You may obtain a copy of the License at
8
+ *
9
+ * http://www.apache.org/licenses/LICENSE-2.0
10
+ *
11
+ * Unless required by applicable law or agreed to in writing, software
12
+ * distributed under the License is distributed on an "AS IS" BASIS,
13
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14
+ * See the License for the specific language governing permissions and
15
+ * limitations under the License.
16
+ */
17
+
18
+ #pragma once
19
+
20
+ #include <couchbase/lookup_in_all_replicas_options.hxx>
21
+ #include <couchbase/lookup_in_replica_result.hxx>
22
+
23
+ #include "core/document_id.hxx"
24
+ #include "core/error_context/key_value.hxx"
25
+ #include "core/utils/movable_function.hxx"
26
+
27
+ #include <vector>
28
+
29
+ namespace couchbase::core::impl
30
+ {
31
+
32
+ class lookup_in_all_replicas_request
33
+ {
34
+ public:
35
+ explicit lookup_in_all_replicas_request(std::string bucket_name,
36
+ std::string scope_name,
37
+ std::string collection_name,
38
+ std::string document_key,
39
+ std::vector<couchbase::core::impl::subdoc::command> specs,
40
+ std::optional<std::chrono::milliseconds> timeout)
41
+ : id_{ std::move(bucket_name), std::move(scope_name), std::move(collection_name), std::move(document_key) }
42
+ , specs_{ std::move(specs) }
43
+ , timeout_{ timeout }
44
+ {
45
+ }
46
+
47
+ [[nodiscard]] const auto& id() const
48
+ {
49
+ return id_;
50
+ }
51
+
52
+ [[nodiscard]] const auto& specs() const
53
+ {
54
+ return specs_;
55
+ }
56
+
57
+ [[nodiscard]] const auto& timeout() const
58
+ {
59
+ return timeout_;
60
+ }
61
+
62
+ private:
63
+ core::document_id id_;
64
+ std::vector<couchbase::core::impl::subdoc::command> specs_;
65
+ std::optional<std::chrono::milliseconds> timeout_{};
66
+ };
67
+
68
+ using movable_lookup_in_all_replicas_handler =
69
+ utils::movable_function<void(couchbase::subdocument_error_context, lookup_in_all_replicas_result)>;
70
+
71
+ void
72
+ initiate_lookup_in_all_replicas_operation(std::shared_ptr<cluster> core,
73
+ const std::string& bucket_name,
74
+ const std::string& scope_name,
75
+ const std::string& collection_name,
76
+ std::string document_key,
77
+ const std::vector<subdoc::command>& specs,
78
+ std::optional<std::chrono::milliseconds> timeout,
79
+ movable_lookup_in_all_replicas_handler&& handler);
80
+ } // namespace couchbase::core::impl
@@ -0,0 +1,167 @@
1
+ /* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
2
+ /*
3
+ * Copyright 2020-Present Couchbase, Inc.
4
+ *
5
+ * Licensed under the Apache License, Version 2.0 (the "License");
6
+ * you may not use this file except in compliance with the License.
7
+ * You may obtain a copy of the License at
8
+ *
9
+ * http://www.apache.org/licenses/LICENSE-2.0
10
+ *
11
+ * Unless required by applicable law or agreed to in writing, software
12
+ * distributed under the License is distributed on an "AS IS" BASIS,
13
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14
+ * See the License for the specific language governing permissions and
15
+ * limitations under the License.
16
+ */
17
+
18
+ #include "lookup_in_any_replica.hxx"
19
+ #include "lookup_in_replica.hxx"
20
+
21
+ #include "core/cluster.hxx"
22
+ #include "core/error_context/key_value.hxx"
23
+ #include "core/operations/document_lookup_in.hxx"
24
+ #include "core/topology/configuration.hxx"
25
+
26
+ namespace couchbase::core::impl
27
+ {
28
+ void
29
+ initiate_lookup_in_any_replica_operation(std::shared_ptr<cluster> core,
30
+ const std::string& bucket_name,
31
+ const std::string& scope_name,
32
+ const std::string& collection_name,
33
+ std::string document_key,
34
+ const std::vector<subdoc::command>& specs,
35
+ lookup_in_any_replica_options::built options,
36
+ lookup_in_any_replica_handler&& handler)
37
+ {
38
+ return initiate_lookup_in_any_replica_operation(std::move(core),
39
+ bucket_name,
40
+ scope_name,
41
+ collection_name,
42
+ std::move(document_key),
43
+ specs,
44
+ options.timeout,
45
+ movable_lookup_in_any_replica_handler{ std::move(handler) });
46
+ }
47
+
48
+ void
49
+ initiate_lookup_in_any_replica_operation(std::shared_ptr<cluster> core,
50
+ const std::string& bucket_name,
51
+ const std::string& scope_name,
52
+ const std::string& collection_name,
53
+ std::string document_key,
54
+ const std::vector<subdoc::command>& specs,
55
+ std::optional<std::chrono::milliseconds> timeout,
56
+ movable_lookup_in_any_replica_handler&& handler)
57
+ {
58
+ auto request = std::make_shared<couchbase::core::impl::lookup_in_any_replica_request>(
59
+ bucket_name, scope_name, collection_name, std::move(document_key), specs, timeout);
60
+ core->with_bucket_configuration(
61
+ bucket_name,
62
+ [core, r = std::move(request), h = std::move(handler)](std::error_code ec, const core::topology::configuration& config) mutable {
63
+ if (!config.supports_subdoc_read_replica()) {
64
+ ec = errc::common::feature_not_available;
65
+ }
66
+ if (r->specs().size() > 16) {
67
+ ec = errc::common::invalid_argument;
68
+ }
69
+ if (ec) {
70
+ std::optional<std::string> first_error_path{};
71
+ std::optional<std::size_t> first_error_index{};
72
+ return h(
73
+ make_subdocument_error_context(make_key_value_error_context(ec, r->id()), ec, first_error_path, first_error_index, false),
74
+ lookup_in_replica_result{});
75
+ }
76
+ struct replica_context {
77
+ replica_context(movable_lookup_in_any_replica_handler handler, std::uint32_t expected_responses)
78
+ : handler_(std::move(handler))
79
+ , expected_responses_(expected_responses)
80
+ {
81
+ }
82
+
83
+ movable_lookup_in_any_replica_handler handler_;
84
+ std::uint32_t expected_responses_;
85
+ bool done_{ false };
86
+ std::mutex mutex_{};
87
+ };
88
+ auto ctx = std::make_shared<replica_context>(std::move(h), config.num_replicas.value_or(0U) + 1U);
89
+
90
+ for (std::size_t idx = 1U; idx <= config.num_replicas.value_or(0U); ++idx) {
91
+ document_id replica_id{ r->id() };
92
+ replica_id.node_index(idx);
93
+ core->execute(impl::lookup_in_replica_request{ std::move(replica_id), r->specs(), r->timeout() },
94
+ [ctx](impl::lookup_in_replica_response&& resp) {
95
+ movable_lookup_in_any_replica_handler local_handler;
96
+ {
97
+ std::scoped_lock lock(ctx->mutex_);
98
+ if (ctx->done_) {
99
+ return;
100
+ }
101
+ --ctx->expected_responses_;
102
+ if (resp.ctx.ec()) {
103
+ if (ctx->expected_responses_ > 0) {
104
+ // just ignore the response
105
+ return;
106
+ }
107
+ // consider document irretrievable and give up
108
+ resp.ctx.override_ec(errc::key_value::document_irretrievable);
109
+ }
110
+ ctx->done_ = true;
111
+ std::swap(local_handler, ctx->handler_);
112
+ }
113
+ if (local_handler) {
114
+ std::vector<lookup_in_replica_result::entry> entries;
115
+ for (auto& field : resp.fields) {
116
+ lookup_in_replica_result::entry entry{};
117
+ entry.path = field.path;
118
+ entry.original_index = field.original_index;
119
+ entry.exists = field.exists;
120
+ entry.value = field.value;
121
+ entries.emplace_back(entry);
122
+ }
123
+ return local_handler(std::move(resp.ctx),
124
+ lookup_in_replica_result{ resp.cas, entries, resp.deleted, true /* replica */ });
125
+ }
126
+ });
127
+ }
128
+
129
+ core::operations::lookup_in_request active{ document_id{ r->id() } };
130
+ active.specs = r->specs();
131
+ active.timeout = r->timeout();
132
+ core->execute(active, [ctx](core::operations::lookup_in_response&& resp) {
133
+ movable_lookup_in_any_replica_handler local_handler{};
134
+ {
135
+ std::scoped_lock lock(ctx->mutex_);
136
+ if (ctx->done_) {
137
+ return;
138
+ }
139
+ --ctx->expected_responses_;
140
+ if (resp.ctx.ec()) {
141
+ if (ctx->expected_responses_ > 0) {
142
+ // just ignore the response
143
+ return;
144
+ }
145
+ // consider document irretrievable and give up
146
+ resp.ctx.override_ec(errc::key_value::document_irretrievable);
147
+ }
148
+ ctx->done_ = true;
149
+ std::swap(local_handler, ctx->handler_);
150
+ }
151
+ if (local_handler) {
152
+ std::vector<lookup_in_replica_result::entry> entries;
153
+ for (auto& field : resp.fields) {
154
+ lookup_in_replica_result::entry entry{};
155
+ entry.path = field.path;
156
+ entry.original_index = field.original_index;
157
+ entry.exists = field.exists;
158
+ entry.value = field.value;
159
+ entries.emplace_back(entry);
160
+ }
161
+ return local_handler(std::move(resp.ctx),
162
+ lookup_in_replica_result{ resp.cas, entries, resp.deleted, false /* active */ });
163
+ }
164
+ });
165
+ });
166
+ }
167
+ } // namespace couchbase::core::impl
@@ -0,0 +1,75 @@
1
+ /* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
2
+ /*
3
+ * Copyright 2020-Present Couchbase, Inc.
4
+ *
5
+ * Licensed under the Apache License, Version 2.0 (the "License");
6
+ * you may not use this file except in compliance with the License.
7
+ * You may obtain a copy of the License at
8
+ *
9
+ * http://www.apache.org/licenses/LICENSE-2.0
10
+ *
11
+ * Unless required by applicable law or agreed to in writing, software
12
+ * distributed under the License is distributed on an "AS IS" BASIS,
13
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14
+ * See the License for the specific language governing permissions and
15
+ * limitations under the License.
16
+ */
17
+
18
+ #pragma once
19
+
20
+ #include <couchbase/lookup_in_any_replica_options.hxx>
21
+ #include <couchbase/lookup_in_replica_result.hxx>
22
+
23
+ #include "core/document_id.hxx"
24
+ #include "core/utils/movable_function.hxx"
25
+
26
+ namespace couchbase::core::impl
27
+ {
28
+ class lookup_in_any_replica_request
29
+ {
30
+ public:
31
+ explicit lookup_in_any_replica_request(std::string bucket_name,
32
+ std::string scope_name,
33
+ std::string collection_name,
34
+ std::string document_key,
35
+ std::vector<couchbase::core::impl::subdoc::command> specs,
36
+ std::optional<std::chrono::milliseconds> timeout)
37
+ : id_{ std::move(bucket_name), std::move(scope_name), std::move(collection_name), std::move(document_key) }
38
+ , specs_{ std::move(specs) }
39
+ , timeout_{ timeout }
40
+ {
41
+ }
42
+
43
+ [[nodiscard]] const auto& id() const
44
+ {
45
+ return id_;
46
+ }
47
+
48
+ [[nodiscard]] const auto& specs() const
49
+ {
50
+ return specs_;
51
+ }
52
+
53
+ [[nodiscard]] const auto& timeout() const
54
+ {
55
+ return timeout_;
56
+ }
57
+
58
+ private:
59
+ core::document_id id_;
60
+ std::vector<couchbase::core::impl::subdoc::command> specs_;
61
+ std::optional<std::chrono::milliseconds> timeout_{};
62
+ };
63
+
64
+ using movable_lookup_in_any_replica_handler = utils::movable_function<void(couchbase::subdocument_error_context, lookup_in_replica_result)>;
65
+
66
+ void
67
+ initiate_lookup_in_any_replica_operation(std::shared_ptr<cluster> core,
68
+ const std::string& bucket_name,
69
+ const std::string& scope_name,
70
+ const std::string& collection_name,
71
+ std::string document_key,
72
+ const std::vector<subdoc::command>& specs,
73
+ std::optional<std::chrono::milliseconds> timeout,
74
+ movable_lookup_in_any_replica_handler&& handler);
75
+ } // namespace couchbase::core::impl
@@ -0,0 +1,97 @@
1
+ /* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
2
+ /*
3
+ * Copyright 2020-Present Couchbase, Inc.
4
+ *
5
+ * Licensed under the Apache License, Version 2.0 (the "License");
6
+ * you may not use this file except in compliance with the License.
7
+ * You may obtain a copy of the License at
8
+ *
9
+ * http://www.apache.org/licenses/LICENSE-2.0
10
+ *
11
+ * Unless required by applicable law or agreed to in writing, software
12
+ * distributed under the License is distributed on an "AS IS" BASIS,
13
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14
+ * See the License for the specific language governing permissions and
15
+ * limitations under the License.
16
+ */
17
+
18
+ #include "lookup_in_replica.hxx"
19
+ #include "core/impl/subdoc/path_flags.hxx"
20
+
21
+ #include <couchbase/error_codes.hxx>
22
+
23
+ namespace couchbase::core::impl
24
+ {
25
+ std::error_code
26
+ lookup_in_replica_request::encode_to(lookup_in_replica_request::encoded_request_type& encoded, mcbp_context&& /* context */)
27
+ {
28
+ for (std::size_t i = 0; i < specs.size(); ++i) {
29
+ specs[i].original_index_ = i;
30
+ }
31
+ std::stable_sort(specs.begin(), specs.end(), [](const auto& lhs, const auto& rhs) {
32
+ /* move XATTRs to the beginning of the vector */
33
+ return core::impl::subdoc::has_xattr_path_flag(lhs.flags_) && !core::impl::subdoc::has_xattr_path_flag(rhs.flags_);
34
+ });
35
+
36
+ encoded.opaque(opaque);
37
+ encoded.partition(partition);
38
+ encoded.body().id(id);
39
+ encoded.body().read_replica(true);
40
+ encoded.body().specs(specs);
41
+ return {};
42
+ }
43
+
44
+ lookup_in_replica_response
45
+ lookup_in_replica_request::make_response(key_value_error_context&& ctx, const encoded_response_type& encoded) const
46
+ {
47
+
48
+ bool deleted = false;
49
+ couchbase::cas cas{};
50
+ std::vector<lookup_in_replica_response::entry> fields{};
51
+ std::error_code ec = ctx.ec();
52
+ std::optional<std::size_t> first_error_index{};
53
+ std::optional<std::string> first_error_path{};
54
+
55
+ if (encoded.status() == key_value_status_code::subdoc_success_deleted ||
56
+ encoded.status() == key_value_status_code::subdoc_multi_path_failure_deleted) {
57
+ deleted = true;
58
+ }
59
+ if (!ctx.ec()) {
60
+ fields.resize(specs.size());
61
+ for (size_t i = 0; i < specs.size(); ++i) {
62
+ const auto& req_entry = specs[i];
63
+ fields[i].original_index = req_entry.original_index_;
64
+ fields[i].path = req_entry.path_;
65
+ fields[i].opcode = static_cast<protocol::subdoc_opcode>(req_entry.opcode_);
66
+ fields[i].status = key_value_status_code::success;
67
+ }
68
+ for (size_t i = 0; i < encoded.body().fields().size(); ++i) {
69
+ const auto& res_entry = encoded.body().fields()[i];
70
+ fields[i].status = res_entry.status;
71
+ fields[i].ec =
72
+ protocol::map_status_code(protocol::client_opcode::subdoc_multi_mutation, static_cast<std::uint16_t>(res_entry.status));
73
+ if (!fields[i].ec && !ctx.ec()) {
74
+ ec = fields[i].ec;
75
+ }
76
+ if (!first_error_index && !fields[i].ec) {
77
+ first_error_index = i;
78
+ first_error_path = fields[i].path;
79
+ }
80
+ fields[i].exists =
81
+ res_entry.status == key_value_status_code::success || res_entry.status == key_value_status_code::subdoc_success_deleted;
82
+ fields[i].value = utils::to_binary(res_entry.value);
83
+ }
84
+ if (!ec) {
85
+ cas = encoded.cas();
86
+ }
87
+ std::sort(fields.begin(), fields.end(), [](const auto& lhs, const auto& rhs) { return lhs.original_index < rhs.original_index; });
88
+ }
89
+
90
+ return lookup_in_replica_response{
91
+ make_subdocument_error_context(ctx, ec, first_error_path, first_error_index, deleted),
92
+ cas,
93
+ std::move(fields),
94
+ deleted,
95
+ };
96
+ }
97
+ } // namespace couchbase::core::impl
@@ -0,0 +1,67 @@
1
+ /* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
2
+ /*
3
+ * Copyright 2020-Present Couchbase, Inc.
4
+ *
5
+ * Licensed under the Apache License, Version 2.0 (the "License");
6
+ * you may not use this file except in compliance with the License.
7
+ * You may obtain a copy of the License at
8
+ *
9
+ * http://www.apache.org/licenses/LICENSE-2.0
10
+ *
11
+ * Unless required by applicable law or agreed to in writing, software
12
+ * distributed under the License is distributed on an "AS IS" BASIS,
13
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14
+ * See the License for the specific language governing permissions and
15
+ * limitations under the License.
16
+ */
17
+
18
+ #pragma once
19
+
20
+ #include "core/error_context/key_value.hxx"
21
+ #include "core/impl/subdoc/command.hxx"
22
+ #include "core/io/mcbp_context.hxx"
23
+ #include "core/io/retry_context.hxx"
24
+ #include "core/protocol/client_request.hxx"
25
+ #include "core/protocol/cmd_lookup_in_replica.hxx"
26
+ #include "core/public_fwd.hxx"
27
+ #include "core/timeout_defaults.hxx"
28
+
29
+ #include <couchbase/lookup_in_result.hxx>
30
+ #include <couchbase/subdocument_error_context.hxx>
31
+
32
+ namespace couchbase::core::impl
33
+ {
34
+ struct lookup_in_replica_response {
35
+ struct entry {
36
+ std::string path;
37
+ couchbase::codec::binary value;
38
+ std::size_t original_index;
39
+ bool exists;
40
+ protocol::subdoc_opcode opcode;
41
+ key_value_status_code status;
42
+ std::error_code ec{};
43
+ };
44
+ subdocument_error_context ctx{};
45
+ couchbase::cas cas{};
46
+ std::vector<entry> fields{};
47
+ bool deleted{ false };
48
+ };
49
+
50
+ struct lookup_in_replica_request {
51
+ using response_type = lookup_in_replica_response;
52
+ using encoded_request_type = protocol::client_request<protocol::lookup_in_replica_request_body>;
53
+ using encoded_response_type = protocol::client_response<protocol::lookup_in_replica_response_body>;
54
+
55
+ document_id id;
56
+ std::vector<couchbase::core::impl::subdoc::command> specs{};
57
+ std::optional<std::chrono::milliseconds> timeout{};
58
+ std::shared_ptr<couchbase::tracing::request_span> parent_span{ nullptr };
59
+ std::uint16_t partition{};
60
+ std::uint32_t opaque{};
61
+ io::retry_context<false> retries{};
62
+
63
+ [[nodiscard]] std::error_code encode_to(encoded_request_type& encoded, mcbp_context&& context);
64
+
65
+ [[nodiscard]] lookup_in_replica_response make_response(key_value_error_context&& ctx, const encoded_response_type& encoded) const;
66
+ };
67
+ } // namespace couchbase::core::impl
@@ -0,0 +1,100 @@
1
+ /* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
2
+ /*
3
+ * Copyright 2023-Present Couchbase, Inc.
4
+ *
5
+ * Licensed under the Apache License, Version 2.0 (the "License");
6
+ * you may not use this file except in compliance with the License.
7
+ * You may obtain a copy of the License at
8
+ *
9
+ * http://www.apache.org/licenses/LICENSE-2.0
10
+ *
11
+ * Unless required by applicable law or agreed to in writing, software
12
+ * distributed under the License is distributed on an "AS IS" BASIS,
13
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14
+ * See the License for the specific language governing permissions and
15
+ * limitations under the License.
16
+ */
17
+
18
+ #include "internal_manager_error_context.hxx"
19
+
20
+ #include <couchbase/manager_error_context.hxx>
21
+
22
+ namespace couchbase
23
+ {
24
+ manager_error_context::manager_error_context()
25
+ : internal_{ nullptr }
26
+ {
27
+ }
28
+
29
+ manager_error_context::manager_error_context(internal_manager_error_context ctx)
30
+ : internal_{ std::make_unique<internal_manager_error_context>(std::move(ctx)) }
31
+ {
32
+ }
33
+
34
+ manager_error_context::manager_error_context(manager_error_context&& other) = default;
35
+
36
+ manager_error_context&
37
+ manager_error_context::operator=(manager_error_context&& other) = default;
38
+
39
+ manager_error_context::~manager_error_context() = default;
40
+
41
+ auto
42
+ manager_error_context::ec() const -> std::error_code
43
+ {
44
+ return internal_->ec();
45
+ }
46
+
47
+ auto
48
+ manager_error_context::last_dispatched_from() const -> const std::optional<std::string>&
49
+ {
50
+ return internal_->last_dispatched_from();
51
+ }
52
+
53
+ auto
54
+ manager_error_context::last_dispatched_to() const -> const std::optional<std::string>&
55
+ {
56
+ return internal_->last_dispatched_to();
57
+ }
58
+
59
+ auto
60
+ manager_error_context::retry_attempts() const -> std::size_t
61
+ {
62
+ return internal_->retry_attempts();
63
+ }
64
+
65
+ auto
66
+ manager_error_context::retry_reasons() const -> const std::set<retry_reason>&
67
+ {
68
+ return internal_->retry_reasons();
69
+ }
70
+
71
+ auto
72
+ manager_error_context::retried_because_of(couchbase::retry_reason reason) const -> bool
73
+ {
74
+ return internal_->retried_because_of(reason);
75
+ }
76
+
77
+ auto
78
+ manager_error_context::path() const -> const std::string&
79
+ {
80
+ return internal_->path();
81
+ }
82
+
83
+ auto
84
+ manager_error_context::content() const -> const std::string&
85
+ {
86
+ return internal_->content();
87
+ }
88
+
89
+ auto
90
+ manager_error_context::client_context_id() const -> const std::string&
91
+ {
92
+ return internal_->client_context_id();
93
+ }
94
+
95
+ auto
96
+ manager_error_context::http_status() const -> std::uint32_t
97
+ {
98
+ return internal_->http_status();
99
+ }
100
+ } // namespace couchbase
@@ -161,6 +161,7 @@ build_query_request(std::string statement, query_options::built options)
161
161
  options.readonly,
162
162
  options.flex_index,
163
163
  options.preserve_expiry,
164
+ options.use_replica,
164
165
  options.max_parallelism,
165
166
  options.scan_cap,
166
167
  options.scan_wait,
@@ -0,0 +1,75 @@
1
+ /* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
2
+ /*
3
+ * Copyright 2020-Present Couchbase, Inc.
4
+ *
5
+ * Licensed under the Apache License, Version 2.0 (the "License");
6
+ * you may not use this file except in compliance with the License.
7
+ * You may obtain a copy of the License at
8
+ *
9
+ * http://www.apache.org/licenses/LICENSE-2.0
10
+ *
11
+ * Unless required by applicable law or agreed to in writing, software
12
+ * distributed under the License is distributed on an "AS IS" BASIS,
13
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14
+ * See the License for the specific language governing permissions and
15
+ * limitations under the License.
16
+ */
17
+
18
+ #include <couchbase/query_error_context.hxx>
19
+
20
+ #include <couchbase/fmt/retry_reason.hxx>
21
+
22
+ #include <tao/json/to_string.hpp>
23
+
24
+ namespace couchbase
25
+ {
26
+ auto
27
+ query_error_context::to_json() const -> std::string
28
+ {
29
+ tao::json::value json = {
30
+ {
31
+ "ec",
32
+ tao::json::value{
33
+ { "value", ec().value() },
34
+ { "message", ec().message() },
35
+ },
36
+ },
37
+ { "operation_id", operation_id() },
38
+ { "retry_attempts", retry_attempts() },
39
+ { "client_context_id", client_context_id_ },
40
+ { "statement", statement_ },
41
+ { "method", statement_ },
42
+ { "path", statement_ },
43
+ { "http_status", http_status_ },
44
+ { "http_body", http_body_ },
45
+ { "hostname", hostname_ },
46
+ { "port", port_ },
47
+ };
48
+
49
+ if (const auto& val = parameters_; val.has_value()) {
50
+ json["parameters"] = val.value();
51
+ }
52
+ if (first_error_code_ > 0) {
53
+ json["first_error_code"] = first_error_code_;
54
+ }
55
+ if (!first_error_message_.empty()) {
56
+ json["first_error_message"] = first_error_message_;
57
+ }
58
+
59
+ if (const auto& reasons = retry_reasons(); !reasons.empty()) {
60
+ tao::json::value reasons_json = tao::json::empty_array;
61
+ for (const auto& reason : reasons) {
62
+ reasons_json.emplace_back(fmt::format("{}", reason));
63
+ }
64
+ json["retry_reasons"] = reasons_json;
65
+ }
66
+ if (const auto& val = last_dispatched_from(); val.has_value()) {
67
+ json["last_dispatched_from"] = val.value();
68
+ }
69
+ if (const auto& val = last_dispatched_to(); val.has_value()) {
70
+ json["last_dispatched_to"] = val.value();
71
+ }
72
+
73
+ return tao::json::to_string(json, 2);
74
+ }
75
+ } // namespace couchbase