couchbase 3.4.1 → 3.4.3
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/README.md +2 -2
- data/ext/couchbase/CMakeLists.txt +59 -7
- data/ext/couchbase/cmake/Documentation.cmake +0 -1
- data/ext/couchbase/cmake/OpenSSL.cmake +98 -3
- data/ext/couchbase/cmake/Testing.cmake +12 -4
- data/ext/couchbase/cmake/ThirdPartyDependencies.cmake +4 -0
- data/ext/couchbase/cmake/build_config.hxx.in +3 -0
- data/ext/couchbase/core/bucket.cxx +3 -2
- data/ext/couchbase/core/bucket.hxx +9 -0
- data/ext/couchbase/core/cluster.hxx +17 -0
- data/ext/couchbase/core/cluster_options.cxx +2 -2
- data/ext/couchbase/core/cluster_options.hxx +4 -7
- data/ext/couchbase/core/{config_profile.cxx → cluster_options_fwd.hxx} +7 -6
- data/ext/couchbase/core/config_profile.hxx +2 -65
- data/ext/couchbase/core/config_profiles.cxx +79 -0
- data/ext/couchbase/core/config_profiles.hxx +56 -0
- data/ext/couchbase/core/error_context/search.hxx +1 -1
- data/ext/couchbase/core/impl/analytics.cxx +237 -0
- data/ext/couchbase/core/impl/boolean_field_query.cxx +40 -0
- data/ext/couchbase/core/impl/boolean_query.cxx +62 -0
- data/ext/couchbase/core/impl/cluster.cxx +2 -1
- data/ext/couchbase/core/impl/conjunction_query.cxx +51 -0
- data/ext/couchbase/core/impl/date_range.cxx +89 -0
- data/ext/couchbase/core/impl/date_range_facet.cxx +54 -0
- data/ext/couchbase/core/impl/date_range_facet_result.cxx +64 -0
- data/ext/couchbase/core/impl/date_range_query.cxx +125 -0
- data/ext/couchbase/core/impl/disjunction_query.cxx +51 -0
- data/ext/couchbase/core/impl/dns_srv_tracker.cxx +5 -3
- data/ext/couchbase/core/impl/encoded_search_facet.hxx +29 -0
- data/ext/couchbase/core/impl/encoded_search_query.hxx +29 -0
- data/ext/couchbase/core/impl/encoded_search_sort.hxx +29 -0
- data/ext/couchbase/core/impl/geo_bounding_box_query.cxx +46 -0
- data/ext/couchbase/core/impl/geo_distance_query.cxx +43 -0
- data/ext/couchbase/core/impl/geo_polygon_query.cxx +46 -0
- data/ext/couchbase/core/impl/internal_date_range_facet_result.cxx +80 -0
- data/ext/couchbase/core/impl/internal_date_range_facet_result.hxx +48 -0
- data/ext/couchbase/core/impl/internal_numeric_range_facet_result.cxx +80 -0
- data/ext/couchbase/core/impl/internal_numeric_range_facet_result.hxx +48 -0
- data/ext/couchbase/core/impl/internal_search_error_context.cxx +141 -0
- data/ext/couchbase/core/impl/internal_search_error_context.hxx +61 -0
- data/ext/couchbase/core/impl/internal_search_meta_data.cxx +60 -0
- data/ext/couchbase/core/impl/internal_search_meta_data.hxx +41 -0
- data/ext/couchbase/core/impl/internal_search_result.cxx +84 -0
- data/ext/couchbase/core/impl/internal_search_result.hxx +43 -0
- data/ext/couchbase/core/impl/internal_search_row.cxx +82 -0
- data/ext/couchbase/core/impl/internal_search_row.hxx +56 -0
- data/ext/couchbase/core/impl/internal_search_row_location.hxx +32 -0
- data/ext/couchbase/core/impl/internal_search_row_locations.cxx +137 -0
- data/ext/couchbase/core/impl/internal_search_row_locations.hxx +45 -0
- data/ext/couchbase/core/impl/internal_term_facet_result.cxx +80 -0
- data/ext/couchbase/core/impl/internal_term_facet_result.hxx +48 -0
- data/ext/couchbase/core/impl/match_all_query.cxx +35 -0
- data/ext/couchbase/core/impl/match_none_query.cxx +35 -0
- data/ext/couchbase/core/impl/match_phrase_query.cxx +43 -0
- data/ext/couchbase/core/impl/match_query.cxx +59 -0
- data/ext/couchbase/core/impl/numeric_range.cxx +49 -0
- data/ext/couchbase/core/impl/numeric_range_facet.cxx +54 -0
- data/ext/couchbase/core/impl/numeric_range_facet_result.cxx +64 -0
- data/ext/couchbase/core/impl/numeric_range_query.cxx +56 -0
- data/ext/couchbase/core/impl/phrase_query.cxx +42 -0
- data/ext/couchbase/core/impl/prefix_query.cxx +40 -0
- data/ext/couchbase/core/impl/query.cxx +5 -5
- data/ext/couchbase/core/impl/query_string_query.cxx +37 -0
- data/ext/couchbase/core/impl/regexp_query.cxx +40 -0
- data/ext/couchbase/core/impl/search.cxx +191 -0
- data/ext/couchbase/core/impl/search_error_context.cxx +147 -0
- data/ext/couchbase/core/impl/search_meta_data.cxx +46 -0
- data/ext/couchbase/core/impl/search_result.cxx +66 -0
- data/ext/couchbase/core/impl/search_row.cxx +74 -0
- data/ext/couchbase/core/impl/search_row_location.cxx +64 -0
- data/ext/couchbase/core/impl/search_row_locations.cxx +66 -0
- data/ext/couchbase/core/impl/search_sort_field.cxx +104 -0
- data/ext/couchbase/core/impl/search_sort_id.cxx +43 -0
- data/ext/couchbase/core/impl/search_sort_score.cxx +43 -0
- data/ext/couchbase/core/impl/term_facet.cxx +36 -0
- data/ext/couchbase/core/impl/term_facet_result.cxx +64 -0
- data/ext/couchbase/core/impl/term_query.cxx +56 -0
- data/ext/couchbase/core/impl/term_range_query.cxx +57 -0
- data/ext/couchbase/core/impl/wildcard_query.cxx +40 -0
- data/ext/couchbase/core/io/dns_client.cxx +225 -0
- data/ext/couchbase/core/io/dns_client.hxx +19 -188
- data/ext/couchbase/core/io/http_context.hxx +1 -1
- data/ext/couchbase/core/io/http_session.hxx +10 -0
- data/ext/couchbase/core/io/http_session_manager.hxx +5 -3
- data/ext/couchbase/core/io/mcbp_session.cxx +28 -1
- data/ext/couchbase/core/io/retry_orchestrator.hxx +3 -2
- data/ext/couchbase/core/json_string.hxx +5 -0
- data/ext/couchbase/core/meta/version.cxx +18 -4
- data/ext/couchbase/core/mozilla_ca_bundle.hxx +39 -0
- data/ext/couchbase/core/operations/document_analytics.cxx +1 -0
- data/ext/couchbase/core/operations/document_analytics.hxx +1 -0
- data/ext/couchbase/core/operations/document_append.hxx +1 -1
- data/ext/couchbase/core/operations/document_decrement.hxx +1 -1
- data/ext/couchbase/core/operations/document_exists.hxx +1 -1
- data/ext/couchbase/core/operations/document_get.hxx +1 -1
- data/ext/couchbase/core/operations/document_get_and_lock.hxx +1 -1
- data/ext/couchbase/core/operations/document_get_and_touch.hxx +1 -1
- data/ext/couchbase/core/operations/document_get_projected.hxx +1 -1
- data/ext/couchbase/core/operations/document_increment.hxx +1 -1
- data/ext/couchbase/core/operations/document_insert.hxx +1 -1
- data/ext/couchbase/core/operations/document_lookup_in.hxx +1 -1
- data/ext/couchbase/core/operations/document_mutate_in.hxx +1 -1
- data/ext/couchbase/core/operations/document_prepend.hxx +1 -1
- data/ext/couchbase/core/operations/document_query.cxx +2 -0
- data/ext/couchbase/core/operations/document_query.hxx +6 -0
- data/ext/couchbase/core/operations/document_remove.hxx +1 -1
- data/ext/couchbase/core/operations/document_replace.hxx +1 -1
- data/ext/couchbase/core/operations/document_search.cxx +4 -1
- data/ext/couchbase/core/operations/document_search.hxx +2 -1
- data/ext/couchbase/core/operations/document_touch.hxx +1 -1
- data/ext/couchbase/core/operations/document_unlock.hxx +1 -1
- data/ext/couchbase/core/operations/document_upsert.hxx +1 -1
- data/ext/couchbase/core/operations/document_view.hxx +1 -0
- data/ext/couchbase/core/protocol/client_request.hxx +11 -2
- data/ext/couchbase/core/public_fwd.hxx +21 -0
- data/ext/couchbase/core/tls_verify_mode.hxx +26 -0
- data/ext/couchbase/core/topology/configuration.cxx +15 -2
- data/ext/couchbase/core/topology/configuration.hxx +5 -1
- data/ext/couchbase/core/transactions/active_transaction_record.hxx +2 -2
- data/ext/couchbase/core/transactions/attempt_context_impl.cxx +3 -0
- data/ext/couchbase/core/transactions/attempt_context_impl.hxx +1 -1
- data/ext/couchbase/core/transactions/attempt_context_testing_hooks.cxx +93 -0
- data/ext/couchbase/core/transactions/attempt_context_testing_hooks.hxx +48 -75
- data/ext/couchbase/core/transactions/cleanup_testing_hooks.cxx +52 -0
- data/ext/couchbase/core/transactions/cleanup_testing_hooks.hxx +17 -31
- data/ext/couchbase/core/transactions/exceptions.hxx +12 -9
- data/ext/couchbase/core/transactions/internal/transaction_context.hxx +12 -12
- data/ext/couchbase/core/transactions/internal/transactions_cleanup.hxx +7 -1
- data/ext/couchbase/core/transactions/transaction_context.cxx +1 -0
- data/ext/couchbase/core/transactions/transactions_cleanup.cxx +144 -155
- data/ext/couchbase/core/utils/connection_string.cxx +27 -3
- data/ext/couchbase/core/utils/connection_string.hxx +3 -3
- data/ext/couchbase/core/utils/json.cxx +4 -1
- data/ext/couchbase/couchbase/analytics_error_context.hxx +143 -0
- data/ext/couchbase/couchbase/analytics_meta_data.hxx +155 -0
- data/ext/couchbase/couchbase/analytics_metrics.hxx +163 -0
- data/ext/couchbase/couchbase/analytics_options.hxx +359 -0
- data/ext/couchbase/couchbase/analytics_result.hxx +102 -0
- data/ext/couchbase/couchbase/analytics_scan_consistency.hxx +46 -0
- data/ext/couchbase/couchbase/analytics_status.hxx +41 -0
- data/ext/couchbase/couchbase/analytics_warning.hxx +85 -0
- data/ext/couchbase/couchbase/behavior_options.hxx +10 -1
- data/ext/couchbase/couchbase/boolean_field_query.hxx +77 -0
- data/ext/couchbase/couchbase/boolean_query.hxx +223 -0
- data/ext/couchbase/couchbase/cluster.hxx +75 -1
- data/ext/couchbase/couchbase/conjunction_query.hxx +88 -0
- data/ext/couchbase/couchbase/date_range.hxx +69 -0
- data/ext/couchbase/couchbase/date_range_facet.hxx +56 -0
- data/ext/couchbase/couchbase/date_range_facet_result.hxx +55 -0
- data/ext/couchbase/couchbase/date_range_query.hxx +265 -0
- data/ext/couchbase/couchbase/disjunction_query.hxx +109 -0
- data/ext/couchbase/couchbase/doc_id_query.hxx +111 -0
- data/ext/couchbase/couchbase/error_context.hxx +7 -6
- data/ext/couchbase/couchbase/fmt/analytics_scan_consistency.hxx +52 -0
- data/ext/couchbase/couchbase/fmt/analytics_status.hxx +76 -0
- data/ext/couchbase/couchbase/fmt/search_scan_consistency.hxx +49 -0
- data/ext/couchbase/couchbase/geo_bounding_box_query.hxx +107 -0
- data/ext/couchbase/couchbase/geo_distance_query.hxx +109 -0
- data/ext/couchbase/couchbase/geo_point.hxx +32 -0
- data/ext/couchbase/couchbase/geo_polygon_query.hxx +85 -0
- data/ext/couchbase/couchbase/highlight_style.hxx +45 -0
- data/ext/couchbase/couchbase/match_all_query.hxx +43 -0
- data/ext/couchbase/couchbase/match_none_query.hxx +43 -0
- data/ext/couchbase/couchbase/match_operator.hxx +45 -0
- data/ext/couchbase/couchbase/match_phrase_query.hxx +108 -0
- data/ext/couchbase/couchbase/match_query.hxx +163 -0
- data/ext/couchbase/couchbase/numeric_range.hxx +58 -0
- data/ext/couchbase/couchbase/numeric_range_facet.hxx +56 -0
- data/ext/couchbase/couchbase/numeric_range_facet_result.hxx +55 -0
- data/ext/couchbase/couchbase/numeric_range_query.hxx +143 -0
- data/ext/couchbase/couchbase/phrase_query.hxx +93 -0
- data/ext/couchbase/couchbase/prefix_query.hxx +82 -0
- data/ext/couchbase/couchbase/query_options.hxx +0 -1
- data/ext/couchbase/couchbase/query_string_query.hxx +72 -0
- data/ext/couchbase/couchbase/regexp_query.hxx +82 -0
- data/ext/couchbase/couchbase/scope.hxx +73 -0
- data/ext/couchbase/couchbase/search_date_range.hxx +68 -0
- data/ext/couchbase/couchbase/search_error_context.hxx +138 -0
- data/ext/couchbase/couchbase/search_facet.hxx +60 -0
- data/ext/couchbase/couchbase/search_facet_result.hxx +50 -0
- data/ext/couchbase/couchbase/search_meta_data.hxx +85 -0
- data/ext/couchbase/couchbase/search_metrics.hxx +127 -0
- data/ext/couchbase/couchbase/search_numeric_range.hxx +69 -0
- data/ext/couchbase/couchbase/search_options.hxx +509 -0
- data/ext/couchbase/couchbase/search_query.hxx +69 -0
- data/ext/couchbase/couchbase/search_result.hxx +77 -0
- data/ext/couchbase/couchbase/search_row.hxx +104 -0
- data/ext/couchbase/couchbase/search_row_location.hxx +55 -0
- data/ext/couchbase/couchbase/search_row_locations.hxx +86 -0
- data/ext/couchbase/couchbase/search_scan_consistency.hxx +34 -0
- data/ext/couchbase/couchbase/search_sort.hxx +58 -0
- data/ext/couchbase/couchbase/search_sort_field.hxx +117 -0
- data/ext/couchbase/couchbase/search_sort_field_missing.hxx +26 -0
- data/ext/couchbase/couchbase/search_sort_field_mode.hxx +27 -0
- data/ext/couchbase/couchbase/search_sort_field_type.hxx +28 -0
- data/ext/couchbase/couchbase/search_sort_id.hxx +60 -0
- data/ext/couchbase/couchbase/search_sort_score.hxx +60 -0
- data/ext/couchbase/couchbase/search_term_range.hxx +51 -0
- data/ext/couchbase/couchbase/security_options.hxx +3 -0
- data/ext/couchbase/couchbase/term_facet.hxx +48 -0
- data/ext/couchbase/couchbase/term_facet_result.hxx +55 -0
- data/ext/couchbase/couchbase/term_query.hxx +151 -0
- data/ext/couchbase/couchbase/term_range_query.hxx +142 -0
- data/ext/couchbase/couchbase/tracing/request_span.hxx +63 -0
- data/ext/couchbase/couchbase/tracing/request_tracer.hxx +2 -40
- data/ext/couchbase/couchbase/transactions/async_attempt_context.hxx +83 -4
- data/ext/couchbase/couchbase/transactions/attempt_context.hxx +68 -1
- data/ext/couchbase/couchbase/transactions/transaction_get_result.hxx +2 -0
- data/ext/couchbase/couchbase/transactions/transaction_keyspace.hxx +11 -1
- data/ext/couchbase/couchbase/transactions/transaction_options.hxx +79 -8
- data/ext/couchbase/couchbase/transactions/transaction_query_options.hxx +128 -15
- data/ext/couchbase/couchbase/transactions/transaction_query_result.hxx +4 -0
- data/ext/couchbase/couchbase/transactions/transaction_result.hxx +1 -1
- data/ext/couchbase/couchbase/transactions/transactions_cleanup_config.hxx +5 -3
- data/ext/couchbase/couchbase/transactions/transactions_config.hxx +9 -5
- data/ext/couchbase/couchbase/transactions/transactions_query_config.hxx +6 -3
- data/ext/couchbase/couchbase/transactions.hxx +34 -1
- data/ext/couchbase/couchbase/wildcard_query.hxx +83 -0
- data/ext/couchbase/test/CMakeLists.txt +8 -7
- data/ext/couchbase/test/benchmark_helper_integration.hxx +2 -2
- data/ext/couchbase/test/test_helper.hxx +6 -6
- data/ext/couchbase/test/test_integration_analytics.cxx +314 -16
- data/ext/couchbase/test/test_integration_collections.cxx +7 -3
- data/ext/couchbase/test/test_integration_connect.cxx +7 -3
- data/ext/couchbase/test/test_integration_crud.cxx +19 -2
- data/ext/couchbase/test/test_integration_diagnostics.cxx +11 -5
- data/ext/couchbase/test/test_integration_durability.cxx +12 -7
- data/ext/couchbase/test/test_integration_examples.cxx +324 -11
- data/ext/couchbase/test/test_integration_management.cxx +162 -94
- data/ext/couchbase/test/test_integration_query.cxx +68 -10
- data/ext/couchbase/test/test_integration_range_scan.cxx +12 -12
- data/ext/couchbase/test/test_integration_read_replica.cxx +48 -11
- data/ext/couchbase/test/test_integration_search.cxx +621 -0
- data/ext/couchbase/test/test_integration_subdoc.cxx +62 -11
- data/ext/couchbase/test/test_integration_tracer.cxx +5 -0
- data/ext/couchbase/test/test_integration_transcoders.cxx +13 -5
- data/ext/couchbase/test/{test_transaction_transaction_context.cxx → test_transaction_context.cxx} +1 -1
- data/ext/couchbase/test/test_transaction_examples.cxx +195 -0
- data/ext/couchbase/test/{test_transaction_transaction_simple.cxx → test_transaction_simple.cxx} +90 -5
- data/ext/couchbase/test/{test_transaction_transaction_simple_async.cxx → test_transaction_simple_async.cxx} +19 -21
- data/ext/couchbase/test/test_unit_config_profiles.cxx +13 -13
- data/ext/couchbase/test/test_unit_connection_string.cxx +35 -0
- data/ext/couchbase/test/test_unit_json_transcoder.cxx +4 -0
- data/ext/couchbase/test/test_unit_search.cxx +427 -0
- data/ext/couchbase/test/test_unit_transaction_utils.cxx +10 -1
- data/ext/couchbase/test/test_unit_utils.cxx +8 -4
- data/ext/couchbase/third_party/snappy/CMakeLists.txt +150 -27
- data/ext/couchbase/third_party/snappy/cmake/config.h.in +28 -24
- data/ext/couchbase/third_party/snappy/snappy-internal.h +189 -25
- data/ext/couchbase/third_party/snappy/snappy-sinksource.cc +26 -9
- data/ext/couchbase/third_party/snappy/snappy-sinksource.h +11 -11
- data/ext/couchbase/third_party/snappy/snappy-stubs-internal.cc +1 -1
- data/ext/couchbase/third_party/snappy/snappy-stubs-internal.h +227 -308
- data/ext/couchbase/third_party/snappy/snappy-stubs-public.h.in +0 -11
- data/ext/couchbase/third_party/snappy/snappy.cc +1176 -410
- data/ext/couchbase/third_party/snappy/snappy.h +19 -4
- data/ext/couchbase.cxx +85 -22
- data/ext/revisions.rb +3 -3
- data/lib/couchbase/authenticator.rb +0 -1
- data/lib/couchbase/cluster.rb +13 -13
- data/lib/couchbase/cluster_registry.rb +7 -2
- data/lib/couchbase/config_profiles.rb +1 -1
- data/lib/couchbase/configuration.rb +3 -4
- data/lib/couchbase/json_transcoder.rb +12 -5
- data/lib/couchbase/management/collection_query_index_manager.rb +54 -15
- data/lib/couchbase/management/query_index_manager.rb +70 -5
- data/lib/couchbase/options.rb +85 -2
- data/lib/couchbase/raw_binary_transcoder.rb +37 -0
- data/lib/couchbase/raw_json_transcoder.rb +38 -0
- data/lib/couchbase/raw_string_transcoder.rb +40 -0
- data/lib/couchbase/search_options.rb +163 -240
- data/lib/couchbase/transcoder_flags.rb +62 -0
- data/lib/couchbase/version.rb +1 -1
- metadata +151 -12
- data/ext/couchbase/core/CMakeLists.txt +0 -0
- /data/ext/couchbase/test/{test_transaction_transaction_public_async_api.cxx → test_transaction_public_async_api.cxx} +0 -0
- /data/ext/couchbase/test/{test_transaction_transaction_public_blocking_api.cxx → test_transaction_public_blocking_api.cxx} +0 -0
@@ -0,0 +1,40 @@
|
|
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 "encoded_search_query.hxx"
|
19
|
+
|
20
|
+
#include <couchbase/wildcard_query.hxx>
|
21
|
+
|
22
|
+
namespace couchbase
|
23
|
+
{
|
24
|
+
auto
|
25
|
+
wildcard_query::encode() const -> encoded_search_query
|
26
|
+
{
|
27
|
+
encoded_search_query built;
|
28
|
+
built.query = tao::json::empty_object;
|
29
|
+
if (boost_) {
|
30
|
+
built.query["boost"] = boost_.value();
|
31
|
+
}
|
32
|
+
|
33
|
+
built.query["wildcard"] = wildcard_;
|
34
|
+
if (field_) {
|
35
|
+
built.query["field"] = field_.value();
|
36
|
+
}
|
37
|
+
|
38
|
+
return built;
|
39
|
+
}
|
40
|
+
} // namespace couchbase
|
@@ -0,0 +1,225 @@
|
|
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 "dns_client.hxx"
|
19
|
+
|
20
|
+
#include "core/logger/logger.hxx"
|
21
|
+
#include "core/utils/join_strings.hxx"
|
22
|
+
#include "dns_codec.hxx"
|
23
|
+
#include "dns_config.hxx"
|
24
|
+
|
25
|
+
#include <couchbase/error_codes.hxx>
|
26
|
+
|
27
|
+
#include <asio/ip/tcp.hpp>
|
28
|
+
#include <asio/read.hpp>
|
29
|
+
#include <asio/write.hpp>
|
30
|
+
|
31
|
+
#include <memory>
|
32
|
+
#include <sstream>
|
33
|
+
|
34
|
+
namespace couchbase::core::io::dns
|
35
|
+
{
|
36
|
+
class dns_srv_command : public std::enable_shared_from_this<dns_srv_command>
|
37
|
+
{
|
38
|
+
public:
|
39
|
+
dns_srv_command(asio::io_context& ctx,
|
40
|
+
const std::string& name,
|
41
|
+
const std::string& service,
|
42
|
+
const asio::ip::address& address,
|
43
|
+
std::uint16_t port,
|
44
|
+
utils::movable_function<void(couchbase::core::io::dns::dns_srv_response&& resp)> handler)
|
45
|
+
: deadline_(ctx)
|
46
|
+
, udp_deadline_(ctx)
|
47
|
+
, udp_(ctx)
|
48
|
+
, tcp_(ctx)
|
49
|
+
, address_(address)
|
50
|
+
, port_(port)
|
51
|
+
, handler_(std::move(handler))
|
52
|
+
{
|
53
|
+
static std::string protocol{ "_tcp" };
|
54
|
+
dns_message request{};
|
55
|
+
question_record qr;
|
56
|
+
qr.klass = resource_class::in;
|
57
|
+
qr.type = resource_type::srv;
|
58
|
+
qr.name.labels.push_back(service);
|
59
|
+
qr.name.labels.push_back(protocol);
|
60
|
+
std::string label;
|
61
|
+
std::istringstream name_stream(name);
|
62
|
+
while (std::getline(name_stream, label, '.')) {
|
63
|
+
qr.name.labels.push_back(label);
|
64
|
+
}
|
65
|
+
request.questions.emplace_back(qr);
|
66
|
+
send_buf_ = dns_codec::encode(request);
|
67
|
+
}
|
68
|
+
|
69
|
+
void execute(std::chrono::milliseconds total_timeout, std::chrono::milliseconds udp_timeout)
|
70
|
+
{
|
71
|
+
asio::ip::udp::endpoint endpoint(address_, port_);
|
72
|
+
udp_.open(endpoint.protocol());
|
73
|
+
udp_.async_send_to(
|
74
|
+
asio::buffer(send_buf_), endpoint, [self = shared_from_this()](std::error_code ec1, std::size_t /* bytes_transferred */) mutable {
|
75
|
+
if (ec1) {
|
76
|
+
self->udp_deadline_.cancel();
|
77
|
+
CB_LOG_DEBUG("DNS UDP write operation has got error {}, retrying with TCP", ec1.message());
|
78
|
+
return self->retry_with_tcp();
|
79
|
+
}
|
80
|
+
|
81
|
+
self->recv_buf_.resize(512);
|
82
|
+
self->udp_.async_receive_from(
|
83
|
+
asio::buffer(self->recv_buf_), self->udp_sender_, [self](std::error_code ec2, std::size_t bytes_transferred) mutable {
|
84
|
+
self->udp_deadline_.cancel();
|
85
|
+
if (ec2) {
|
86
|
+
CB_LOG_DEBUG("DNS UDP read operation has got error {}, retrying with TCP", ec2.message());
|
87
|
+
return self->retry_with_tcp();
|
88
|
+
}
|
89
|
+
self->recv_buf_.resize(bytes_transferred);
|
90
|
+
const dns_message message = dns_codec::decode(self->recv_buf_);
|
91
|
+
if (message.header.flags.tc == truncation::yes) {
|
92
|
+
self->udp_.close();
|
93
|
+
CB_LOG_DEBUG("DNS UDP read operation returned truncated response, retrying with TCP");
|
94
|
+
return self->retry_with_tcp();
|
95
|
+
}
|
96
|
+
self->deadline_.cancel();
|
97
|
+
dns_srv_response resp{ ec2 };
|
98
|
+
resp.targets.reserve(message.answers.size());
|
99
|
+
for (const auto& answer : message.answers) {
|
100
|
+
resp.targets.emplace_back(dns_srv_response::address{ utils::join_strings(answer.target.labels, "."), answer.port });
|
101
|
+
}
|
102
|
+
CB_LOG_DEBUG("DNS UDP returned {} records", resp.targets.size());
|
103
|
+
return self->handler_(std::move(resp));
|
104
|
+
});
|
105
|
+
});
|
106
|
+
udp_deadline_.expires_after(udp_timeout);
|
107
|
+
deadline_.async_wait([self = shared_from_this()](std::error_code ec) {
|
108
|
+
if (ec == asio::error::operation_aborted) {
|
109
|
+
return;
|
110
|
+
}
|
111
|
+
CB_LOG_DEBUG("DNS UDP deadline has been reached, cancelling UDP operation and fall back to TCP");
|
112
|
+
self->udp_.cancel();
|
113
|
+
return self->retry_with_tcp();
|
114
|
+
});
|
115
|
+
|
116
|
+
deadline_.expires_after(total_timeout);
|
117
|
+
deadline_.async_wait([self = shared_from_this()](std::error_code ec) {
|
118
|
+
if (ec == asio::error::operation_aborted) {
|
119
|
+
return;
|
120
|
+
}
|
121
|
+
CB_LOG_DEBUG("DNS deadline has been reached, cancelling in-flight operations (tcp.is_open={})", self->tcp_.is_open());
|
122
|
+
self->udp_.cancel();
|
123
|
+
if (self->tcp_.is_open()) {
|
124
|
+
self->tcp_.cancel();
|
125
|
+
}
|
126
|
+
});
|
127
|
+
}
|
128
|
+
|
129
|
+
private:
|
130
|
+
void retry_with_tcp()
|
131
|
+
{
|
132
|
+
if (bool expected_state{ false }; !retrying_with_tcp_.compare_exchange_strong(expected_state, true)) {
|
133
|
+
return;
|
134
|
+
}
|
135
|
+
|
136
|
+
const asio::ip::tcp::no_delay no_delay(true);
|
137
|
+
std::error_code ignore_ec;
|
138
|
+
tcp_.set_option(no_delay, ignore_ec);
|
139
|
+
const asio::ip::tcp::endpoint endpoint(address_, port_);
|
140
|
+
tcp_.async_connect(endpoint, [self = shared_from_this()](std::error_code ec1) mutable {
|
141
|
+
if (ec1) {
|
142
|
+
self->deadline_.cancel();
|
143
|
+
CB_LOG_DEBUG("DNS TCP connection has been aborted, {}", ec1.message());
|
144
|
+
return self->handler_({ ec1 });
|
145
|
+
}
|
146
|
+
auto send_size = static_cast<std::uint16_t>(self->send_buf_.size());
|
147
|
+
self->send_buf_.insert(self->send_buf_.begin(), static_cast<std::uint8_t>(send_size & 0xffU));
|
148
|
+
self->send_buf_.insert(self->send_buf_.begin(), static_cast<std::uint8_t>(send_size >> 8U));
|
149
|
+
asio::async_write(
|
150
|
+
self->tcp_, asio::buffer(self->send_buf_), [self](std::error_code ec2, std::size_t /* bytes_transferred */) mutable {
|
151
|
+
if (ec2) {
|
152
|
+
CB_LOG_DEBUG("DNS TCP write operation has been aborted, {}", ec2.message());
|
153
|
+
self->deadline_.cancel();
|
154
|
+
if (ec2 == asio::error::operation_aborted) {
|
155
|
+
ec2 = errc::common::unambiguous_timeout;
|
156
|
+
}
|
157
|
+
return self->handler_({ ec2 });
|
158
|
+
}
|
159
|
+
asio::async_read(self->tcp_,
|
160
|
+
asio::buffer(&self->recv_buf_size_, sizeof(std::uint16_t)),
|
161
|
+
[self](std::error_code ec3, std::size_t /* bytes_transferred */) mutable {
|
162
|
+
if (ec3) {
|
163
|
+
CB_LOG_DEBUG("DNS TCP buf size read operation has been aborted, {}", ec3.message());
|
164
|
+
self->deadline_.cancel();
|
165
|
+
return self->handler_({ ec3 });
|
166
|
+
}
|
167
|
+
self->recv_buf_size_ = utils::byte_swap(self->recv_buf_size_);
|
168
|
+
self->recv_buf_.resize(self->recv_buf_size_);
|
169
|
+
CB_LOG_DEBUG("DNS TCP schedule read of {} bytes", self->recv_buf_size_);
|
170
|
+
asio::async_read(self->tcp_,
|
171
|
+
asio::buffer(self->recv_buf_),
|
172
|
+
[self](std::error_code ec4, std::size_t bytes_transferred) mutable {
|
173
|
+
self->deadline_.cancel();
|
174
|
+
if (ec4) {
|
175
|
+
CB_LOG_DEBUG("DNS TCP read operation has been aborted, {}", ec4.message());
|
176
|
+
return self->handler_({ ec4 });
|
177
|
+
}
|
178
|
+
self->recv_buf_.resize(bytes_transferred);
|
179
|
+
const dns_message message = dns_codec::decode(self->recv_buf_);
|
180
|
+
dns_srv_response resp{ ec4 };
|
181
|
+
resp.targets.reserve(message.answers.size());
|
182
|
+
for (const auto& answer : message.answers) {
|
183
|
+
resp.targets.emplace_back(dns_srv_response::address{
|
184
|
+
utils::join_strings(answer.target.labels, "."), answer.port });
|
185
|
+
}
|
186
|
+
CB_LOG_DEBUG("DNS TCP returned {} records", resp.targets.size());
|
187
|
+
return self->handler_(std::move(resp));
|
188
|
+
});
|
189
|
+
});
|
190
|
+
});
|
191
|
+
});
|
192
|
+
}
|
193
|
+
|
194
|
+
asio::steady_timer deadline_;
|
195
|
+
asio::steady_timer udp_deadline_;
|
196
|
+
asio::ip::udp::socket udp_;
|
197
|
+
asio::ip::udp::endpoint udp_sender_{};
|
198
|
+
asio::ip::tcp::socket tcp_;
|
199
|
+
|
200
|
+
asio::ip::address address_;
|
201
|
+
std::uint16_t port_;
|
202
|
+
utils::movable_function<void(couchbase::core::io::dns::dns_srv_response&& resp)> handler_;
|
203
|
+
|
204
|
+
std::vector<std::uint8_t> send_buf_{};
|
205
|
+
std::uint16_t recv_buf_size_{ 0 };
|
206
|
+
std::vector<std::uint8_t> recv_buf_{};
|
207
|
+
|
208
|
+
std::atomic_bool retrying_with_tcp_{ false };
|
209
|
+
};
|
210
|
+
|
211
|
+
void
|
212
|
+
dns_client::query_srv(const std::string& name,
|
213
|
+
const std::string& service,
|
214
|
+
const dns_config& config,
|
215
|
+
utils::movable_function<void(dns_srv_response&&)>&& handler)
|
216
|
+
{
|
217
|
+
std::error_code ec;
|
218
|
+
auto address = asio::ip::address::from_string(config.nameserver(), ec);
|
219
|
+
if (ec) {
|
220
|
+
return handler({ ec });
|
221
|
+
}
|
222
|
+
auto cmd = std::make_shared<dns_srv_command>(ctx_, name, service, address, config.port(), std::move(handler));
|
223
|
+
return cmd->execute(config.timeout(), config.timeout() / 2);
|
224
|
+
}
|
225
|
+
} // namespace couchbase::core::io::dns
|
@@ -1,6 +1,6 @@
|
|
1
1
|
/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
|
2
2
|
/*
|
3
|
-
* Copyright 2020-
|
3
|
+
* Copyright 2020-Present Couchbase, Inc.
|
4
4
|
*
|
5
5
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
6
6
|
* you may not use this file except in compliance with the License.
|
@@ -17,207 +17,38 @@
|
|
17
17
|
|
18
18
|
#pragma once
|
19
19
|
|
20
|
-
#include "core/utils/
|
21
|
-
#include "dns_codec.hxx"
|
20
|
+
#include "core/utils/movable_function.hxx"
|
22
21
|
#include "dns_config.hxx"
|
23
22
|
|
24
|
-
#include <
|
23
|
+
#include <asio/io_context.hpp>
|
25
24
|
|
26
|
-
#include <
|
27
|
-
#include <
|
28
|
-
#include <
|
29
|
-
|
30
|
-
#include <memory>
|
31
|
-
#include <sstream>
|
25
|
+
#include <cinttypes>
|
26
|
+
#include <string>
|
27
|
+
#include <vector>
|
32
28
|
|
33
29
|
namespace couchbase::core::io::dns
|
34
30
|
{
|
31
|
+
struct dns_srv_response {
|
32
|
+
struct address {
|
33
|
+
std::string hostname;
|
34
|
+
std::uint16_t port;
|
35
|
+
};
|
36
|
+
std::error_code ec;
|
37
|
+
std::vector<address> targets{};
|
38
|
+
};
|
39
|
+
|
35
40
|
class dns_client
|
36
41
|
{
|
37
42
|
public:
|
38
|
-
struct dns_srv_response {
|
39
|
-
struct address {
|
40
|
-
std::string hostname;
|
41
|
-
std::uint16_t port;
|
42
|
-
};
|
43
|
-
std::error_code ec;
|
44
|
-
std::vector<address> targets{};
|
45
|
-
};
|
46
|
-
|
47
|
-
class dns_srv_command : public std::enable_shared_from_this<dns_srv_command>
|
48
|
-
{
|
49
|
-
public:
|
50
|
-
dns_srv_command(asio::io_context& ctx,
|
51
|
-
const std::string& name,
|
52
|
-
const std::string& service,
|
53
|
-
const asio::ip::address& address,
|
54
|
-
std::uint16_t port)
|
55
|
-
: deadline_(ctx)
|
56
|
-
, udp_(ctx)
|
57
|
-
, tcp_(ctx)
|
58
|
-
, address_(address)
|
59
|
-
, port_(port)
|
60
|
-
{
|
61
|
-
static std::string protocol{ "_tcp" };
|
62
|
-
dns_message request{};
|
63
|
-
question_record qr;
|
64
|
-
qr.klass = resource_class::in;
|
65
|
-
qr.type = resource_type::srv;
|
66
|
-
qr.name.labels.push_back(service);
|
67
|
-
qr.name.labels.push_back(protocol);
|
68
|
-
std::string label;
|
69
|
-
std::istringstream name_stream(name);
|
70
|
-
while (std::getline(name_stream, label, '.')) {
|
71
|
-
qr.name.labels.push_back(label);
|
72
|
-
}
|
73
|
-
request.questions.emplace_back(qr);
|
74
|
-
send_buf_ = dns_codec::encode(request);
|
75
|
-
}
|
76
|
-
|
77
|
-
template<class Handler>
|
78
|
-
void execute(std::chrono::milliseconds timeout, Handler&& handler)
|
79
|
-
{
|
80
|
-
asio::ip::udp::endpoint endpoint(address_, port_);
|
81
|
-
udp_.open(endpoint.protocol());
|
82
|
-
udp_.async_send_to(asio::buffer(send_buf_),
|
83
|
-
endpoint,
|
84
|
-
[self = shared_from_this(), handler = std::forward<Handler>(handler)](
|
85
|
-
std::error_code ec1, std::size_t /* bytes_transferred */) mutable {
|
86
|
-
if (ec1 == asio::error::operation_aborted) {
|
87
|
-
self->deadline_.cancel();
|
88
|
-
return handler({ errc::common::unambiguous_timeout });
|
89
|
-
}
|
90
|
-
if (ec1) {
|
91
|
-
self->deadline_.cancel();
|
92
|
-
return handler({ ec1 });
|
93
|
-
}
|
94
|
-
|
95
|
-
self->recv_buf_.resize(512);
|
96
|
-
self->udp_.async_receive_from(asio::buffer(self->recv_buf_),
|
97
|
-
self->udp_sender_,
|
98
|
-
[self, handler = std::forward<Handler>(handler)](
|
99
|
-
std::error_code ec2, std::size_t bytes_transferred) mutable {
|
100
|
-
self->deadline_.cancel();
|
101
|
-
if (ec2) {
|
102
|
-
return handler({ ec2 });
|
103
|
-
}
|
104
|
-
self->recv_buf_.resize(bytes_transferred);
|
105
|
-
dns_message message = dns_codec::decode(self->recv_buf_);
|
106
|
-
if (message.header.flags.tc == truncation::yes) {
|
107
|
-
self->udp_.close();
|
108
|
-
return self->retry_with_tcp(std::forward<Handler>(handler));
|
109
|
-
}
|
110
|
-
dns_srv_response resp{ ec2 };
|
111
|
-
resp.targets.reserve(message.answers.size());
|
112
|
-
for (const auto& answer : message.answers) {
|
113
|
-
resp.targets.emplace_back(dns_srv_response::address{
|
114
|
-
utils::join_strings(answer.target.labels, "."), answer.port });
|
115
|
-
}
|
116
|
-
return handler(std::move(resp));
|
117
|
-
});
|
118
|
-
});
|
119
|
-
deadline_.expires_after(timeout);
|
120
|
-
deadline_.async_wait([self = shared_from_this()](std::error_code ec) {
|
121
|
-
if (ec == asio::error::operation_aborted) {
|
122
|
-
return;
|
123
|
-
}
|
124
|
-
self->udp_.cancel();
|
125
|
-
if (self->tcp_.is_open()) {
|
126
|
-
self->tcp_.cancel();
|
127
|
-
}
|
128
|
-
});
|
129
|
-
}
|
130
|
-
|
131
|
-
private:
|
132
|
-
template<class Handler>
|
133
|
-
void retry_with_tcp(Handler&& handler)
|
134
|
-
{
|
135
|
-
asio::ip::tcp::no_delay no_delay(true);
|
136
|
-
std::error_code ignore_ec;
|
137
|
-
tcp_.set_option(no_delay, ignore_ec);
|
138
|
-
asio::ip::tcp::endpoint endpoint(address_, port_);
|
139
|
-
tcp_.async_connect(
|
140
|
-
endpoint, [self = shared_from_this(), handler = std::forward<Handler>(handler)](std::error_code ec1) mutable {
|
141
|
-
if (ec1) {
|
142
|
-
self->deadline_.cancel();
|
143
|
-
return handler({ ec1 });
|
144
|
-
}
|
145
|
-
auto send_size = static_cast<std::uint16_t>(self->send_buf_.size());
|
146
|
-
self->send_buf_.insert(self->send_buf_.begin(), std::uint8_t(send_size & 0xffU));
|
147
|
-
self->send_buf_.insert(self->send_buf_.begin(), std::uint8_t(send_size >> 8U));
|
148
|
-
asio::async_write(
|
149
|
-
self->tcp_,
|
150
|
-
asio::buffer(self->send_buf_),
|
151
|
-
[self, handler = std::forward<Handler>(handler)](std::error_code ec2, std::size_t /* bytes_transferred */) mutable {
|
152
|
-
if (ec2) {
|
153
|
-
self->deadline_.cancel();
|
154
|
-
if (ec2 == asio::error::operation_aborted) {
|
155
|
-
ec2 = errc::common::unambiguous_timeout;
|
156
|
-
}
|
157
|
-
return handler({ ec2 });
|
158
|
-
}
|
159
|
-
asio::async_read(self->tcp_,
|
160
|
-
asio::buffer(&self->recv_buf_size_, sizeof(std::uint16_t)),
|
161
|
-
[self, handler = std::forward<Handler>(handler)](std::error_code ec3,
|
162
|
-
std::size_t /* bytes_transferred */) mutable {
|
163
|
-
if (ec3) {
|
164
|
-
self->deadline_.cancel();
|
165
|
-
return handler({ ec3 });
|
166
|
-
}
|
167
|
-
self->recv_buf_size_ = utils::byte_swap(self->recv_buf_size_);
|
168
|
-
self->recv_buf_.resize(self->recv_buf_size_);
|
169
|
-
asio::async_read(self->tcp_,
|
170
|
-
asio::buffer(self->recv_buf_),
|
171
|
-
[self, handler = std::forward<Handler>(handler)](
|
172
|
-
std::error_code ec4, std::size_t bytes_transferred) mutable {
|
173
|
-
self->deadline_.cancel();
|
174
|
-
if (ec4) {
|
175
|
-
return handler({ ec4 });
|
176
|
-
}
|
177
|
-
self->recv_buf_.resize(bytes_transferred);
|
178
|
-
dns_message message = dns_codec::decode(self->recv_buf_);
|
179
|
-
dns_srv_response resp{ ec4 };
|
180
|
-
resp.targets.reserve(message.answers.size());
|
181
|
-
for (const auto& answer : message.answers) {
|
182
|
-
resp.targets.emplace_back(dns_srv_response::address{
|
183
|
-
utils::join_strings(answer.target.labels, "."), answer.port });
|
184
|
-
}
|
185
|
-
return handler(std::move(resp));
|
186
|
-
});
|
187
|
-
});
|
188
|
-
});
|
189
|
-
});
|
190
|
-
}
|
191
|
-
|
192
|
-
asio::steady_timer deadline_;
|
193
|
-
asio::ip::udp::socket udp_;
|
194
|
-
asio::ip::udp::endpoint udp_sender_{};
|
195
|
-
asio::ip::tcp::socket tcp_;
|
196
|
-
|
197
|
-
asio::ip::address address_;
|
198
|
-
std::uint16_t port_;
|
199
|
-
|
200
|
-
std::vector<std::uint8_t> send_buf_{};
|
201
|
-
std::uint16_t recv_buf_size_{ 0 };
|
202
|
-
std::vector<std::uint8_t> recv_buf_{};
|
203
|
-
};
|
204
|
-
|
205
43
|
explicit dns_client(asio::io_context& ctx)
|
206
44
|
: ctx_(ctx)
|
207
45
|
{
|
208
46
|
}
|
209
47
|
|
210
|
-
|
211
|
-
|
212
|
-
|
213
|
-
|
214
|
-
auto address = asio::ip::address::from_string(config.nameserver(), ec);
|
215
|
-
if (ec) {
|
216
|
-
return handler({ ec });
|
217
|
-
}
|
218
|
-
auto cmd = std::make_shared<dns_srv_command>(ctx_, name, service, address, config.port());
|
219
|
-
cmd->execute(config.timeout(), std::forward<Handler>(handler));
|
220
|
-
}
|
48
|
+
void query_srv(const std::string& name,
|
49
|
+
const std::string& service,
|
50
|
+
const dns_config& config,
|
51
|
+
utils::movable_function<void(couchbase::core::io::dns::dns_srv_response&& resp)>&& handler);
|
221
52
|
|
222
53
|
asio::io_context& ctx_;
|
223
54
|
};
|
@@ -216,11 +216,21 @@ class http_session : public std::enable_shared_from_this<http_session>
|
|
216
216
|
return id_;
|
217
217
|
}
|
218
218
|
|
219
|
+
[[nodiscard]] service_type type() const
|
220
|
+
{
|
221
|
+
return type_;
|
222
|
+
}
|
223
|
+
|
219
224
|
[[nodiscard]] const std::string& hostname() const
|
220
225
|
{
|
221
226
|
return hostname_;
|
222
227
|
}
|
223
228
|
|
229
|
+
[[nodiscard]] const std::string& port() const
|
230
|
+
{
|
231
|
+
return service_;
|
232
|
+
}
|
233
|
+
|
224
234
|
[[nodiscard]] const asio::ip::tcp::endpoint& endpoint()
|
225
235
|
{
|
226
236
|
std::scoped_lock lock(info_mutex_);
|
@@ -61,8 +61,9 @@ class http_session_manager
|
|
61
61
|
std::scoped_lock config_lock(config_mutex_, sessions_mutex_);
|
62
62
|
config_ = std::move(config);
|
63
63
|
for (auto& [type, sessions] : idle_sessions_) {
|
64
|
-
sessions.remove_if(
|
65
|
-
|
64
|
+
sessions.remove_if([&opts = options_, &cfg = config_](const auto& session) {
|
65
|
+
return session && !cfg.has_node(opts.network, session->type(), opts.enable_tls, session->hostname(), session->port());
|
66
|
+
});
|
66
67
|
}
|
67
68
|
}
|
68
69
|
|
@@ -224,7 +225,8 @@ class http_session_manager
|
|
224
225
|
{
|
225
226
|
{
|
226
227
|
std::scoped_lock lock(config_mutex_);
|
227
|
-
if (!session->keep_alive() ||
|
228
|
+
if (!session->keep_alive() ||
|
229
|
+
!config_.has_node(options_.network, session->type(), options_.enable_tls, session->hostname(), session->port())) {
|
228
230
|
return asio::post(session->get_executor(), [session]() { session->stop(); });
|
229
231
|
}
|
230
232
|
}
|
@@ -72,7 +72,7 @@ class collection_cache
|
|
72
72
|
if (auto ptr = cid_map_.find(path); ptr != cid_map_.end()) {
|
73
73
|
return ptr->second;
|
74
74
|
}
|
75
|
-
return
|
75
|
+
return std::nullopt;
|
76
76
|
}
|
77
77
|
|
78
78
|
void update(const std::string& path, std::uint32_t id)
|
@@ -380,6 +380,16 @@ class mcbp_session_impl
|
|
380
380
|
switch (static_cast<protocol::server_opcode>(msg.header.opcode)) {
|
381
381
|
case protocol::server_opcode::cluster_map_change_notification: {
|
382
382
|
protocol::cmd_info info{ session_->endpoint_address_, session_->endpoint_.port() };
|
383
|
+
if (session_->origin_.options().dump_configuration) {
|
384
|
+
std::string_view config_text{ reinterpret_cast<const char*>(msg.body.data()), msg.body.size() };
|
385
|
+
CB_LOG_TRACE(
|
386
|
+
"{} configuration from cluster_map_change_notification request (size={}, endpoint=\"{}:{}\"), {}",
|
387
|
+
session_->log_prefix_,
|
388
|
+
config_text.size(),
|
389
|
+
info.endpoint_address,
|
390
|
+
info.endpoint_port,
|
391
|
+
config_text);
|
392
|
+
}
|
383
393
|
protocol::server_request<protocol::cluster_map_change_notification_request_body> req(std::move(msg), info);
|
384
394
|
std::optional<topology::configuration> config = req.body().config();
|
385
395
|
if (session_ && config.has_value()) {
|
@@ -464,6 +474,15 @@ class mcbp_session_impl
|
|
464
474
|
switch (auto opcode = static_cast<protocol::client_opcode>(msg.header.opcode)) {
|
465
475
|
case protocol::client_opcode::get_cluster_config: {
|
466
476
|
protocol::cmd_info info{ session_->endpoint_address_, session_->endpoint_.port() };
|
477
|
+
if (session_->origin_.options().dump_configuration) {
|
478
|
+
std::string_view config_text{ reinterpret_cast<const char*>(msg.body.data()), msg.body.size() };
|
479
|
+
CB_LOG_TRACE("{} configuration from get_cluster_config response (size={}, endpoint=\"{}:{}\"), {}",
|
480
|
+
session_->log_prefix_,
|
481
|
+
config_text.size(),
|
482
|
+
info.endpoint_address,
|
483
|
+
info.endpoint_port,
|
484
|
+
config_text);
|
485
|
+
}
|
467
486
|
protocol::client_response<protocol::get_cluster_config_response_body> resp(std::move(msg), info);
|
468
487
|
if (resp.status() == key_value_status_code::success) {
|
469
488
|
if (session_) {
|
@@ -1138,6 +1157,14 @@ class mcbp_session_impl
|
|
1138
1157
|
if (utils::byte_swap(msg.header.bodylen) - offset > 0) {
|
1139
1158
|
std::string_view config_text{ reinterpret_cast<const char*>(msg.body.data()) + offset,
|
1140
1159
|
msg.body.size() - static_cast<std::size_t>(offset) };
|
1160
|
+
if (origin_.options().dump_configuration) {
|
1161
|
+
CB_LOG_TRACE("{} configuration from not_my_vbucket response (size={}, endpoint=\"{}:{}\"), {}",
|
1162
|
+
log_prefix_,
|
1163
|
+
config_text.size(),
|
1164
|
+
endpoint_address_,
|
1165
|
+
endpoint_.port(),
|
1166
|
+
config_text);
|
1167
|
+
}
|
1141
1168
|
auto config = protocol::parse_config(config_text, endpoint_address_, endpoint_.port());
|
1142
1169
|
CB_LOG_DEBUG("{} received not_my_vbucket status for {}, opaque={} with config rev={} in the payload",
|
1143
1170
|
log_prefix_,
|
@@ -55,13 +55,14 @@ retry_with_duration(std::shared_ptr<Manager> manager,
|
|
55
55
|
std::chrono::milliseconds duration)
|
56
56
|
{
|
57
57
|
command->request.retries.record_retry_attempt(reason);
|
58
|
-
CB_LOG_TRACE(R"({} retrying operation {} (duration={}ms, id="{}", reason={}, attempts={}))",
|
58
|
+
CB_LOG_TRACE(R"({} retrying operation {} (duration={}ms, id="{}", reason={}, attempts={}, last_dispatched_to=\"{}\"))",
|
59
59
|
manager->log_prefix(),
|
60
60
|
decltype(command->request)::encoded_request_type::body_type::opcode,
|
61
61
|
duration.count(),
|
62
62
|
command->id_,
|
63
63
|
reason,
|
64
|
-
command->request.retries.retry_attempts()
|
64
|
+
command->request.retries.retry_attempts(),
|
65
|
+
command->session_ ? command->session_->remote_address() : "");
|
65
66
|
manager->schedule_for_retry(command, duration);
|
66
67
|
}
|
67
68
|
|