acl/app/master/tools/master_ctl/main.cpp
2022-07-01 19:02:30 +08:00

830 lines
18 KiB
C++

#include "stdafx.h"
#include <signal.h>
#ifdef HAS_READLINE
#include <readline/readline.h>
#include <readline/history.h>
#endif
#include "http_request.h"
static bool __verbose = false;
static long long __timeout = 0;
static int __signum = -1;
static acl::string __server_addr;
static acl::string __extname;
static void print_space(int n)
{
for (int i = 0; i < n; i++)
putchar(' ');
printf("\r\n");
}
static void print_name(const char* name, int tabs = 1)
{
#ifdef HAS_READLINE
printf("\033[1;33;40m%-s\033[0m", name);
#else
printf("%-s", name);
#endif
int len = (int) strlen(name);
int n = tabs * 8;
if (len >= n) {
n = 0;
} else {
n -= len;
}
char fmt[64];
snprintf(fmt, sizeof(fmt), "%%-%ds", n);
printf(fmt, "");
}
static void print_value(const char* value, int tabs = 1)
{
#ifdef HAS_READLINE
printf("\033[1;36;40m%-s\033[0m", value);
#else
printf("%-s", value);
#endif
int len = (int) strlen(value);
int n = tabs * 8;
if (len >= n) {
n = 0;
} else {
n -= len;
}
char fmt[64];
snprintf(fmt, sizeof(fmt), "%%-%ds", n);
printf(fmt, "");
}
static void print_value(int value, int tabs = 1)
{
char buf[32];
snprintf(buf, sizeof(buf), "%d", value);
print_value(buf, tabs);
}
static void print_head(void)
{
print_name("status");
print_name("service", 6);
print_name("type");
print_name("proc", 1);
print_name("owner");
print_name("conf");
printf("\r\n");
}
static void print_server(const serv_info_t& server)
{
print_value(server.status);
print_value(server.name, 6);
print_value(server.type);
print_value(server.proc_total, 1);
print_value(server.owner);
print_value(server.conf);
printf("\r\n");
}
static void println_underline(const char* name, const char* value)
{
#ifdef HAS_READLINE
printf("\033[4m\033[1;36;40m%c\033[0m\033[0m", *name);
#else
printf("%c", *name);
#endif
name++;
#ifdef HAS_READLINE
printf("\033[1;33;40m%-18s\033[0m: %s\r\n", name, value);
#else
printf("%-18s: %s\r\n", name, value);
#endif
}
static void println(const char* name, const char* value)
{
#ifdef HAS_READLINE
printf("\033[1;33;40m%-18s\033[0m: %s\r\n", name, value);
#else
printf("%-18s: %s\r\n", name, value);
#endif
}
static void println(const char* name, int value)
{
char buf[32];
snprintf(buf, sizeof(buf), "%d", value);
println(name, buf);
}
static const char* time_format(time_t t, char* buf, size_t size)
{
struct tm local_time;
localtime_r(&t, &local_time);
strftime(buf, size, "%Y/%m/%d-%H:%M:%S", &local_time);
return buf;
}
static void println_server(const serv_info_t& server)
{
println("status", server.status);
println("name", server.name.c_str());
println("type", server.type);
println("owner", server.owner.c_str());
println("path", server.path.c_str());
println("conf", server.conf.c_str());
println("proc_max", server.proc_max);
println("proc_prefork", server.proc_prefork);
println("proc_total", server.proc_total);
println("proc_avail", server.proc_avail);
println("throttle_delay", server.throttle_delay);
println("listen_fd_count", server.listen_fd_count);
println("notify_addr", server.notify_addr.c_str());
println("notify_recipients", server.notify_recipients.c_str());
println("version", server.version.c_str());
acl::string buf;
size_t i = 0;
for (std::list<proc_info_t>::const_iterator cit = server.procs.begin();
cit != server.procs.end(); ++cit)
{
if (i++ > 0)
buf << " ";
char tmp[128];
buf << "pid=" << (*cit).pid << ", start="
<< time_format((time_t) (*cit).start, tmp, sizeof(tmp));
buf += ";\r\n";
}
println("processes", buf);
}
static void print_servers(const std::vector<serv_info_t>& servers, bool verbose)
{
if (!verbose)
print_head();
for (std::vector<serv_info_t>::const_iterator cit = servers.begin();
cit != servers.end(); ++cit)
{
if (!verbose)
{
print_server(*cit);
continue;
}
if (cit != servers.begin())
print_space(100);
println_server(*cit);
}
}
static bool do_list(const std::vector<acl::string>&,
const char* addr, const char*)
{
list_req_t req;
req.cmd = "list";
list_res_t res;
if (!http_request<list_req_t, list_res_t>(addr, req, res))
return false;
print_servers(res.data, __verbose);
char buf[32];
snprintf(buf, sizeof(buf), "\r\nservers count: %d\r\n",
(int) res.data.size());
print_name(buf);
return true;
}
static bool do_stat(const std::vector<acl::string>&,
const char* addr, const char* fpath)
{
if (*fpath == 0)
{
printf("filepath null\r\n");
return false;
}
stat_req_t req;
req.cmd = "stat";
stat_req_data_t req_data;
req_data.path = fpath;
req.data.push_back(req_data);
stat_res_t res;
if (!http_request<stat_req_t, stat_res_t>(addr, req, res))
return false;
print_servers(res.data, true);
return true;
}
static void print_start_result(const start_res_data_t& data)
{
println("status", data.status);
println("name", data.name.c_str());
println("path", data.path.c_str());
}
static void print_start_results(const std::vector<start_res_data_t>& data)
{
for (std::vector<start_res_data_t>::const_iterator
cit = data.begin(); cit != data.end(); ++cit)
{
print_start_result(*cit);
}
}
static bool do_start(const std::vector<acl::string>&,
const char* addr, const char* fpath)
{
if (*fpath == 0)
{
printf("filepath null\r\n");
return false;
}
start_req_t req;
req.cmd = "start";
req.timeout = __timeout;
start_req_data_t req_data;
req_data.path = fpath;
if (!__extname.empty())
req_data.ext = __extname.c_str();
req.data.push_back(req_data);
start_res_t res;
if (!http_request<start_req_t, start_res_t>(addr, req, res))
return false;
print_start_results(res.data);
return true;
}
static void print_restart_result(const restart_res_data_t& data)
{
println("status", data.status);
println("name", data.name.c_str());
println("path", data.path.c_str());
}
static void print_restart_results(const std::vector<restart_res_data_t>& data)
{
for (std::vector<restart_res_data_t>::const_iterator
cit = data.begin(); cit != data.end(); ++cit)
{
print_restart_result(*cit);
}
}
static bool do_restart(const std::vector<acl::string>&,
const char* addr, const char* fpath)
{
if (*fpath == 0)
{
printf("filepath null\r\n");
return false;
}
restart_req_t req;
req.cmd = "restart";
restart_req_data_t req_data;
req_data.path = fpath;
if (!__extname.empty())
req_data.ext = __extname.c_str();
req.data.push_back(req_data);
restart_res_t res;
if (!http_request<restart_req_t, restart_res_t>(addr, req, res))
return false;
print_restart_results(res.data);
return true;
}
static void print_stop_result(const stop_res_data_t& data)
{
println("status", data.status);
println("path", data.path.c_str());
}
static void print_stop_results(const std::vector<stop_res_data_t>& data)
{
for (std::vector<stop_res_data_t>::const_iterator cit = data.begin();
cit != data.end(); ++cit)
{
print_stop_result(*cit);
}
}
static bool do_stop(const std::vector<acl::string>&,
const char* addr, const char* fpath)
{
if (*fpath == 0)
{
printf("filepath null\r\n");
return false;
}
stop_req_t req;
req.cmd = "stop";
stop_req_data_t req_data;
req_data.path = fpath;
req.data.push_back(req_data);
stop_res_t res;
if (!http_request<stop_req_t, stop_res_t>(addr, req, res))
return false;
print_stop_results(res.data);
return true;
}
static void print_kill_result(const kill_res_data_t& data)
{
println("status", data.status);
println("path", data.path.c_str());
}
static void print_kill_results(const std::vector<kill_res_data_t>& data)
{
for (std::vector<kill_res_data_t>::const_iterator cit = data.begin();
cit != data.end(); ++cit)
{
print_kill_result(*cit);
}
}
static bool do_kill(const std::vector<acl::string>&,
const char* addr, const char* fpath)
{
if (*fpath == 0)
{
printf("filepath null\r\n");
return false;
}
kill_req_t req;
req.cmd = "kill";
kill_req_data_t req_data;
req_data.path = fpath;
req.data.push_back(req_data);
kill_res_t res;
if (!http_request<kill_req_t, kill_res_t>(addr, req, res))
return false;
print_kill_results(res.data);
return true;
}
static void print_reload(const reload_res_data_t& data)
{
println("status", data.status);
println("proc_count", data.proc_count);
println("proc_signaled", data.proc_signaled);
println("proc_ok", data.proc_ok);
println("path", data.path.c_str());
}
static void print_reload_results(const std::vector<reload_res_data_t>& res)
{
for (std::vector<reload_res_data_t>::const_iterator cit = res.begin();
cit != res.end(); ++cit)
{
print_reload(*cit);
}
}
static void print_signal(const signal_res_data_t& data)
{
println("status", data.status);
println("proc_count", data.proc_count);
println("proc_signaled", data.proc_signaled);
println("proc_ok", data.proc_ok);
println("path", data.path.c_str());
}
static void print_signal_results(const std::vector<signal_res_data_t>& res)
{
for (std::vector<signal_res_data_t>::const_iterator cit = res.begin();
cit != res.end(); ++cit)
{
print_signal(*cit);
}
}
static bool do_reload(const std::vector<acl::string>& tokens,
const char* addr, const char* fpath)
{
if (*fpath == 0)
{
#ifdef HAS_READLINE
printf("\033[1;34;40musage\033[0m: "
"\033[1;33;40mreload\033[0m configure_path timeout\r\n");
#else
printf("usage: reload configure_path timeout\r\n");
#endif
return false;
}
if (tokens.size() >= 3)
__timeout = atoll(tokens[2]);
reload_req_t req;
req.cmd = "reload";
req.timeout = __timeout;
reload_req_data_t req_data;
req_data.path = fpath;
req.data.push_back(req_data);
reload_res_t res;
if (!http_request<reload_req_t, reload_res_t>(addr, req, res))
return false;
print_reload_results(res.data);
return true;
}
static int get_signum(const char* signame)
{
static struct {
const char* name;
int signum;
} sigmap[] = {
{ "SIGINT", SIGINT },
{ "SIGHUP", SIGHUP },
{ "SIGUSR1", SIGUSR1 },
{ "SIGUSR2", SIGUSR2 },
{ NULL, -1 },
};
for (int i = 0; sigmap[i].name != NULL; i++) {
if (strcasecmp(sigmap[i].name, signame) == 0) {
return sigmap[i].signum;
}
}
return -1;
}
static bool do_signal(const std::vector<acl::string>& tokens,
const char* addr, const char* fpath)
{
if (*fpath == 0)
{
#ifdef HAS_READLINE
printf("\033[1;34;40musage\033[0m: "
"\033[1;33;40msignal\033[0m configure_path SIGXXX\r\n");
#else
printf("usage: signal configure_path SIGXXX\r\n");
#endif
return false;
}
if (tokens.size() >= 3)
__signum = get_signum(tokens[2]);
signal_req_t req;
req.cmd = "reload";
req.signum = __timeout;
signal_req_data_t req_data;
req_data.path = fpath;
req.data.push_back(req_data);
signal_res_t res;
if (!http_request<signal_req_t, signal_res_t>(addr, req, res))
return false;
print_signal_results(res.data);
return true;
}
static bool do_master_config(const std::vector<acl::string>&,
const char* addr, const char*)
{
master_config_req_t req;
req.cmd = "master_config";
master_config_res_t res;
if (!http_request<master_config_req_t, master_config_res_t>
(addr, req, res)) {
return false;
}
for (std::map<acl::string, acl::string>::const_iterator cit =
res.data.begin(); cit != res.data.end(); ++cit) {
println(cit->first.c_str(), cit->second.c_str());
}
return true;
}
static bool do_set(const std::vector<acl::string>& tokens,
const char*, const char* fpath)
{
if (*fpath == 0)
{
#ifdef HAS_READLINE
printf("\033[1;34;40musage\033[0m: "
"\033[1;33;40mset\033[0m configure_path timeout\r\n");
#else
printf("usage: set configure_path timeout\r\n");
#endif
return true;
}
if (tokens.size() >= 3)
__timeout = atoll(tokens[2]);
printf("set service to %s ok, timeout=%lld\r\n", fpath, __timeout);
return true;
}
static bool do_server(const std::vector<acl::string>& tokens,
const char* addr, const char*)
{
if (tokens.size() > 1)
{
addr = tokens[1].c_str();
__server_addr = addr;
const char* ptr = strchr(addr, ':');
if (ptr == NULL || *++ptr == 0 || atoi(ptr) <= 0)
__server_addr += ":8290";
printf("set server to %s ok\r\n", __server_addr.c_str());
}
else if (*addr == 0)
#ifdef HAS_READLINE
printf("\033[1;34;40musage\033[0m: "
"\033[1;33;40mserver\033[0m addr\r\n");
#else
printf("usage: server addr\r\n");
#endif
else
printf("server addr is %s\r\n", addr);
return true;
}
static bool do_timeout(const std::vector<acl::string>& tokens,
const char*, const char*)
{
if (tokens.size() >= 2)
__timeout = atoll(tokens[1]);
else
#ifdef HAS_READLINE
printf("\033[1;34;40musage\033[0m: "
"\033[1;33;40mtimeout\033[0m timeout\r\n");
#else
printf("usage: timeout timeout\r\n");
#endif
return true;
}
static bool do_verbose(const std::vector<acl::string>&,
const char*, const char*)
{
__verbose = __verbose ? false : true;
printf("set verbose %s\r\n", __verbose ? "on" : "off");
return true;
}
static bool do_help(const std::vector<acl::string>&, const char*, const char*)
{
println("set", "set the service's configure");
println("server", "set the service's addr");
println_underline("verbose", "display verbose");
println_underline("clear", "clear the screan");
println_underline("list", "list all the running services");
println_underline("stat", "show one service's running status");
println("start", "start one service");
println("restart", "restart one service");
println("stop", "stop one service");
println("kill", "kill one service");
println("reload", "reload one service");
println("signal", "signal one service");
println("master_config", "get master's configure entries");
println_underline("timeout", "show one service's running status");
return true;
}
static bool do_quit(const std::vector<acl::string>&, const char*, const char*)
{
printf("Bye!\r\n");
exit(0);
// not reached here
return false;
}
static bool do_clear(const std::vector<acl::string>&, const char*, const char*)
{
#if !defined(__APPLE__)
#ifdef HAS_READLINE
rl_clear_screen(0, 0);
#endif
#endif
printf("\r\n");
return true;
}
static void getline(acl::string& out)
{
#ifdef HAS_READLINE
const char* prompt = "\033[1;34;40mmaster_ctl>\033[0m ";
char* ptr = readline(prompt);
#else
printf("master_ctl> ");
fflush(stdout);
char buf[1024];
char* ptr = fgets(buf, (int) sizeof(buf), stdin);
#endif
if (ptr == NULL || *ptr == 0)
{
printf("Bye!\r\n");
exit(0);
}
out = ptr;
out.trim_right_line();
#ifdef HAS_READLINE
if (!out.empty() && !out.equal("y", false) && !out.equal("n", false))
add_history(out.c_str());
#endif
}
static struct {
const char* cmd;
char shortcut;
bool has_path;
bool (*fn)(const std::vector<acl::string>&, const char*, const char*);
} __actions[] = {
{ "list", 'l', false, do_list },
{ "stat", 's', true, do_stat },
{ "start", '\0', true, do_start },
{ "restart", '\0', true, do_restart },
{ "stop", '\0', true, do_stop },
{ "kill", '\0', true, do_kill },
{ "reload", 'r', true, do_reload },
{ "signal", 'k', true, do_signal },
{ "master_config", '\0', false, do_master_config },
{ "help", 'h', false, do_help },
{ "clear", 'c', false, do_clear },
{ "quit", 'q', false, do_quit },
{ "exit", 'e', false, do_quit },
{ "set", '\0', true, do_set },
{ "server", '\0', false, do_server },
{ "timeout", 't', false, do_timeout },
{ "verbose", 'v', false, do_verbose },
{ 0, 0, false, 0 },
};
static void run(const char* server, const char* filepath)
{
acl::string buf, fpath;
__server_addr = server;
printf("server addr is %s\r\n", server);
if (filepath && *filepath)
fpath = filepath;
while (true)
{
getline(buf);
if (buf.empty())
continue;
std::vector<acl::string>& tokens = buf.split2(" \t", true);
acl::string cmd = tokens[0];
cmd.lower();
bool ret = false;
int i;
for (i = 0; __actions[i].cmd; i++)
{
if (cmd == __actions[i].cmd ||
(__actions[i].shortcut &&
cmd.size() == 1 &&
cmd[0] == __actions[i].shortcut))
{
break;
}
}
if (__actions[i].cmd == NULL)
{
do_help(tokens, __server_addr, fpath);
continue;
}
if (__actions[i].has_path && tokens.size() >= 2)
fpath = tokens[1];
ret = __actions[i].fn(tokens, __server_addr, fpath);
if (!__verbose)
print_space(100);
#ifdef HAS_READLINE
printf("\033[1;36;40m%s\033[0m ==> \033[1;32;40m%s\033[0m\r\n",
__actions[i].cmd, ret ? "ok" : "err");
#else
printf("%s ==> %s\r\n", __actions[i].cmd, ret ? "ok" : "err");
#endif
}
}
static void usage(const char* procname)
{
printf("usage: %s -h[help]\r\n"
" -s master_manage_addr[default: /opt/soft/acl-master/var/public/master.sock]\r\n"
" -f servicde_path\r\n"
" -t timeout[waiting the result from master, default: 0]\r\n"
" -a cmd[list|stat|start|stop|reload|restart|signal]\r\n"
" -n signum[specify the signal number if command is signal]\r\n"
" -e extname[specified the extname of service's path, just for start and restart]\r\n",
procname);
}
int main(int argc, char* argv[])
{
acl::string filepath, action, addr("/opt/soft/acl-master/var/public/master.sock");
int ch;
while ((ch = getopt(argc, argv, "hs:f:a:t:e:n:")) > 0)
{
switch (ch)
{
case 'h':
usage(argv[0]);
return 0;
case 's':
addr = optarg;
break;
case 'f':
filepath = optarg;
break;
case 'a':
action = optarg;
break;
case 't':
__timeout = atoi(optarg);
if (__timeout < 0)
__timeout = 0;
break;
case 'e':
__extname = optarg;
break;
case 'n':
__signum = get_signum(optarg);
break;
default:
usage(argv[0]);
return 1;
}
}
action.lower();
acl::log::stdout_open(true);
std::vector<acl::string> tokens;
int i;
for (i = 0; __actions[i].cmd; i++)
{
if (action == __actions[i].cmd)
{
(void) __actions[i].fn(tokens, addr, filepath);
break;
}
}
if (__actions[i].cmd == NULL)
run(addr, filepath);
return 0;
}