couchbase 3.4.3 → 3.4.5

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

@@ -0,0 +1,67 @@
+/* -*- 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.
+ */
+
+#pragma once
+
+#include "core/error_context/key_value.hxx"
+#include "core/impl/subdoc/command.hxx"
+#include "core/io/mcbp_context.hxx"
+#include "core/io/retry_context.hxx"
+#include "core/protocol/client_request.hxx"
+#include "core/protocol/cmd_lookup_in_replica.hxx"
+#include "core/public_fwd.hxx"
+#include "core/timeout_defaults.hxx"
+
+#include <couchbase/lookup_in_result.hxx>
+#include <couchbase/subdocument_error_context.hxx>
+
+namespace couchbase::core::impl
+{
+struct lookup_in_replica_response {
+    struct entry {
+        std::string path;
+        couchbase::codec::binary value;
+        std::size_t original_index;
+        bool exists;
+        protocol::subdoc_opcode opcode;
+        key_value_status_code status;
+        std::error_code ec{};
+    };
+    subdocument_error_context ctx{};
+    couchbase::cas cas{};
+    std::vector<entry> fields{};
+    bool deleted{ false };
+};
+
+struct lookup_in_replica_request {
+    using response_type = lookup_in_replica_response;
+    using encoded_request_type = protocol::client_request<protocol::lookup_in_replica_request_body>;
+    using encoded_response_type = protocol::client_response<protocol::lookup_in_replica_response_body>;
+
+    document_id id;
+    std::vector<couchbase::core::impl::subdoc::command> specs{};
+    std::optional<std::chrono::milliseconds> timeout{};
+    std::shared_ptr<couchbase::tracing::request_span> parent_span{ nullptr };
+    std::uint16_t partition{};
+    std::uint32_t opaque{};
+    io::retry_context<false> retries{};
+
+    [[nodiscard]] std::error_code encode_to(encoded_request_type& encoded, mcbp_context&& context);
+
+    [[nodiscard]] lookup_in_replica_response make_response(key_value_error_context&& ctx, const encoded_response_type& encoded) const;
+};
+} // namespace couchbase::core::impl

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

