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

data/ext/couchbase/operations/document_upsert.hxx

@@ -0,0 +1,74 @@
+/* -*- 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 <document_id.hxx>
+#include <protocol/cmd_upsert.hxx>
+#include <protocol/durability_level.hxx>
+
+namespace couchbase::operations
+{
+
+struct upsert_response {
+    document_id id;
+    std::error_code ec{};
+    std::uint64_t cas{};
+    mutation_token token{};
+};
+
+struct upsert_request {
+    using encoded_request_type = protocol::client_request<protocol::upsert_request_body>;
+    using encoded_response_type = protocol::client_response<protocol::upsert_response_body>;
+
+    document_id id;
+    std::string value;
+    uint16_t partition{};
+    uint32_t opaque{};
+    uint32_t flags{ 0 };
+    uint32_t expiration{ 0 };
+    protocol::durability_level durability_level{ protocol::durability_level::none };
+    std::optional<std::uint16_t> durability_timeout{};
+
+    void encode_to(encoded_request_type& encoded)
+    {
+        encoded.opaque(opaque);
+        encoded.partition(partition);
+        encoded.body().id(id);
+        encoded.body().expiration(expiration);
+        encoded.body().flags(flags);
+        encoded.body().content(value);
+        if (durability_level != protocol::durability_level::none) {
+            encoded.body().durability(durability_level, durability_timeout);
+        }
+    }
+};
+
+upsert_response
+make_response(std::error_code ec, upsert_request& request, upsert_request::encoded_response_type encoded)
+{
+    upsert_response response{ request.id, ec };
+    if (!ec) {
+        response.cas = encoded.cas();
+        response.token = encoded.body().token();
+        response.token.partition_id = request.partition;
+        response.token.bucket_name = response.id.bucket;
+    }
+    return response;
+}
+
+} // namespace couchbase::operations

data/ext/couchbase/operations/query_index_build_deferred.hxx

@@ -0,0 +1,85 @@
+/* -*- 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 query_index_build_deferred_response {
+    struct query_problem {
+        std::uint64_t code;
+        std::string message;
+    };
+    uuid::uuid_t client_context_id;
+    std::error_code ec;
+    std::string status{};
+    std::vector<query_problem> errors{};
+};
+
+struct query_index_build_deferred_request {
+    using response_type = query_index_build_deferred_response;
+    using encoded_request_type = io::http_request;
+    using encoded_response_type = io::http_response;
+
+    static const inline service_type type = service_type::query;
+
+    uuid::uuid_t client_context_id{ uuid::random() };
+    std::string bucket_name;
+
+    void encode_to(encoded_request_type& encoded)
+    {
+        encoded.headers["content-type"] = "application/json";
+        tao::json::value body{
+            { "statement",
+              fmt::format(R"(BUILD INDEX ON `{}` ((SELECT RAW name FROM system:indexes WHERE keyspace_id = "{}" AND state = "deferred")))",
+                          bucket_name,
+                          bucket_name) },
+            { "client_context_id", uuid::to_string(client_context_id) }
+        };
+        encoded.method = "POST";
+        encoded.path = "/query/service";
+        encoded.body = tao::json::to_string(body);
+    }
+};
+
+query_index_build_deferred_response
+make_response(std::error_code ec,
+              query_index_build_deferred_request& request,
+              query_index_build_deferred_request::encoded_response_type encoded)
+{
+    query_index_build_deferred_response response{ request.client_context_id, ec };
+    if (!ec) {
+        auto payload = tao::json::from_string(encoded.body);
+        response.status = payload.at("status").get_string();
+        if (response.status != "success") {
+            for (const auto& entry : payload.at("errors").get_array()) {
+                query_index_build_deferred_response::query_problem error;
+                error.code = entry.at("code").get_unsigned();
+                error.message = entry.at("msg").get_string();
+                response.errors.emplace_back(error);
+            }
+            response.ec = std::make_error_code(error::common_errc::internal_server_failure);
+        }
+    }
+    return response;
+}
+
+} // namespace couchbase::operations

data/ext/couchbase/operations/query_index_create.hxx

@@ -0,0 +1,134 @@
+/* -*- 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 query_index_create_response {
+    struct query_problem {
+        std::uint64_t code;
+        std::string message;
+    };
+    uuid::uuid_t client_context_id;
+    std::error_code ec;
+    std::string status{};
+    std::vector<query_problem> errors{};
+};
+
+struct query_index_create_request {
+    using response_type = query_index_create_response;
+    using encoded_request_type = io::http_request;
+    using encoded_response_type = io::http_response;
+
+    static const inline service_type type = service_type::query;
+
+    uuid::uuid_t client_context_id{ uuid::random() };
+    std::string bucket_name;
+    std::string index_name{};
+    std::vector<std::string> fields;
+    bool is_primary{ false };
+    bool ignore_if_exists{ false };
+    std::optional<std::string> condition{};
+    std::optional<bool> deferred{};
+    std::optional<int> num_replicas{};
+
+    void encode_to(encoded_request_type& encoded)
+    {
+        encoded.headers["content-type"] = "application/json";
+        tao::json::value with{};
+        if (deferred) {
+            with["defer_build"] = *deferred;
+        }
+        if (num_replicas) {
+            with["num_replica"] = *num_replicas; /* no 's' in key name */
+        }
+        std::string where_clause{};
+        if (condition) {
+            where_clause = fmt::format("WHERE {}", *condition);
+        }
+        std::string with_clause{};
+        if (with) {
+            with_clause = fmt::format("WITH {}", tao::json::to_string(with));
+        }
+        tao::json::value body{ { "statement",
+                                 is_primary ? fmt::format(R"(CREATE PRIMARY INDEX {} ON `{}` USING GSI {})",
+                                                          index_name.empty() ? "" : fmt::format("`{}`", index_name),
+                                                          bucket_name,
+                                                          with_clause)
+                                            : fmt::format(R"(CREATE INDEX `{}` ON `{}`({}) {} USING GSI {})",
+                                                          index_name,
+                                                          bucket_name,
+                                                          fmt::join(fields, ", "),
+                                                          where_clause,
+                                                          with_clause) },
+                               { "client_context_id", uuid::to_string(client_context_id) } };
+        encoded.method = "POST";
+        encoded.path = "/query/service";
+        encoded.body = tao::json::to_string(body);
+    }
+};
+
+query_index_create_response
+make_response(std::error_code ec, query_index_create_request& request, query_index_create_request::encoded_response_type encoded)
+{
+    query_index_create_response response{ request.client_context_id, ec };
+    if (!ec) {
+        auto payload = tao::json::from_string(encoded.body);
+        response.status = payload.at("status").get_string();
+
+        if (response.status != "success") {
+            bool index_already_exists = false;
+            bool bucket_not_found = false;
+            for (const auto& entry : payload.at("errors").get_array()) {
+                query_index_create_response::query_problem error;
+                error.code = entry.at("code").get_unsigned();
+                error.message = entry.at("msg").get_string();
+                switch (error.code) {
+                    case 5000: /* IKey: "Internal Error" */
+                        if (error.message.find(" already exists") != std::string::npos) {
+                            index_already_exists = true;
+                        }
+                        break;
+                    case 12003: /* IKey: "datastore.couchbase.keyspace_not_found" */
+                        bucket_not_found = true;
+                        break;
+                    case 4300: /* IKey: "plan.new_index_already_exists" */
+                        index_already_exists = true;
+                        break;
+                }
+                response.errors.emplace_back(error);
+            }
+            if (index_already_exists) {
+                if (!request.ignore_if_exists) {
+                    response.ec = std::make_error_code(error::common_errc::index_exists);
+                }
+            } else if (bucket_not_found) {
+                response.ec = std::make_error_code(error::common_errc::bucket_not_found);
+            } else if (!response.errors.empty()) {
+                response.ec = std::make_error_code(error::common_errc::internal_server_failure);
+            }
+        }
+    }
+    return response;
+}
+
+} // namespace couchbase::operations

