couchbase 3.0.0.alpha.2-universal-darwin-19 → 3.0.0.alpha.3-universal-darwin-19

data/ext/couchbase/operations/document_touch.hxx

@@ -37,6 +37,7 @@ struct touch_request {
     std::uint16_t partition{};
     std::uint32_t opaque{};
     std::uint32_t expiration{};
+    std::chrono::milliseconds timeout{ timeout_defaults::key_value_timeout };
 
     void encode_to(encoded_request_type& encoded)
     {

data/ext/couchbase/operations/document_unlock.hxx

@@ -37,6 +37,7 @@ struct unlock_request {
     uint16_t partition{};
     uint32_t opaque{};
     uint64_t cas{};
+    std::chrono::milliseconds timeout{ timeout_defaults::key_value_timeout };
 
     void encode_to(encoded_request_type& encoded)
     {

data/ext/couchbase/operations/document_upsert.hxx

@@ -43,6 +43,7 @@ struct upsert_request {
     uint32_t expiration{ 0 };
     protocol::durability_level durability_level{ protocol::durability_level::none };
     std::optional<std::uint16_t> durability_timeout{};
+    std::chrono::milliseconds timeout{ timeout_defaults::key_value_timeout };
 
     void encode_to(encoded_request_type& encoded)
     {

data/ext/couchbase/operations/query_index_build_deferred.hxx

@@ -43,6 +43,7 @@ struct query_index_build_deferred_request {
 
     uuid::uuid_t client_context_id{ uuid::random() };
     std::string bucket_name;
+    std::chrono::milliseconds timeout{ timeout_defaults::management_timeout };
 
     void encode_to(encoded_request_type& encoded)
     {

data/ext/couchbase/operations/query_index_create.hxx

@@ -50,6 +50,7 @@ struct query_index_create_request {
     std::optional<std::string> condition{};
     std::optional<bool> deferred{};
     std::optional<int> num_replicas{};
+    std::chrono::milliseconds timeout{ timeout_defaults::management_timeout };
 
     void encode_to(encoded_request_type& encoded)
     {

data/ext/couchbase/operations/query_index_drop.hxx

@@ -46,6 +46,7 @@ struct query_index_drop_request {
     std::string index_name;
     bool is_primary{ false };
     bool ignore_if_does_not_exist{ false };
+    std::chrono::milliseconds timeout{ timeout_defaults::management_timeout };
 
     void encode_to(encoded_request_type& encoded)
     {

data/ext/couchbase/operations/query_index_get_all.hxx

@@ -51,6 +51,7 @@ struct query_index_get_all_request {
 
     uuid::uuid_t client_context_id{ uuid::random() };
     std::string bucket_name;
+    std::chrono::milliseconds timeout{ timeout_defaults::management_timeout };
 
     void encode_to(encoded_request_type& encoded)
     {

data/ext/couchbase/operations/scope_create.hxx

@@ -40,6 +40,7 @@ struct scope_create_request {
 
     std::string bucket_name;
     std::string scope_name;
+    std::chrono::milliseconds timeout{ timeout_defaults::management_timeout };
 
     void encode_to(encoded_request_type& encoded)
     {

data/ext/couchbase/operations/scope_drop.hxx

@@ -40,6 +40,7 @@ struct scope_drop_request {
 
     std::string bucket_name;
     std::string scope_name;
+    std::chrono::milliseconds timeout{ timeout_defaults::management_timeout };
 
     void encode_to(encoded_request_type& encoded)
     {

data/ext/couchbase/operations/scope_get_all.hxx

@@ -21,6 +21,7 @@
 
 #include <version.hxx>
 #include <operations/bucket_settings.hxx>
+#include <collections_manifest.hxx>
 
 namespace couchbase::operations
 {
@@ -38,6 +39,7 @@ struct scope_get_all_request {
     static const inline service_type type = service_type::management;
 
     std::string bucket_name;
+    std::chrono::milliseconds timeout{ timeout_defaults::management_timeout };
 
     void encode_to(encoded_request_type& encoded)
     {

data/ext/couchbase/operations/search_index.hxx

@@ -0,0 +1,62 @@
+#pragma once
+
+namespace couchbase::operations
+{
+struct search_index {
+    std::string uuid;
+    std::string name;
+    std::string type;
+    std::string params_json;
+
+    std::string source_uuid;
+    std::string source_name;
+    std::string source_type;
+    std::string source_params_json;
+
+    std::string plan_params_json;
+};
+
+} // namespace couchbase::operations
+
+namespace tao::json
+{
+template<>
+struct traits<couchbase::operations::search_index> {
+    template<template<typename...> class Traits>
+    static couchbase::operations::search_index as(const tao::json::basic_value<Traits>& v)
+    {
+        couchbase::operations::search_index result;
+        result.uuid = v.at("uuid").get_string();
+        result.name = v.at("name").get_string();
+        result.type = v.at("type").get_string();
+        {
+            const auto* params = v.find("params");
+            if (params != nullptr && params->is_object()) {
+                result.params_json = tao::json::to_string(*params);
+            }
+        }
+        if (v.find("sourceUUID") != nullptr) {
+            result.source_uuid = v.at("sourceUUID").get_string();
+        }
+        if (v.find("sourceName") != nullptr) {
+            result.source_name = v.at("sourceName").get_string();
+        }
+        if (v.find("sourceType") != nullptr) {
+            result.source_type = v.at("sourceType").get_string();
+        }
+        {
+            const auto* params = v.find("sourceParams");
+            if (params != nullptr && params->is_object()) {
+                result.source_params_json = tao::json::to_string(*params);
+            }
+        }
+        {
+            const auto* params = v.find("planParams");
+            if (params != nullptr && params->is_object()) {
+                result.plan_params_json = tao::json::to_string(*params);
+            }
+        }
+        return result;
+    }
+};
+} // namespace tao::json

data/ext/couchbase/operations/search_index_analyze_document.hxx

@@ -0,0 +1,92 @@
+/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *     Copyright 2020 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.hpp>
+
+#include <version.hxx>
+
+namespace couchbase::operations
+{
+struct search_index_analyze_document_response {
+    uuid::uuid_t client_context_id;
+    std::error_code ec;
+    std::string status{};
+    std::string error{};
+    std::string analysis{};
+};
+
+struct search_index_analyze_document_request {
+    using response_type = search_index_analyze_document_response;
+    using encoded_request_type = io::http_request;
+    using encoded_response_type = io::http_response;
+
+    static const inline service_type type = service_type::search;
+
+    uuid::uuid_t client_context_id{ uuid::random() };
+    std::chrono::milliseconds timeout{ timeout_defaults::management_timeout };
+
+    std::string index_name;
+    std::string encoded_document;
+
+    void encode_to(encoded_request_type& encoded)
+    {
+        encoded.method = "POST";
+        encoded.headers["cache-control"] = "no-cache";
+        encoded.headers["content-type"] = "application/json";
+        encoded.path = fmt::format("/api/index/{}/analyzeDoc", index_name);
+        encoded.body = encoded_document;
+    }
+};
+
+search_index_analyze_document_response
+make_response(std::error_code ec,
+              search_index_analyze_document_request& request,
+              search_index_analyze_document_request::encoded_response_type encoded)
+{
+    search_index_analyze_document_response response{ request.client_context_id, ec };
+    if (!ec) {
+        if (encoded.status_code == 200) {
+            auto payload = tao::json::from_string(encoded.body);
+            response.status = payload.at("status").get_string();
+            if (response.status == "ok") {
+                response.analysis = tao::json::to_string(payload.at("analyzed"));
+                return response;
+            }
+        } else if (encoded.status_code == 400) {
+            if (encoded.body.find("no indexName:") != std::string::npos) {
+                response.ec = std::make_error_code(error::common_errc::index_not_found);
+                return response;
+            }
+            auto payload = tao::json::from_string(encoded.body);
+            response.status = payload.at("status").get_string();
+            response.error = payload.at("error").get_string();
+            if (response.error.find("index not found") != std::string::npos) {
+                response.ec = std::make_error_code(error::common_errc::index_not_found);
+                return response;
+            } else if (response.error.find("index with the same name already exists") != std::string::npos) {
+                response.ec = std::make_error_code(error::common_errc::index_exists);
+                return response;
+            }
+        }
+        response.ec = std::make_error_code(error::common_errc::internal_server_failure);
+    }
+    return response;
+}
+
+} // namespace couchbase::operations

data/ext/couchbase/operations/search_index_control_ingest.hxx

@@ -0,0 +1,78 @@
+/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *     Copyright 2020 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.hpp>
+
+#include <version.hxx>
+
+namespace couchbase::operations
+{
+struct search_index_control_ingest_response {
+    uuid::uuid_t client_context_id;
+    std::error_code ec;
+    std::string status{};
+    std::string error{};
+};
+
+struct search_index_control_ingest_request {
+    using response_type = search_index_control_ingest_response;
+    using encoded_request_type = io::http_request;
+    using encoded_response_type = io::http_response;
+
+    static const inline service_type type = service_type::search;
+
+    uuid::uuid_t client_context_id{ uuid::random() };
+    std::chrono::milliseconds timeout{ timeout_defaults::management_timeout };
+
+    std::string index_name;
+    bool pause;
+
+    void encode_to(encoded_request_type& encoded)
+    {
+        encoded.method = "POST";
+        encoded.path = fmt::format("/api/index/{}/ingestControl/{}", index_name, pause ? "pause" : "resume");
+    }
+};
+
+search_index_control_ingest_response
+make_response(std::error_code ec, search_index_control_ingest_request& request, search_index_control_ingest_request::encoded_response_type encoded)
+{
+    search_index_control_ingest_response response{ request.client_context_id, ec };
+    if (!ec) {
+        if (encoded.status_code == 200) {
+            auto payload = tao::json::from_string(encoded.body);
+            response.status = payload.at("status").get_string();
+            if (response.status == "ok") {
+                return response;
+            }
+        } else if (encoded.status_code == 400) {
+            auto payload = tao::json::from_string(encoded.body);
+            response.status = payload.at("status").get_string();
+            response.error = payload.at("error").get_string();
+            if (response.error.find("index not found") != std::string::npos) {
+                response.ec = std::make_error_code(error::common_errc::index_not_found);
+                return response;
+            }
+        }
+        response.ec = std::make_error_code(error::common_errc::internal_server_failure);
+    }
+    return response;
+}
+
+} // namespace couchbase::operations

data/ext/couchbase/operations/search_index_control_plan_freeze.hxx

@@ -0,0 +1,80 @@
+/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *     Copyright 2020 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.hpp>
+
+#include <version.hxx>
+
+namespace couchbase::operations
+{
+struct search_index_control_plan_freeze_response {
+    uuid::uuid_t client_context_id;
+    std::error_code ec;
+    std::string status{};
+    std::string error{};
+};
+
+struct search_index_control_plan_freeze_request {
+    using response_type = search_index_control_plan_freeze_response;
+    using encoded_request_type = io::http_request;
+    using encoded_response_type = io::http_response;
+
+    static const inline service_type type = service_type::search;
+
+    uuid::uuid_t client_context_id{ uuid::random() };
+    std::chrono::milliseconds timeout{ timeout_defaults::management_timeout };
+
+    std::string index_name;
+    bool freeze;
+
+    void encode_to(encoded_request_type& encoded)
+    {
+        encoded.method = "POST";
+        encoded.path = fmt::format("/api/index/{}/planFreezeControl/{}", index_name, freeze ? "freeze" : "unfreeze");
+    }
+};
+
+search_index_control_plan_freeze_response
+make_response(std::error_code ec,
+              search_index_control_plan_freeze_request& request,
+              search_index_control_plan_freeze_request::encoded_response_type encoded)
+{
+    search_index_control_plan_freeze_response response{ request.client_context_id, ec };
+    if (!ec) {
+        if (encoded.status_code == 200) {
+            auto payload = tao::json::from_string(encoded.body);
+            response.status = payload.at("status").get_string();
+            if (response.status == "ok") {
+                return response;
+            }
+        } else if (encoded.status_code == 400) {
+            auto payload = tao::json::from_string(encoded.body);
+            response.status = payload.at("status").get_string();
+            response.error = payload.at("error").get_string();
+            if (response.error.find("index not found") != std::string::npos) {
+                response.ec = std::make_error_code(error::common_errc::index_not_found);
+                return response;
+            }
+        }
+        response.ec = std::make_error_code(error::common_errc::internal_server_failure);
+    }
+    return response;
+}
+
+} // namespace couchbase::operations

data/ext/couchbase/operations/search_index_control_query.hxx

@@ -0,0 +1,80 @@
+/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *     Copyright 2020 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.hpp>
+
+#include <version.hxx>
+
+namespace couchbase::operations
+{
+struct search_index_control_query_response {
+    uuid::uuid_t client_context_id;
+    std::error_code ec;
+    std::string status{};
+    std::string error{};
+};
+
+struct search_index_control_query_request {
+    using response_type = search_index_control_query_response;
+    using encoded_request_type = io::http_request;
+    using encoded_response_type = io::http_response;
+
+    static const inline service_type type = service_type::search;
+
+    uuid::uuid_t client_context_id{ uuid::random() };
+    std::chrono::milliseconds timeout{ timeout_defaults::management_timeout };
+
+    std::string index_name;
+    bool allow;
+
+    void encode_to(encoded_request_type& encoded)
+    {
+        encoded.method = "POST";
+        encoded.path = fmt::format("/api/index/{}/queryControl/{}", index_name, allow ? "allow" : "disallow");
+    }
+};
+
+search_index_control_query_response
+make_response(std::error_code ec,
+              search_index_control_query_request& request,
+              search_index_control_query_request::encoded_response_type encoded)
+{
+    search_index_control_query_response response{ request.client_context_id, ec };
+    if (!ec) {
+        if (encoded.status_code == 200) {
+            auto payload = tao::json::from_string(encoded.body);
+            response.status = payload.at("status").get_string();
+            if (response.status == "ok") {
+                return response;
+            }
+        } else if (encoded.status_code == 400) {
+            auto payload = tao::json::from_string(encoded.body);
+            response.status = payload.at("status").get_string();
+            response.error = payload.at("error").get_string();
+            if (response.error.find("index not found") != std::string::npos) {
+                response.ec = std::make_error_code(error::common_errc::index_not_found);
+                return response;
+            }
+        }
+        response.ec = std::make_error_code(error::common_errc::internal_server_failure);
+    }
+    return response;
+}
+
+} // namespace couchbase::operations