couchbase 3.4.1 → 3.4.3

data/ext/couchbase/core/impl/date_range_facet.cxx

@@ -0,0 +1,54 @@
+/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *   Copyright 2023-Present Couchbase, Inc.
+ *
+ *   Licensed under the Apache License, Version 2.0 (the "License");
+ *   you may not use this file except in compliance with the License.
+ *   You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *   Unless required by applicable law or agreed to in writing, software
+ *   distributed under the License is distributed on an "AS IS" BASIS,
+ *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *   See the License for the specific language governing permissions and
+ *   limitations under the License.
+ */
+
+#include "encoded_search_facet.hxx"
+
+#include <couchbase/date_range_facet.hxx>
+#include <couchbase/error_codes.hxx>
+
+namespace couchbase
+{
+auto
+date_range_facet::encode() const -> encoded_search_facet
+{
+    encoded_search_facet built;
+    built.facet = {
+        { "field", field_ },
+    };
+    if (size_) {
+        built.facet["size"] = size_.value();
+    }
+    if (ranges_.empty()) {
+        return { errc::common::invalid_argument };
+    }
+    tao::json::value ranges = tao::json::empty_array;
+    for (const auto& range : ranges_) {
+        tao::json::value entry = {
+            { "field", range.name() },
+        };
+        if (const auto& start = range.start(); start.has_value()) {
+            entry["start"] = start.value();
+        }
+        if (const auto& end = range.end(); end.has_value()) {
+            entry["end"] = end.value();
+        }
+        ranges.emplace_back(entry);
+    }
+    built.facet["date_ranges"] = ranges;
+    return built;
+}
+} // namespace couchbase

data/ext/couchbase/core/impl/date_range_facet_result.cxx

@@ -0,0 +1,64 @@
+/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *   Copyright 2023-Present Couchbase, Inc.
+ *
+ *   Licensed under the Apache License, Version 2.0 (the "License");
+ *   you may not use this file except in compliance with the License.
+ *   You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *   Unless required by applicable law or agreed to in writing, software
+ *   distributed under the License is distributed on an "AS IS" BASIS,
+ *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *   See the License for the specific language governing permissions and
+ *   limitations under the License.
+ */
+
+#include "internal_date_range_facet_result.hxx"
+
+#include <couchbase/date_range_facet_result.hxx>
+
+namespace couchbase
+{
+date_range_facet_result::date_range_facet_result(internal_date_range_facet_result internal)
+  : internal_{ std::make_unique<internal_date_range_facet_result>(std::move(internal)) }
+{
+}
+
+auto
+date_range_facet_result::name() const -> const std::string&
+{
+    return internal_->name();
+}
+
+auto
+date_range_facet_result::field() const -> const std::string&
+{
+    return internal_->field();
+}
+
+auto
+date_range_facet_result::total() const -> std::uint64_t
+{
+    return internal_->total();
+}
+
+auto
+date_range_facet_result::missing() const -> std::uint64_t
+{
+    return internal_->missing();
+}
+
+auto
+date_range_facet_result::other() const -> std::uint64_t
+{
+    return internal_->other();
+}
+
+auto
+date_range_facet_result::date_ranges() const -> const std::vector<search_date_range>&
+{
+    return internal_->date_ranges();
+}
+} // namespace couchbase

data/ext/couchbase/core/impl/date_range_query.cxx

