2022-04-03 20:14:51 +03:00
|
|
|
// Copyright (c) Vitaliy Filippov, 2019+
|
|
|
|
// License: VNPL-1.1 (see README.md for details)
|
|
|
|
|
|
|
|
#include "base64.h"
|
|
|
|
#include "cluster_client.h"
|
|
|
|
#include "cli.h"
|
|
|
|
|
|
|
|
void cli_tool_t::change_parent(inode_t cur, inode_t new_parent, cli_result_t *result)
|
|
|
|
{
|
|
|
|
auto cur_cfg_it = cli->st_cli.inode_config.find(cur);
|
|
|
|
if (cur_cfg_it == cli->st_cli.inode_config.end())
|
|
|
|
{
|
|
|
|
char buf[128];
|
|
|
|
snprintf(buf, 128, "Inode 0x%lx disappeared", cur);
|
2022-05-10 12:26:47 +03:00
|
|
|
*result = (cli_result_t){ .err = EIO, .text = buf };
|
2022-04-03 20:14:51 +03:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
inode_config_t new_cfg = cur_cfg_it->second;
|
|
|
|
std::string cur_name = new_cfg.name;
|
|
|
|
std::string cur_cfg_key = base64_encode(cli->st_cli.etcd_prefix+
|
|
|
|
"/config/inode/"+std::to_string(INODE_POOL(cur))+
|
|
|
|
"/"+std::to_string(INODE_NO_POOL(cur)));
|
|
|
|
new_cfg.parent_id = new_parent;
|
|
|
|
json11::Json::object cur_cfg_json = cli->st_cli.serialize_inode_cfg(&new_cfg);
|
|
|
|
waiting++;
|
|
|
|
cli->st_cli.etcd_txn_slow(json11::Json::object {
|
|
|
|
{ "compare", json11::Json::array {
|
|
|
|
json11::Json::object {
|
|
|
|
{ "target", "MOD" },
|
|
|
|
{ "key", cur_cfg_key },
|
|
|
|
{ "result", "LESS" },
|
|
|
|
{ "mod_revision", new_cfg.mod_revision+1 },
|
|
|
|
},
|
|
|
|
} },
|
|
|
|
{ "success", json11::Json::array {
|
|
|
|
json11::Json::object {
|
|
|
|
{ "request_put", json11::Json::object {
|
|
|
|
{ "key", cur_cfg_key },
|
|
|
|
{ "value", base64_encode(json11::Json(cur_cfg_json).dump()) },
|
|
|
|
} }
|
|
|
|
},
|
|
|
|
} },
|
|
|
|
}, [this, result, new_parent, cur, cur_name](std::string err, json11::Json res)
|
|
|
|
{
|
|
|
|
if (err != "")
|
|
|
|
{
|
2022-05-10 12:26:47 +03:00
|
|
|
*result = (cli_result_t){ .err = EIO, .text = "Error changing parent of "+cur_name+": "+err };
|
2022-04-03 20:14:51 +03:00
|
|
|
}
|
|
|
|
else if (!res["succeeded"].bool_value())
|
|
|
|
{
|
2022-05-10 12:26:47 +03:00
|
|
|
*result = (cli_result_t){ .err = EAGAIN, .text = "Image "+cur_name+" was modified during change" };
|
2022-04-03 20:14:51 +03:00
|
|
|
}
|
|
|
|
else if (new_parent)
|
|
|
|
{
|
|
|
|
auto new_parent_it = cli->st_cli.inode_config.find(new_parent);
|
|
|
|
std::string new_parent_name = new_parent_it != cli->st_cli.inode_config.end()
|
|
|
|
? new_parent_it->second.name : "<unknown>";
|
2022-05-10 12:26:47 +03:00
|
|
|
*result = (cli_result_t){
|
2022-04-03 20:14:51 +03:00
|
|
|
.text = "Parent of layer "+cur_name+" (inode "+std::to_string(INODE_NO_POOL(cur))+
|
|
|
|
" in pool "+std::to_string(INODE_POOL(cur))+") changed to "+new_parent_name+
|
|
|
|
" (inode "+std::to_string(INODE_NO_POOL(new_parent))+" in pool "+std::to_string(INODE_POOL(new_parent))+")",
|
|
|
|
};
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2022-05-10 12:26:47 +03:00
|
|
|
*result = (cli_result_t){
|
2022-04-03 20:14:51 +03:00
|
|
|
.text = "Parent of layer "+cur_name+" (inode "+std::to_string(INODE_NO_POOL(cur))+
|
|
|
|
" in pool "+std::to_string(INODE_POOL(cur))+") detached",
|
|
|
|
};
|
|
|
|
}
|
|
|
|
waiting--;
|
|
|
|
ringloop->wakeup();
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
void cli_tool_t::etcd_txn(json11::Json txn)
|
|
|
|
{
|
|
|
|
waiting++;
|
|
|
|
cli->st_cli.etcd_txn_slow(txn, [this](std::string err, json11::Json res)
|
|
|
|
{
|
|
|
|
waiting--;
|
|
|
|
if (err != "")
|
2022-05-10 12:26:47 +03:00
|
|
|
etcd_err = (cli_result_t){ .err = EIO, .text = "Error communicating with etcd: "+err };
|
2022-04-03 20:14:51 +03:00
|
|
|
else
|
2022-05-10 12:26:47 +03:00
|
|
|
etcd_err = (cli_result_t){ .err = 0 };
|
2022-04-03 20:14:51 +03:00
|
|
|
etcd_result = res;
|
|
|
|
ringloop->wakeup();
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
inode_config_t* cli_tool_t::get_inode_cfg(const std::string & name)
|
|
|
|
{
|
|
|
|
for (auto & ic: cli->st_cli.inode_config)
|
|
|
|
{
|
|
|
|
if (ic.second.name == name)
|
|
|
|
{
|
|
|
|
return &ic.second;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
void cli_tool_t::parse_config(json11::Json cfg)
|
|
|
|
{
|
|
|
|
color = !cfg["no-color"].bool_value();
|
|
|
|
json_output = cfg["json"].bool_value();
|
|
|
|
iodepth = cfg["iodepth"].uint64_value();
|
|
|
|
if (!iodepth)
|
|
|
|
iodepth = 32;
|
|
|
|
parallel_osds = cfg["parallel_osds"].uint64_value();
|
|
|
|
if (!parallel_osds)
|
|
|
|
parallel_osds = 4;
|
|
|
|
log_level = cfg["log_level"].int64_value();
|
|
|
|
progress = cfg["progress"].uint64_value() ? true : false;
|
|
|
|
list_first = cfg["wait-list"].uint64_value() ? true : false;
|
|
|
|
}
|
|
|
|
|
|
|
|
struct cli_result_looper_t
|
|
|
|
{
|
|
|
|
ring_consumer_t consumer;
|
|
|
|
cli_result_t result;
|
|
|
|
std::function<bool(cli_result_t &)> loop_cb;
|
|
|
|
std::function<void(const cli_result_t &)> complete_cb;
|
|
|
|
};
|
|
|
|
|
|
|
|
void cli_tool_t::loop_and_wait(std::function<bool(cli_result_t &)> loop_cb, std::function<void(const cli_result_t &)> complete_cb)
|
|
|
|
{
|
|
|
|
auto *looper = new cli_result_looper_t();
|
|
|
|
looper->loop_cb = loop_cb;
|
|
|
|
looper->complete_cb = complete_cb;
|
|
|
|
looper->consumer.loop = [this, looper]()
|
|
|
|
{
|
|
|
|
bool done = looper->loop_cb(looper->result);
|
|
|
|
if (done)
|
|
|
|
{
|
|
|
|
ringloop->unregister_consumer(&looper->consumer);
|
|
|
|
looper->loop_cb = NULL;
|
|
|
|
looper->complete_cb(looper->result);
|
|
|
|
delete looper;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
ringloop->submit();
|
|
|
|
};
|
|
|
|
cli->on_ready([this, looper]()
|
|
|
|
{
|
|
|
|
ringloop->register_consumer(&looper->consumer);
|
|
|
|
ringloop->wakeup();
|
|
|
|
});
|
|
|
|
}
|