couchbase 3.0.0.alpha.2 → 3.0.0.alpha.3

data/ext/couchbase/operations/search_index_drop.hxx

@@ -0,0 +1,77 @@
+/* -*- 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_drop_response {
+    uuid::uuid_t client_context_id;
+    std::error_code ec;
+    std::string status{};
+    std::string error{};
+};
+
+struct search_index_drop_request {
+    using response_type = search_index_drop_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;
+
+    void encode_to(encoded_request_type& encoded)
+    {
+        encoded.method = "DELETE";
+        encoded.path = fmt::format("/api/index/{}", index_name);
+    }
+};
+
+search_index_drop_response
+make_response(std::error_code ec, search_index_drop_request& request, search_index_drop_request::encoded_response_type encoded)
+{
+    search_index_drop_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_get.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 <operations/search_index.hxx>
+
+#include <version.hxx>
+
+namespace couchbase::operations
+{
+struct search_index_get_response {
+    uuid::uuid_t client_context_id;
+    std::error_code ec;
+    std::string status{};
+    search_index index{};
+    std::string error{};
+};
+
+struct search_index_get_request {
+    using response_type = search_index_get_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;
+
+    void encode_to(encoded_request_type& encoded)
+    {
+        encoded.method = "GET";
+        encoded.path = fmt::format("/api/index/{}", index_name);
+    }
+};
+
+search_index_get_response
+make_response(std::error_code ec, search_index_get_request& request, search_index_get_request::encoded_response_type encoded)
+{
+    search_index_get_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.index = payload.at("indexDef").as<search_index>();
+                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_get_all.hxx

@@ -0,0 +1,82 @@
+/* -*- 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 <operations/search_index.hxx>
+
+#include <version.hxx>
+
+namespace couchbase::operations
+{
+struct search_index_get_all_response {
+    uuid::uuid_t client_context_id;
+    std::error_code ec;
+    std::string status{};
+    std::string impl_version{};
+    std::vector<search_index> indexes{};
+};
+
+struct search_index_get_all_request {
+    using response_type = search_index_get_all_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;
+
+    void encode_to(encoded_request_type& encoded)
+    {
+        encoded.method = "GET";
+        encoded.path = fmt::format("/api/index");
+    }
+};
+
+search_index_get_all_response
+make_response(std::error_code ec, search_index_get_all_request& request, search_index_get_all_request::encoded_response_type encoded)
+{
+    search_index_get_all_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") {
+                const auto* indexDefs = payload.find("indexDefs");
+                if (indexDefs != nullptr && indexDefs->is_object()) {
+                    const auto* impl_ver = indexDefs->find("implVersion");
+                    if (impl_ver != nullptr && impl_ver->is_string()) {
+                        response.impl_version = impl_ver->get_string();
+                    }
+                    const auto* indexes = indexDefs->find("indexDefs");
+                    for (const auto& entry : indexes->get_object()) {
+                        response.indexes.emplace_back(entry.second.as<search_index>());
+                    }
+                }
+                return response;
+            }
+        }
+        return response;
+    }
+    return response;
+}
+
+} // namespace couchbase::operations

data/ext/couchbase/operations/search_index_get_documents_count.hxx

@@ -0,0 +1,81 @@
+/* -*- 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_get_documents_count_response {
+    uuid::uuid_t client_context_id;
+    std::error_code ec;
+    std::string status{};
+    std::uint64_t count{ 0 };
+    std::string error{};
+};
+
+struct search_index_get_documents_count_request {
+    using response_type = search_index_get_documents_count_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;
+
+    void encode_to(encoded_request_type& encoded)
+    {
+        encoded.method = "GET";
+        encoded.path = fmt::format("/api/index/{}/count", index_name);
+    }
+};
+
+search_index_get_documents_count_response
+make_response(std::error_code ec,
+              search_index_get_documents_count_request& request,
+              search_index_get_documents_count_request::encoded_response_type encoded)
+{
+    search_index_get_documents_count_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.count = payload.at("count").get_unsigned();
+                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_upsert.hxx

@@ -0,0 +1,106 @@
+/* -*- 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_upsert_response {
+    uuid::uuid_t client_context_id;
+    std::error_code ec;
+    std::string status{};
+    std::string error{};
+};
+
+struct search_index_upsert_request {
+    using response_type = search_index_upsert_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 };
+
+    search_index index;
+
+    void encode_to(encoded_request_type& encoded)
+    {
+        encoded.method = "PUT";
+        encoded.headers["cache-control"] = "no-cache";
+        encoded.headers["content-type"] = "application/json";
+        encoded.path = fmt::format("/api/index/{}", index.name);
+        tao::json::value body{
+            { "name", index.name },
+            { "type", index.type },
+            { "sourceType", index.source_type },
+        };
+        if (!index.uuid.empty()) {
+            body["uuid"] = index.uuid;
+        }
+        if (!index.params_json.empty()) {
+            body["params"] = tao::json::from_string(index.params_json);
+        }
+        if (!index.source_name.empty()) {
+            body["sourceName"] = index.source_name;
+        }
+        if (!index.source_uuid.empty()) {
+            body["sourceUUID"] = index.source_uuid;
+        }
+        if (!index.source_params_json.empty()) {
+            body["sourceParams"] = tao::json::from_string(index.source_params_json);
+        }
+        if (!index.plan_params_json.empty()) {
+            body["planParams"] = tao::json::from_string(index.plan_params_json);
+        }
+        encoded.body = tao::json::to_string(body);
+    }
+};
+
+search_index_upsert_response
+make_response(std::error_code ec, search_index_upsert_request& request, search_index_upsert_request::encoded_response_type encoded)
+{
+    search_index_upsert_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;
+            } 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/protocol/client_opcode.hxx

@@ -41,6 +41,7 @@ enum class client_opcode : uint8_t {
     get_and_lock = 0x94,
     unlock = 0x95,
     get_collections_manifest = 0xba,
+    get_collection_id = 0xbb,
     subdoc_multi_lookup = 0xd0,
     subdoc_multi_mutation = 0xd1,
     get_cluster_config = 0xb5,
@@ -52,6 +53,7 @@ enum class client_opcode : uint8_t {
  * subdocument opcodes are listed separately, because we are not going to implement/support single-op messages
  */
 enum class subdoc_opcode : uint8_t {
+    get_doc = 0x00,
     get = 0xc5,
     exists = 0xc6,
     dict_add = 0xc7,
@@ -93,6 +95,7 @@ is_valid_client_opcode(uint8_t code)
         case client_opcode::get_and_touch:
         case client_opcode::increment:
         case client_opcode::decrement:
+        case client_opcode::get_collection_id:
             return true;
     }
     return false;
@@ -114,6 +117,7 @@ is_valid_subdoc_opcode(uint8_t code)
         case subdoc_opcode::array_add_unique:
         case subdoc_opcode::counter:
         case subdoc_opcode::get_count:
+        case subdoc_opcode::get_doc:
             return true;
     }
     return false;
@@ -196,6 +200,9 @@ struct fmt::formatter : formatter
             case couchbase::protocol::client_opcode::decrement:
                 name = "decrement";
                 break;
+            case couchbase::protocol::client_opcode::get_collection_id:
+                name = "get_collection_uid";
+                break;
         }
         return formatter<string_view>::format(name, ctx);
     }
@@ -244,6 +251,9 @@ struct fmt::formatter : formatter
             case couchbase::protocol::subdoc_opcode::get_count:
                 name = "get_count";
                 break;
+            case couchbase::protocol::subdoc_opcode::get_doc:
+                name = "get_doc";
+                break;
         }
         return formatter<string_view>::format(name, ctx);
     }