@@ -0,0 +1,125 @@
+/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *   Copyright 2020-Present Couchbase, Inc.
+ *
+ *   Licensed under the Apache License, Version 2.0 (the "License");
+ *   you may not use this file except in compliance with the License.
+ *   You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *   Unless required by applicable law or agreed to in writing, software
+ *   distributed under the License is distributed on an "AS IS" BASIS,
+ *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *   See the License for the specific language governing permissions and
+ *   limitations under the License.
+ */
+
+#include "encoded_search_query.hxx"
+
+#include <couchbase/date_range_query.hxx>
+
+#include <couchbase/error_codes.hxx>
+
+#include <fmt/chrono.h>
+
+namespace couchbase
+{
+
+static constexpr const char* iso_8601_format = "{:%Y-%m-%dT%H:%M:%S%z}";
+
+auto
+date_range_query::start(std::chrono::system_clock::time_point value) -> date_range_query&
+{
+    start_ = fmt::format(iso_8601_format, value);
+    return *this;
+}
+
+auto
+date_range_query::start(std::tm value) -> date_range_query&
+{
+    start_ = fmt::format(iso_8601_format, value);
+    return *this;
+}
+
+auto
+date_range_query::start(std::chrono::system_clock::time_point value, bool inclusive) -> date_range_query&
+{
+    start_ = fmt::format(iso_8601_format, value);
+    inclusive_start_ = inclusive;
+    return *this;
+}
+
+auto
+date_range_query::start(std::tm value, bool inclusive) -> date_range_query&
+{
+    start_ = fmt::format(iso_8601_format, value);
+    inclusive_start_ = inclusive;
+    return *this;
+}
+
+auto
+date_range_query::end(std::chrono::system_clock::time_point value) -> date_range_query&
+{
+
+    end_ = fmt::format(iso_8601_format, value);
+    return *this;
+}
+
+auto
+date_range_query::end(std::tm value) -> date_range_query&
+{
+    end_ = fmt::format(iso_8601_format, value);
+    return *this;
+}
+
+auto
+date_range_query::end(std::chrono::system_clock::time_point value, bool inclusive) -> date_range_query&
+{
+    end_ = fmt::format(iso_8601_format, value);
+    inclusive_end_ = inclusive;
+    return *this;
+}
+
+auto
+date_range_query::end(std::tm value, bool inclusive) -> date_range_query&
+{
+    end_ = fmt::format(iso_8601_format, value);
+    inclusive_end_ = inclusive;
+    return *this;
+}
+
+auto
+date_range_query::encode() const -> encoded_search_query
+{
+    if ((!start_ || start_->empty()) && (!end_ || end_->empty())) {
+        return { errc::common::invalid_argument };
+    }
+
+    encoded_search_query built;
+    built.query = tao::json::empty_object;
+    if (boost_) {
+        built.query["boost"] = boost_.value();
+    }
+    if (field_) {
+        built.query["field"] = field_.value();
+    }
+    if (start_) {
+        built.query["start"] = start_.value();
+        if (inclusive_start_.has_value()) {
+            built.query["inclusive_start"] = inclusive_start_.value();
+        }
+    }
+    if (end_) {
+        built.query["end"] = end_.value();
+        if (inclusive_end_.has_value()) {
+            built.query["inclusive_end"] = inclusive_end_.value();
+        }
+    }
+    if (date_time_parser_) {
+        built.query["datetime_parser"] = date_time_parser_.value();
+    }
+
+    return built;
+}
+} // namespace couchbase

data/ext/couchbase/core/impl/disjunction_query.cxx

@@ -0,0 +1,51 @@
+/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *   Copyright 2023-Present Couchbase, Inc.
+ *
+ *   Licensed under the Apache License, Version 2.0 (the "License");
+ *   you may not use this file except in compliance with the License.
+ *   You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *   Unless required by applicable law or agreed to in writing, software
+ *   distributed under the License is distributed on an "AS IS" BASIS,
+ *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *   See the License for the specific language governing permissions and
+ *   limitations under the License.
+ */
+
+#include "encoded_search_query.hxx"
+
+#include <couchbase/disjunction_query.hxx>
+
+#include <couchbase/error_codes.hxx>
+
+namespace couchbase
+{
+auto
+disjunction_query::encode() const -> encoded_search_query
+{
+    if (disjuncts_.empty()) {
+        return { errc::common::invalid_argument };
+    }
+
+    encoded_search_query built;
+    built.query = tao::json::empty_object;
+    if (boost_) {
+        built.query["boost"] = boost_.value();
+    }
+
+    tao::json::value disjuncts = tao::json::empty_array;
+    for (const auto& disjunct : disjuncts_) {
+        auto encoded = disjunct->encode();
+        if (encoded.ec) {
+            return { encoded.ec };
+        }
+        disjuncts.push_back(encoded.query);
+    }
+    built.query["disjuncts"] = disjuncts;
+
+    return built;
+}
+} // namespace couchbase

