couchbase 3.4.2 → 3.4.4

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

@@ -0,0 +1,66 @@
+/* -*- 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_drop.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_drop_request
+build_drop_bucket_request(std::string bucket_name, const drop_bucket_options::built& options)
+{
+    core::operations::management::bucket_drop_request request{ std::move(bucket_name), {}, options.timeout };
+    return request;
+}
+
+void
+bucket_manager::drop_bucket(std::string bucket_name, const drop_bucket_options& options, drop_bucket_handler&& handler) const
+{
+    auto request = build_drop_bucket_request(std::move(bucket_name), options.build());
+
+    core_->execute(std::move(request), [handler = std::move(handler)](core::operations::management::bucket_drop_response resp) mutable {
+        return handler(build_context(resp));
+    });
+}
+
+auto
+bucket_manager::drop_bucket(std::string bucket_name, const drop_bucket_options& options) const -> std::future<manager_error_context>
+{
+    auto barrier = std::make_shared<std::promise<manager_error_context>>();
+    drop_bucket(std::move(bucket_name), options, [barrier](auto ctx) mutable { barrier->set_value(std::move(ctx)); });
+    return barrier->get_future();
+}
+} // namespace couchbase

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

@@ -21,88 +21,167 @@
 #include "core/cluster.hxx"
 #include "core/operations/management/query_index_drop.hxx"
 
-namespace couchbase::core::impl
+namespace couchbase
 {
 template<typename Response>
 static manager_error_context
 build_context(Response& resp)
 {
-    return { 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) };
+    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) });
 }
 
-void
-initiate_drop_query_index(std::shared_ptr<couchbase::core::cluster> core,
-                          std::string bucket_name,
-                          std::string index_name,
-                          drop_query_index_options::built options,
-                          query_context query_context,
-                          std::string collection_name,
-                          drop_query_index_handler&& handler)
+static core::operations::management::query_index_drop_request
+build_drop_primary_index_request(std::string bucket_name, const drop_primary_query_index_options::built& options)
+{
+    core::operations::management::query_index_drop_request request{
+        std::move(bucket_name), "", "", options.index_name.value_or(""), {}, true, options.ignore_if_not_exists, {}, options.timeout,
+    };
+    return request;
+}
+
+static core::operations::management::query_index_drop_request
+build_drop_index_request(std::string bucket_name, std::string index_name, const drop_query_index_options::built& options)
+{
+    core::operations::management::query_index_drop_request request{
+        std::move(bucket_name), "", "", std::move(index_name), {}, false, options.ignore_if_not_exists, {}, options.timeout,
+    };
+    return request;
+}
+
+static core::operations::management::query_index_drop_request
+build_drop_primary_index_request(std::string bucket_name,
+                                 std::string scope_name,
+                                 std::string collection_name,
+                                 const drop_primary_query_index_options::built& options)
+{
+    core::operations::management::query_index_drop_request request{
+        "",
+        "",
+        std::move(collection_name),
+        options.index_name.value_or(""),
+        core::query_context(std::move(bucket_name), std::move(scope_name)),
+        true,
+        options.ignore_if_not_exists,
+        {},
+        options.timeout,
+    };
+    return request;
+}
+
+static core::operations::management::query_index_drop_request
+build_drop_index_request(std::string bucket_name,
+                         std::string scope_name,
+                         std::string collection_name,
+                         std::string index_name,
+                         const drop_query_index_options::built& options)
 {
-    core->execute(
-      operations::management::query_index_drop_request{
-        bucket_name,
+    core::operations::management::query_index_drop_request request{
         "",
-        collection_name,
-        index_name,
-        query_context,
+        "",
+        std::move(collection_name),
+        std::move(index_name),
+        core::query_context(std::move(bucket_name), std::move(scope_name)),
         false,
         options.ignore_if_not_exists,
         {},
         options.timeout,
-      },
-      [core, bucket_name, options = std::move(options), handler = std::move(handler)](
-        operations::management::query_index_drop_response resp) { handler(build_context(resp)); });
+    };
+    return request;
+}
+
+void
+query_index_manager::drop_primary_index(std::string bucket_name,
+                                        const couchbase::drop_primary_query_index_options& options,
+                                        couchbase::drop_query_index_handler&& handler)
+{
+    auto request = build_drop_primary_index_request(std::move(bucket_name), options.build());
+
+    core_->execute(std::move(request),
+                   [handler = std::move(handler)](core::operations::management::query_index_drop_response resp) mutable {
+                       return handler(build_context(resp));
+                   });
+}
+
+auto
+query_index_manager::drop_primary_index(std::string bucket_name, const couchbase::drop_primary_query_index_options& options)
+  -> std::future<manager_error_context>
+{
+    auto barrier = std::make_shared<std::promise<manager_error_context>>();
+    drop_primary_index(std::move(bucket_name), options, [barrier](auto ctx) mutable { barrier->set_value(std::move(ctx)); });
+    return barrier->get_future();
 }
 
 void
-initiate_drop_query_index(std::shared_ptr<couchbase::core::cluster> core,
-                          std::string bucket_name,
-                          std::string index_name,
-                          couchbase::drop_query_index_options::built options,
-                          drop_query_index_handler&& handler)
+query_index_manager::drop_index(std::string bucket_name,
+                                std::string index_name,
+                                const couchbase::drop_query_index_options& options,
+                                couchbase::drop_query_index_handler&& handler)
 {
-    initiate_drop_query_index(core, std::move(bucket_name), std::move(index_name), options, {}, "", std::move(handler));
+    auto request = build_drop_index_request(std::move(bucket_name), std::move(index_name), options.build());
+
+    core_->execute(std::move(request),
+                   [handler = std::move(handler)](core::operations::management::query_index_drop_response resp) mutable {
+                       return handler(build_context(resp));
+                   });
+}
+
+auto
+query_index_manager::drop_index(std::string bucket_name, std::string index_name, const couchbase::drop_query_index_options& options)
+  -> std::future<manager_error_context>
+{
+    auto barrier = std::make_shared<std::promise<manager_error_context>>();
+    drop_index(std::move(bucket_name), std::move(index_name), options, [barrier](auto ctx) mutable { barrier->set_value(std::move(ctx)); });
+    return barrier->get_future();
 }
 
 void
-initiate_drop_primary_query_index(std::shared_ptr<couchbase::core::cluster> core,
-                                  std::string bucket_name,
-                                  couchbase::drop_primary_query_index_options::built options,
-                                  query_context query_ctx,
-                                  std::string collection_name,
-                                  drop_primary_query_index_handler&& handler)
+collection_query_index_manager::drop_primary_index(const drop_primary_query_index_options& options,
+                                                   drop_query_index_handler&& handler) const
 {
-    core->execute(
-      operations::management::query_index_drop_request{
-        bucket_name,
-        "",
-        collection_name,
-        options.index_name.value_or(""),
-        query_ctx,
-        true,
-        options.ignore_if_not_exists,
-        {},
-        options.timeout,
-      },
-      [core, bucket_name, options = std::move(options), handler = std::move(handler)](
-        operations::management::query_index_drop_response resp) { handler(build_context(resp)); });
+    auto request = build_drop_primary_index_request(bucket_name_, scope_name_, collection_name_, options.build());
+
+    core_->execute(std::move(request),
+                   [handler = std::move(handler)](core::operations::management::query_index_drop_response resp) mutable {
+                       return handler(build_context(resp));
+                   });
 }
+
+auto
+collection_query_index_manager::drop_primary_index(const couchbase::drop_primary_query_index_options& options) const
+  -> std::future<manager_error_context>
+{
+    auto barrier = std::make_shared<std::promise<manager_error_context>>();
+    drop_primary_index(options, [barrier](auto ctx) mutable { barrier->set_value(std::move(ctx)); });
+    return barrier->get_future();
+}
+
 void
-initiate_drop_primary_query_index(std::shared_ptr<couchbase::core::cluster> core,
-                                  std::string bucket_name,
-                                  couchbase::drop_primary_query_index_options::built options,
-                                  drop_primary_query_index_handler&& handler)
+collection_query_index_manager::drop_index(std::string index_name,
+                                           const drop_query_index_options& options,
+                                           drop_query_index_handler&& handler) const
 {
-    return initiate_drop_primary_query_index(core, std::move(bucket_name), options, {}, "", std::move(handler));
+    auto request = build_drop_index_request(bucket_name_, scope_name_, collection_name_, std::move(index_name), options.build());
+
+    core_->execute(std::move(request),
+                   [handler = std::move(handler)](core::operations::management::query_index_drop_response resp) mutable {
+                       return handler(build_context(resp));
+                   });
 }
 
-} // namespace couchbase::core::impl
+auto
+collection_query_index_manager::drop_index(std::string index_name, const couchbase::drop_query_index_options& options)
+  -> std::future<manager_error_context>
+{
+    auto barrier = std::make_shared<std::promise<manager_error_context>>();
+    drop_index(std::move(index_name), options, [barrier](auto ctx) mutable { barrier->set_value(std::move(ctx)); });
+    return barrier->get_future();
+}
+} // namespace couchbase

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/flush_bucket.cxx

@@ -0,0 +1,66 @@
+/* -*- 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_flush.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_flush_request
+build_flush_bucket_request(std::string bucket_name, const flush_bucket_options::built& options)
+{
+    core::operations::management::bucket_flush_request request{ std::move(bucket_name), {}, options.timeout };
+    return request;
+}
+
+void
+bucket_manager::flush_bucket(std::string bucket_name, const flush_bucket_options& options, flush_bucket_handler&& handler) const
+{
+    auto request = build_flush_bucket_request(std::move(bucket_name), options.build());
+
+    core_->execute(std::move(request), [handler = std::move(handler)](core::operations::management::bucket_flush_response resp) mutable {
+        return handler(build_context(resp));
+    });
+}
+
+auto
+bucket_manager::flush_bucket(std::string bucket_name, const flush_bucket_options& options) const -> std::future<manager_error_context>
+{
+    auto barrier = std::make_shared<std::promise<manager_error_context>>();
+    flush_bucket(std::move(bucket_name), options, [barrier](auto ctx) mutable { barrier->set_value(std::move(ctx)); });
+    return barrier->get_future();
+}
+} // 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