data/ext/couchbase/operations/query_index_drop.hxx

@@ -0,0 +1,108 @@
+/* -*- 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 query_index_drop_response {
+    struct query_problem {
+        std::uint64_t code;
+        std::string message;
+    };
+    uuid::uuid_t client_context_id;
+    std::error_code ec;
+    std::string status{};
+    std::vector<query_problem> errors{};
+};
+
+struct query_index_drop_request {
+    using response_type = query_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::query;
+
+    uuid::uuid_t client_context_id{ uuid::random() };
+    std::string bucket_name;
+    std::string index_name;
+    bool is_primary{ false };
+    bool ignore_if_does_not_exist{ false };
+
+    void encode_to(encoded_request_type& encoded)
+    {
+        encoded.headers["content-type"] = "application/json";
+        tao::json::value body{ { "statement",
+                                 is_primary ? fmt::format(R"(DROP PRIMARY INDEX ON `{}` USING GSI)", bucket_name)
+                                            : fmt::format(R"(DROP INDEX `{}`.`{}` USING GSI)", bucket_name, index_name) },
+                               { "client_context_id", uuid::to_string(client_context_id) } };
+        encoded.method = "POST";
+        encoded.path = "/query/service";
+        encoded.body = tao::json::to_string(body);
+    }
+};
+
+query_index_drop_response
+make_response(std::error_code ec, query_index_drop_request& request, query_index_drop_request::encoded_response_type encoded)
+{
+    query_index_drop_response response{ request.client_context_id, ec };
+    if (!ec) {
+        auto payload = tao::json::from_string(encoded.body);
+        response.status = payload.at("status").get_string();
+
+        if (response.status != "success") {
+            bool bucket_not_found = false;
+            bool index_not_found = false;
+            for (const auto& entry : payload.at("errors").get_array()) {
+                query_index_drop_response::query_problem error;
+                error.code = entry.at("code").get_unsigned();
+                error.message = entry.at("msg").get_string();
+                switch (error.code) {
+                    case 5000: /* IKey: "Internal Error" */
+                        if (error.message.find("not found.") != std::string::npos) {
+                            index_not_found = true;
+                        }
+                        break;
+                    case 12003: /* IKey: "datastore.couchbase.keyspace_not_found" */
+                        bucket_not_found = true;
+                        break;
+                    case 12004: /* IKey: "datastore.couchbase.primary_idx_not_found" */
+                    case 12006: /* IKey: "datastore.couchbase.keyspace_count_error" */
+                        index_not_found = true;
+                        break;
+                }
+                response.errors.emplace_back(error);
+            }
+            if (index_not_found) {
+                if (!request.ignore_if_does_not_exist) {
+                    response.ec = std::make_error_code(error::common_errc::index_not_found);
+                }
+            } else if (bucket_not_found) {
+                response.ec = std::make_error_code(error::common_errc::bucket_not_found);
+            } else if (!response.errors.empty()) {
+                response.ec = std::make_error_code(error::common_errc::internal_server_failure);
+            }
+        }
+    }
+    return response;
+}
+
+} // namespace couchbase::operations