@@ -0,0 +1,100 @@
+/* -*- 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_manager_error_context.hxx"
+
+#include <couchbase/manager_error_context.hxx>
+
+namespace couchbase
+{
+manager_error_context::manager_error_context()
+  : internal_{ nullptr }
+{
+}
+
+manager_error_context::manager_error_context(internal_manager_error_context ctx)
+  : internal_{ std::make_unique<internal_manager_error_context>(std::move(ctx)) }
+{
+}
+
+manager_error_context::manager_error_context(manager_error_context&& other) = default;
+
+manager_error_context&
+manager_error_context::operator=(manager_error_context&& other) = default;
+
+manager_error_context::~manager_error_context() = default;
+
+auto
+manager_error_context::ec() const -> std::error_code
+{
+    return internal_->ec();
+}
+
+auto
+manager_error_context::last_dispatched_from() const -> const std::optional<std::string>&
+{
+    return internal_->last_dispatched_from();
+}
+
+auto
+manager_error_context::last_dispatched_to() const -> const std::optional<std::string>&
+{
+    return internal_->last_dispatched_to();
+}
+
+auto
+manager_error_context::retry_attempts() const -> std::size_t
+{
+    return internal_->retry_attempts();
+}
+
+auto
+manager_error_context::retry_reasons() const -> const std::set<retry_reason>&
+{
+    return internal_->retry_reasons();
+}
+
+auto
+manager_error_context::retried_because_of(couchbase::retry_reason reason) const -> bool
+{
+    return internal_->retried_because_of(reason);
+}
+
+auto
+manager_error_context::path() const -> const std::string&
+{
+    return internal_->path();
+}
+
+auto
+manager_error_context::content() const -> const std::string&
+{
+    return internal_->content();
+}
+
+auto
+manager_error_context::client_context_id() const -> const std::string&
+{
+    return internal_->client_context_id();
+}
+
+auto
+manager_error_context::http_status() const -> std::uint32_t
+{
+    return internal_->http_status();
+}
+} // namespace couchbase

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

@@ -161,6 +161,7 @@ build_query_request(std::string statement, query_options::built options)
         options.readonly,
         options.flex_index,
         options.preserve_expiry,
+        options.use_replica,
         options.max_parallelism,
         options.scan_cap,
         options.scan_wait,

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

@@ -0,0 +1,75 @@
+/* -*- 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 <couchbase/query_error_context.hxx>
+
+#include <couchbase/fmt/retry_reason.hxx>
+
+#include <tao/json/to_string.hpp>
+
+namespace couchbase
+{
+auto
+query_error_context::to_json() const -> std::string
+{
+    tao::json::value json = {
+        {
+          "ec",
+          tao::json::value{
+            { "value", ec().value() },
+            { "message", ec().message() },
+          },
+        },
+        { "operation_id", operation_id() },
+        { "retry_attempts", retry_attempts() },
+        { "client_context_id", client_context_id_ },
+        { "statement", statement_ },
+        { "method", statement_ },
+        { "path", statement_ },
+        { "http_status", http_status_ },
+        { "http_body", http_body_ },
+        { "hostname", hostname_ },
+        { "port", port_ },
+    };
+
+    if (const auto& val = parameters_; val.has_value()) {
+        json["parameters"] = val.value();
+    }
+    if (first_error_code_ > 0) {
+        json["first_error_code"] = first_error_code_;
+    }
+    if (!first_error_message_.empty()) {
+        json["first_error_message"] = first_error_message_;
+    }
+
+    if (const auto& reasons = retry_reasons(); !reasons.empty()) {
+        tao::json::value reasons_json = tao::json::empty_array;
+        for (const auto& reason : reasons) {
+            reasons_json.emplace_back(fmt::format("{}", reason));
+        }
+        json["retry_reasons"] = reasons_json;
+    }
+    if (const auto& val = last_dispatched_from(); val.has_value()) {
+        json["last_dispatched_from"] = val.value();
+    }
+    if (const auto& val = last_dispatched_to(); val.has_value()) {
+        json["last_dispatched_to"] = val.value();
+    }
+
+    return tao::json::to_string(json, 2);
+}
+} // namespace couchbase

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

@@ -0,0 +1,133 @@
+/* -*- 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 <couchbase/manager_error_context.hxx>
+#include <utility>
+
+#include "core/cluster.hxx"
+#include "core/operations/management/bucket_update.hxx"
+#include "couchbase/bucket_manager.hxx"
+
+namespace couchbase
+{
+template<typename Response>
+static manager_error_context
+build_context(Response& resp)
+{
+    return manager_error_context(internal_manager_error_context{ resp.ctx.ec,
+                                                                 resp.ctx.last_dispatched_to,
+                                                                 resp.ctx.last_dispatched_from,
+                                                                 resp.ctx.retry_attempts,
+                                                                 std::move(resp.ctx.retry_reasons),
+                                                                 std::move(resp.ctx.client_context_id),
+                                                                 resp.ctx.http_status,
+                                                                 std::move(resp.ctx.http_body),
+                                                                 std::move(resp.ctx.path) });
+}
+
+static core::operations::management::bucket_update_request
+build_update_bucket_request(couchbase::core::management::cluster::bucket_settings bucket_settings,
+                            const update_bucket_options::built& options)
+{
+    core::operations::management::bucket_update_request request{ std::move(bucket_settings), {}, options.timeout };
+    return request;
+}
+
+static couchbase::core::management::cluster::bucket_settings
+map_bucket_settings(const couchbase::management::cluster::bucket_settings& bucket)
+{
+    couchbase::core::management::cluster::bucket_settings bucket_settings{};
+
+    bucket_settings.name = bucket.name;
+    bucket_settings.ram_quota_mb = bucket.ram_quota_mb;
+    bucket_settings.max_expiry = bucket.max_expiry;
+    bucket_settings.minimum_durability_level = bucket.minimum_durability_level;
+    bucket_settings.num_replicas = bucket.num_replicas;
+    bucket_settings.replica_indexes = bucket.replica_indexes;
+    bucket_settings.flush_enabled = bucket.flush_enabled;
+    bucket_settings.history_retention_collection_default = bucket.history_retention_collection_default;
+    bucket_settings.history_retention_bytes = bucket.history_retention_bytes;
+    bucket_settings.history_retention_duration = bucket.history_retention_duration;
+    switch (bucket.conflict_resolution_type) {
+        case management::cluster::bucket_conflict_resolution::unknown:
+            bucket_settings.conflict_resolution_type = core::management::cluster::bucket_conflict_resolution::unknown;
+            break;
+        case management::cluster::bucket_conflict_resolution::timestamp:
+            bucket_settings.conflict_resolution_type = core::management::cluster::bucket_conflict_resolution::timestamp;
+            break;
+        case management::cluster::bucket_conflict_resolution::sequence_number:
+            bucket_settings.conflict_resolution_type = core::management::cluster::bucket_conflict_resolution::sequence_number;
+            break;
+        case management::cluster::bucket_conflict_resolution::custom:
+            bucket_settings.conflict_resolution_type = core::management::cluster::bucket_conflict_resolution::custom;
+            break;
+    }
+    switch (bucket.eviction_policy) {
+        case management::cluster::bucket_eviction_policy::unknown:
+            bucket_settings.eviction_policy = core::management::cluster::bucket_eviction_policy::unknown;
+            break;
+        case management::cluster::bucket_eviction_policy::full:
+            bucket_settings.eviction_policy = core::management::cluster::bucket_eviction_policy::full;
+            break;
+        case management::cluster::bucket_eviction_policy::value_only:
+            bucket_settings.eviction_policy = core::management::cluster::bucket_eviction_policy::value_only;
+            break;
+        case management::cluster::bucket_eviction_policy::no_eviction:
+            bucket_settings.eviction_policy = core::management::cluster::bucket_eviction_policy::no_eviction;
+            break;
+        case management::cluster::bucket_eviction_policy::not_recently_used:
+            bucket_settings.eviction_policy = core::management::cluster::bucket_eviction_policy::not_recently_used;
+            break;
+    }
+    switch (bucket.bucket_type) {
+        case management::cluster::bucket_type::unknown:
+            bucket_settings.bucket_type = core::management::cluster::bucket_type::unknown;
+            break;
+        case management::cluster::bucket_type::couchbase:
+            bucket_settings.bucket_type = core::management::cluster::bucket_type::couchbase;
+            break;
+        case management::cluster::bucket_type::memcached:
+            bucket_settings.bucket_type = core::management::cluster::bucket_type::memcached;
+            break;
+        case management::cluster::bucket_type::ephemeral:
+            bucket_settings.bucket_type = core::management::cluster::bucket_type::ephemeral;
+            break;
+    }
+    return bucket_settings;
+}
+
+void
+bucket_manager::update_bucket(const management::cluster::bucket_settings& bucket_settings,
+                              const update_bucket_options& options,
+                              update_bucket_handler&& handler) const
+{
+    auto request = build_update_bucket_request(map_bucket_settings(bucket_settings), options.build());
+
+    core_->execute(std::move(request), [handler = std::move(handler)](core::operations::management::bucket_update_response resp) mutable {
+        return handler(build_context(resp));
+    });
+}
+
+auto
+bucket_manager::update_bucket(const management::cluster::bucket_settings& bucket_settings, const update_bucket_options& options) const
+  -> std::future<manager_error_context>
+{
+    auto barrier = std::make_shared<std::promise<manager_error_context>>();
+    update_bucket(bucket_settings, options, [barrier](auto ctx) mutable { barrier->set_value(std::move(ctx)); });
+    return barrier->get_future();
+}
+} // namespace couchbase

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

@@ -0,0 +1,83 @@
+/* -*- 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 <couchbase/manager_error_context.hxx>
+
+#include "core/cluster.hxx"
+#include "core/operations/management/collection_update.hxx"
+#include "couchbase/collection_manager.hxx"
+
+namespace couchbase
+{
+template<typename Response>
+static manager_error_context
+build_context(Response& resp)
+{
+    return manager_error_context(internal_manager_error_context{ resp.ctx.ec,
+                                                                 resp.ctx.last_dispatched_to,
+                                                                 resp.ctx.last_dispatched_from,
+                                                                 resp.ctx.retry_attempts,
+                                                                 std::move(resp.ctx.retry_reasons),
+                                                                 std::move(resp.ctx.client_context_id),
+                                                                 resp.ctx.http_status,
+                                                                 std::move(resp.ctx.http_body),
+                                                                 std::move(resp.ctx.path) });
+}
+
+static core::operations::management::collection_update_request
+build_collection_update_request(std::string bucket_name,
+                                std::string scope_name,
+                                std::string collection_name,
+                                const update_collection_settings& settings,
+                                const update_collection_options::built& options)
+{
+    core::operations::management::collection_update_request request{
+        std::move(bucket_name), std::move(scope_name), std::move(collection_name), settings.max_expiry, settings.history, {},
+        options.timeout
+    };
+    return request;
+}
+
+void
+collection_manager::update_collection(std::string scope_name,
+                                      std::string collection_name,
+                                      const couchbase::update_collection_settings& settings,
+                                      const couchbase::update_collection_options& options,
+                                      couchbase::update_collection_handler&& handler) const
+{
+    auto request =
+      build_collection_update_request(bucket_name_, std::move(scope_name), std::move(collection_name), settings, options.build());
+
+    core_->execute(std::move(request),
+                   [handler = std::move(handler)](core::operations::management::collection_update_response resp) mutable {
+                       return handler(build_context(resp));
+                   });
+}
+
+auto
+collection_manager::update_collection(std::string scope_name,
+                                      std::string collection_name,
+                                      const couchbase::update_collection_settings& settings,
+                                      const couchbase::update_collection_options& options) const -> std::future<manager_error_context>
+{
+    auto barrier = std::make_shared<std::promise<manager_error_context>>();
+    update_collection(std::move(scope_name), std::move(collection_name), settings, options, [barrier](auto ctx) mutable {
+        barrier->set_value(std::move(ctx));
+    });
+    return barrier->get_future();
+}
+} // namespace couchbase

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

@@ -24,16 +24,6 @@
 
 namespace couchbase::core::impl
 {
-
-template<typename Response>
-static manager_error_context
-build_context(Response& resp, std::optional<std::error_code> ec = {})
-{
-    return { ec ? ec.value() : resp.ctx.ec, resp.ctx.last_dispatched_to,       resp.ctx.last_dispatched_from,
-             resp.ctx.retry_attempts,       std::move(resp.ctx.retry_reasons), std::move(resp.ctx.client_context_id),
-             resp.ctx.http_status,          std::move(resp.ctx.http_body),     std::move(resp.ctx.path) };
-}
-
 class watch_context : public std::enable_shared_from_this<watch_context>
 {
 
@@ -50,9 +40,18 @@ class watch_context : public std::enable_shared_from_this<watch_context>
     std::chrono::milliseconds timeout_{ options_.timeout.value_or(core_->origin().second.options().query_timeout) };
     std::atomic<size_t> attempts_{ 0 };
 
-    void finish(manager_error_context error_ctx)
+    template<typename Response>
+    void finish(Response& resp, std::optional<std::error_code> ec = {})
     {
-        handler_(error_ctx);
+        handler_({ manager_error_context(internal_manager_error_context{ ec ? ec.value() : resp.ctx.ec,
+                                                                         resp.ctx.last_dispatched_to,
+                                                                         resp.ctx.last_dispatched_from,
+                                                                         resp.ctx.retry_attempts,
+                                                                         std::move(resp.ctx.retry_reasons),
+                                                                         std::move(resp.ctx.client_context_id),
+                                                                         resp.ctx.http_status,
+                                                                         std::move(resp.ctx.http_body),
+                                                                         std::move(resp.ctx.path) }) });
         timer_.cancel();
     }
     std::chrono::milliseconds remaining()
@@ -79,9 +78,9 @@ class watch_context : public std::enable_shared_from_this<watch_context>
             complete &= it != resp.indexes.end() && it->state == "online";
         }
         if (complete || resp.ctx.ec == couchbase::errc::common::ambiguous_timeout) {
-            finish(build_context(resp));
+            finish(resp);
         } else if (remaining().count() <= 0) {
-            finish(build_context(resp, couchbase::errc::common::ambiguous_timeout));
+            finish(resp, couchbase::errc::common::ambiguous_timeout);
             complete = true;
         }
         return complete;
@@ -141,28 +140,53 @@ class watch_context : public std::enable_shared_from_this<watch_context>
         core_->execute(req, resp_fn);
     }
 };
+} // namespace couchbase::core::impl
 
+namespace couchbase
+{
 void
-initiate_watch_query_indexes(std::shared_ptr<couchbase::core::cluster> core,
-                             std::string bucket_name,
-                             std::vector<std::string> index_names,
-                             couchbase::watch_query_indexes_options::built options,
-                             query_context query_ctx,
-                             std::string collection_name,
-                             watch_query_indexes_handler&& handler)
+query_index_manager::watch_indexes(std::string bucket_name,
+                                   std::vector<std::string> index_names,
+                                   const couchbase::watch_query_indexes_options& options,
+                                   couchbase::watch_query_indexes_handler&& handler)
 {
-    auto ctx = std::make_shared<watch_context>(core, bucket_name, index_names, options, query_ctx, collection_name, std::move(handler));
+    auto ctx = std::make_shared<couchbase::core::impl::watch_context>(
+      core_, std::move(bucket_name), std::move(index_names), options.build(), core::query_context{}, "", std::move(handler));
     ctx->execute();
 }
 
+auto
+query_index_manager::watch_indexes(std::string bucket_name,
+                                   std::vector<std::string> index_names,
+                                   const couchbase::watch_query_indexes_options& options) -> std::future<manager_error_context>
+{
+    auto barrier = std::make_shared<std::promise<manager_error_context>>();
+    watch_indexes(
+      std::move(bucket_name), std::move(index_names), options, [barrier](auto ctx) mutable { barrier->set_value(std::move(ctx)); });
+    return barrier->get_future();
+}
+
 void
-initiate_watch_query_indexes(std::shared_ptr<couchbase::core::cluster> core,
-                             std::string bucket_name,
-                             std::vector<std::string> index_names,
-                             couchbase::watch_query_indexes_options::built options,
-                             watch_query_indexes_handler&& handler)
+collection_query_index_manager::watch_indexes(std::vector<std::string> index_names,
+                                              const watch_query_indexes_options& options,
+                                              watch_query_indexes_handler&& handler) const
 {
-    initiate_watch_query_indexes(core, std::move(bucket_name), std::move(index_names), options, {}, "", std::move(handler));
+    auto ctx = std::make_shared<couchbase::core::impl::watch_context>(core_,
+                                                                      bucket_name_,
+                                                                      std::move(index_names),
+                                                                      options.build(),
+                                                                      core::query_context(bucket_name_, scope_name_),
+                                                                      collection_name_,
+                                                                      std::move(handler));
+    ctx->execute();
 }
 
-} // namespace couchbase::core::impl
+auto
+collection_query_index_manager::watch_indexes(std::vector<std::string> index_names, const couchbase::watch_query_indexes_options& options)
+  -> std::future<manager_error_context>
+{
+    auto barrier = std::make_shared<std::promise<manager_error_context>>();
+    watch_indexes(std::move(index_names), options, [barrier](auto ctx) mutable { barrier->set_value(std::move(ctx)); });
+    return barrier->get_future();
+}
+} // namespace couchbase