data/ext/couchbase/core/impl/dns_srv_tracker.cxx

@@ -16,11 +16,13 @@
  */
 
 #include "dns_srv_tracker.hxx"
-#include "../logger/logger.hxx"
-#include "../topology/configuration.hxx"
+
+#include "core/logger/logger.hxx"
+#include "core/utils/join_strings.hxx"
 
 #include <asio/bind_executor.hpp>
 #include <asio/io_context.hpp>
+#include <asio/post.hpp>
 
 #include <memory>
 
@@ -44,7 +46,7 @@ dns_srv_tracker::get_srv_nodes(utils::movable_function<void(origin::node_list, s
       address_,
       service_,
       config_,
-      [self = shared_from_this(), callback = std::move(callback)](couchbase::core::io::dns::dns_client::dns_srv_response&& resp) mutable {
+      [self = shared_from_this(), callback = std::move(callback)](couchbase::core::io::dns::dns_srv_response&& resp) mutable {
           origin::node_list nodes;
           if (resp.ec) {
               CB_LOG_WARNING("failed to fetch DNS SRV records for \"{}\" ({}), assuming that cluster is listening this address",

data/ext/couchbase/core/impl/encoded_search_facet.hxx

@@ -0,0 +1,29 @@
+/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *   Copyright 2023-Present Couchbase, Inc.
+ *
+ *   Licensed under the Apache License, Version 2.0 (the "License");
+ *   you may not use this file except in compliance with the License.
+ *   You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *   Unless required by applicable law or agreed to in writing, software
+ *   distributed under the License is distributed on an "AS IS" BASIS,
+ *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *   See the License for the specific language governing permissions and
+ *   limitations under the License.
+ */
+
+#pragma once
+
+#include <tao/json/value.hpp>
+
+namespace couchbase
+{
+struct encoded_search_facet {
+    std::error_code ec{};
+    tao::json::value facet{};
+};
+
+} // namespace couchbase

data/ext/couchbase/core/impl/encoded_search_query.hxx

@@ -0,0 +1,29 @@
+/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *   Copyright 2023-Present Couchbase, Inc.
+ *
+ *   Licensed under the Apache License, Version 2.0 (the "License");
+ *   you may not use this file except in compliance with the License.
+ *   You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *   Unless required by applicable law or agreed to in writing, software
+ *   distributed under the License is distributed on an "AS IS" BASIS,
+ *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *   See the License for the specific language governing permissions and
+ *   limitations under the License.
+ */
+
+#pragma once
+
+#include <tao/json/value.hpp>
+
+namespace couchbase
+{
+struct encoded_search_query {
+    std::error_code ec{};
+    tao::json::value query{};
+};
+
+} // namespace couchbase

data/ext/couchbase/core/impl/encoded_search_sort.hxx

@@ -0,0 +1,29 @@
+/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *   Copyright 2023-Present Couchbase, Inc.
+ *
+ *   Licensed under the Apache License, Version 2.0 (the "License");
+ *   you may not use this file except in compliance with the License.
+ *   You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *   Unless required by applicable law or agreed to in writing, software
+ *   distributed under the License is distributed on an "AS IS" BASIS,
+ *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *   See the License for the specific language governing permissions and
+ *   limitations under the License.
+ */
+
+#pragma once
+
+#include <tao/json/value.hpp>
+
+namespace couchbase
+{
+struct encoded_search_sort {
+    std::error_code ec{};
+    tao::json::value sort{};
+};
+
+} // namespace couchbase

data/ext/couchbase/core/impl/geo_bounding_box_query.cxx

@@ -0,0 +1,46 @@
+/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *   Copyright 2020-Present Couchbase, Inc.
+ *
+ *   Licensed under the Apache License, Version 2.0 (the "License");
+ *   you may not use this file except in compliance with the License.
+ *   You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *   Unless required by applicable law or agreed to in writing, software
+ *   distributed under the License is distributed on an "AS IS" BASIS,
+ *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *   See the License for the specific language governing permissions and
+ *   limitations under the License.
+ */
+
+#include "encoded_search_query.hxx"
+
+#include <couchbase/geo_bounding_box_query.hxx>
+
+namespace couchbase
+{
+auto
+geo_bounding_box_query::encode() const -> encoded_search_query
+{
+    encoded_search_query built;
+    built.query = tao::json::empty_object;
+    if (boost_) {
+        built.query["boost"] = boost_.value();
+    }
+    if (field_) {
+        built.query["field"] = field_.value();
+    }
+    built.query["top_left"] = {
+        { "lat", top_left_.latitude },
+        { "lon", top_left_.longitude },
+    };
+    built.query["bottom_right"] = {
+        { "lat", bottom_right_.latitude },
+        { "lon", bottom_right_.longitude },
+    };
+
+    return built;
+}
+} // namespace couchbase

data/ext/couchbase/core/impl/geo_distance_query.cxx

@@ -0,0 +1,43 @@
+/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *   Copyright 2020-Present Couchbase, Inc.
+ *
+ *   Licensed under the Apache License, Version 2.0 (the "License");
+ *   you may not use this file except in compliance with the License.
+ *   You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *   Unless required by applicable law or agreed to in writing, software
+ *   distributed under the License is distributed on an "AS IS" BASIS,
+ *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *   See the License for the specific language governing permissions and
+ *   limitations under the License.
+ */
+
+#include "encoded_search_query.hxx"
+
+#include <couchbase/geo_distance_query.hxx>
+
+namespace couchbase
+{
+auto
+geo_distance_query::encode() const -> encoded_search_query
+{
+    encoded_search_query built;
+    built.query = tao::json::empty_object;
+    if (boost_) {
+        built.query["boost"] = boost_.value();
+    }
+    if (field_) {
+        built.query["field"] = field_.value();
+    }
+    built.query["distance"] = distance_;
+    built.query["location"] = {
+        { "lat", location_.latitude },
+        { "lon", location_.longitude },
+    };
+
+    return built;
+}
+} // namespace couchbase

data/ext/couchbase/core/impl/geo_polygon_query.cxx

@@ -0,0 +1,46 @@
+/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *   Copyright 2020-Present Couchbase, Inc.
+ *
+ *   Licensed under the Apache License, Version 2.0 (the "License");
+ *   you may not use this file except in compliance with the License.
+ *   You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *   Unless required by applicable law or agreed to in writing, software
+ *   distributed under the License is distributed on an "AS IS" BASIS,
+ *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *   See the License for the specific language governing permissions and
+ *   limitations under the License.
+ */
+
+#include "encoded_search_query.hxx"
+
+#include <couchbase/geo_polygon_query.hxx>
+
+namespace couchbase
+{
+auto
+geo_polygon_query::encode() const -> encoded_search_query
+{
+    encoded_search_query built;
+    built.query = tao::json::empty_object;
+    if (boost_) {
+        built.query["boost"] = boost_.value();
+    }
+    if (field_) {
+        built.query["field"] = field_.value();
+    }
+    tao::json::value polygon_points = tao::json::empty_array;
+    for (const auto& point : polygon_points_) {
+        polygon_points.emplace_back(tao::json::value{
+          { "lat", point.latitude },
+          { "lon", point.longitude },
+        });
+    }
+    built.query["polygon_points"] = polygon_points;
+
+    return built;
+}
+} // namespace couchbase

data/ext/couchbase/core/impl/internal_date_range_facet_result.cxx

@@ -0,0 +1,80 @@
+/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *   Copyright 2023-Present Couchbase, Inc.
+ *
+ *   Licensed under the Apache License, Version 2.0 (the "License");
+ *   you may not use this file except in compliance with the License.
+ *   You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *   Unless required by applicable law or agreed to in writing, software
+ *   distributed under the License is distributed on an "AS IS" BASIS,
+ *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *   See the License for the specific language governing permissions and
+ *   limitations under the License.
+ */
+
+#include "internal_date_range_facet_result.hxx"
+
+namespace couchbase
+{
+
+static std::vector<search_date_range>
+map_ranges(const core::operations::search_response::search_facet& facet)
+{
+    std::vector<search_date_range> ranges;
+    ranges.reserve(facet.date_ranges.size());
+
+    for (const auto& range : facet.date_ranges) {
+        ranges.emplace_back(range.name, range.start, range.end, range.count);
+    }
+
+    return ranges;
+}
+
+internal_date_range_facet_result::internal_date_range_facet_result(const core::operations::search_response::search_facet& facet)
+  : name_{ facet.name }
+  , field_{ facet.field }
+  , total_{ facet.total }
+  , missing_{ facet.missing }
+  , other_{ facet.other }
+  , ranges_{ map_ranges(facet) }
+{
+}
+
+auto
+internal_date_range_facet_result::name() const -> const std::string&
+{
+    return name_;
+}
+auto
+internal_date_range_facet_result::field() const -> const std::string&
+{
+    return field_;
+}
+
+auto
+internal_date_range_facet_result::total() const -> std::uint64_t
+{
+    return total_;
+}
+
+auto
+internal_date_range_facet_result::missing() const -> std::uint64_t
+{
+    return missing_;
+}
+
+auto
+internal_date_range_facet_result::other() const -> std::uint64_t
+{
+    return other_;
+}
+
+auto
+internal_date_range_facet_result::date_ranges() const -> const std::vector<search_date_range>&
+{
+    return ranges_;
+}
+} // namespace couchbase

data/ext/couchbase/core/impl/internal_date_range_facet_result.hxx

@@ -0,0 +1,48 @@
+/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *   Copyright 2023-Present Couchbase, Inc.
+ *
+ *   Licensed under the Apache License, Version 2.0 (the "License");
+ *   you may not use this file except in compliance with the License.
+ *   You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *   Unless required by applicable law or agreed to in writing, software
+ *   distributed under the License is distributed on an "AS IS" BASIS,
+ *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *   See the License for the specific language governing permissions and
+ *   limitations under the License.
+ */
+
+#pragma once
+
+#include "core/operations/document_search.hxx"
+
+#include <couchbase/search_date_range.hxx>
+#include <couchbase/search_meta_data.hxx>
+
+namespace couchbase
+{
+class internal_date_range_facet_result
+{
+  public:
+    explicit internal_date_range_facet_result(const core::operations::search_response::search_facet& facet);
+
+    [[nodiscard]] auto name() const -> const std::string&;
+    [[nodiscard]] auto field() const -> const std::string&;
+    [[nodiscard]] auto total() const -> std::uint64_t;
+    [[nodiscard]] auto missing() const -> std::uint64_t;
+    [[nodiscard]] auto other() const -> std::uint64_t;
+    [[nodiscard]] auto date_ranges() const -> const std::vector<search_date_range>&;
+
+  private:
+    std::string name_;
+    std::string field_;
+    std::uint64_t total_{};
+    std::uint64_t missing_{};
+    std::uint64_t other_{};
+    std::vector<search_date_range> ranges_;
+};
+
+} // namespace couchbase