data/ext/couchbase/operations/query_index_get_all.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 query_index_get_all_response {
+    struct query_index {
+        bool is_primary{ false };
+        std::string id;
+        std::string name;
+        std::string state;
+        std::string datastore_id;
+        std::string keyspace_id;
+        std::string namespace_id;
+        std::string type;
+        std::vector<std::string> index_key{};
+        std::optional<std::string> condition{};
+    };
+    uuid::uuid_t client_context_id;
+    std::error_code ec;
+    std::string status{};
+    std::vector<query_index> indexes{};
+};
+
+struct query_index_get_all_request {
+    using response_type = query_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::query;
+
+    uuid::uuid_t client_context_id{ uuid::random() };
+    std::string bucket_name;
+
+    void encode_to(encoded_request_type& encoded)
+    {
+        encoded.headers["content-type"] = "application/json";
+        tao::json::value body{
+            { "statement",
+              fmt::format(
+                R"(SELECT idx.* FROM system:indexes AS idx WHERE keyspace_id = "{}" AND `using`="gsi" ORDER BY is_primary DESC, name ASC)",
+                bucket_name) },
+            { "client_context_id", uuid::to_string(client_context_id) }
+        };
+        encoded.method = "POST";
+        encoded.path = "/query/service";
+        encoded.body = tao::json::to_string(body);
+    }
+};
+
+query_index_get_all_response
+make_response(std::error_code ec, query_index_get_all_request& request, query_index_get_all_request::encoded_response_type encoded)
+{
+    query_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 == "success") {
+                for (const auto& entry : payload.at("results").get_array()) {
+                    query_index_get_all_response::query_index index;
+                    index.id = entry.at("id").get_string();
+                    index.datastore_id = entry.at("datastore_id").get_string();
+                    index.namespace_id = entry.at("namespace_id").get_string();
+                    index.keyspace_id = entry.at("keyspace_id").get_string();
+                    index.type = entry.at("using").get_string();
+                    index.name = entry.at("name").get_string();
+                    index.state = entry.at("state").get_string();
+                    if (const auto* prop = entry.find("is_primary")) {
+                        index.is_primary = prop->get_boolean();
+                    }
+                    if (const auto* prop = entry.find("condition")) {
+                        index.condition = prop->get_string();
+                    }
+                    for (const auto& key : entry.at("index_key").get_array()) {
+                        index.index_key.emplace_back(key.get_string());
+                    }
+                    response.indexes.emplace_back(index);
+                }
+            }
+        }
+    }
+    return response;
+}
+
+} // namespace couchbase::operations