couchbase 3.4.1 → 3.4.3
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- 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
|
|