Rename cli/ to cmd/, fix cmake install

2024-05-15 23:04:50 +03:00
parent c79b38bd26
commit a12d328793
35 changed files with 25 additions and 25 deletions
@@ -0,0 +1,131 @@
+// Copyright (c) Vitaliy Filippov, 2019+
+// License: VNPL-1.1 (see README.md for details)
+
+#include <ctype.h>
+#include "cli.h"
+#include "cluster_client.h"
+#include "str_util.h"
+
+#include <algorithm>
+
+// Safely allocate an OSD number
+struct alloc_osd_t
+{
+    cli_tool_t *parent;
+
+    uint64_t new_id = 1;
+
+    int state = 0;
+    cli_result_t result;
+
+    bool is_done()
+    {
+        return state == 100;
+    }
+
+    void loop()
+    {
+        if (state == 1)
+            goto resume_1;
+        do
+        {
+            parent->etcd_txn(json11::Json::object {
+                { "compare", json11::Json::array {
+                    json11::Json::object {
+                        { "target", "VERSION" },
+                        { "version", 0 },
+                        { "key", base64_encode(
+                            parent->cli->st_cli.etcd_prefix+"/osd/stats/"+std::to_string(new_id)
+                        ) },
+                    },
+                } },
+                { "success", json11::Json::array {
+                    json11::Json::object {
+                        { "request_put", json11::Json::object {
+                            { "key", base64_encode(
+                                parent->cli->st_cli.etcd_prefix+"/osd/stats/"+std::to_string(new_id)
+                            ) },
+                            { "value", base64_encode("{}") },
+                        } },
+                    },
+                } },
+                { "failure", json11::Json::array {
+                    json11::Json::object {
+                        { "request_range", json11::Json::object {
+                            { "key", base64_encode(parent->cli->st_cli.etcd_prefix+"/osd/stats/") },
+                            { "range_end", base64_encode(parent->cli->st_cli.etcd_prefix+"/osd/stats0") },
+                            { "keys_only", true },
+                        } },
+                    },
+                } },
+            });
+        resume_1:
+            state = 1;
+            if (parent->waiting > 0)
+                return;
+            if (parent->etcd_err.err)
+            {
+                result = parent->etcd_err;
+                state = 100;
+                return;
+            }
+            if (!parent->etcd_result["succeeded"].bool_value())
+            {
+                std::vector<osd_num_t> used;
+                for (auto kv: parent->etcd_result["responses"][0]["response_range"]["kvs"].array_items())
+                {
+                    std::string key = base64_decode(kv["key"].string_value());
+                    osd_num_t cur_osd;
+                    char null_byte = 0;
+                    int scanned = sscanf(key.c_str() + parent->cli->st_cli.etcd_prefix.length(), "/osd/stats/%ju%c", &cur_osd, &null_byte);
+                    if (scanned != 1 || !cur_osd)
+                    {
+                        fprintf(stderr, "Invalid key in etcd: %s\n", key.c_str());
+                        continue;
+                    }
+                    used.push_back(cur_osd);
+                }
+                std::sort(used.begin(), used.end());
+                if (used[used.size()-1] == used.size())
+                {
+                    new_id = used.size()+1;
+                }
+                else
+                {
+                    int s = 0, e = used.size();
+                    while (e > s+1)
+                    {
+                        int c = (s+e)/2;
+                        if (used[c] == c+1)
+                            s = c;
+                        else
+                            e = c;
+                    }
+                    new_id = used[e-1]+1;
+                }
+            }
+        } while (!parent->etcd_result["succeeded"].bool_value());
+        state = 100;
+        result = (cli_result_t){
+            .text = std::to_string(new_id),
+            .data = json11::Json(new_id),
+        };
+    }
+};
+
+std::function<bool(cli_result_t &)> cli_tool_t::start_alloc_osd(json11::Json cfg)
+{
+    auto alloc_osd = new alloc_osd_t();
+    alloc_osd->parent = this;
+    return [alloc_osd](cli_result_t & result)
+    {
+        alloc_osd->loop();
+        if (alloc_osd->is_done())
+        {
+            result = alloc_osd->result;
+            delete alloc_osd;
+            return true;
+        }
+        return false;
+    };
+}