acl/app/master/tools/master_ctl/main.cpp

#include "stdafx.h"
#include <readline/readline.h>
#include <readline/history.h>
#include "http_request.h"

static void print_server(const serv_info_t& server)
{
	printf("status: %d\r\n", server.status);
	printf("name: %s\r\n", server.name.c_str());
	printf("type: %d\r\n", server.type);
	printf("owner: %s\r\n", server.owner.c_str());
	printf("path: %s\r\n", server.path.c_str());
	printf("conf: %s\r\n", server.conf.c_str());
	printf("proc_max: %d\r\n", server.proc_max);
	printf("proc_prefork: %d\r\n", server.proc_prefork);
	printf("proc_total: %d\r\n", server.proc_total);
	printf("proc_avail: %d\r\n", server.proc_avail);
	printf("throttle_delay: %d\r\n", server.throttle_delay);
	printf("listen_fd_count: %d\r\n", server.listen_fd_count);
	printf("notify_addr: %s\r\n", server.notify_addr.c_str());
	printf("notify_recipients: %s\r\n", server.notify_recipients.c_str());
}

static void print_servers(const std::list<serv_info_t>& servers)
{
	for (std::list<serv_info_t>::const_iterator cit = servers.begin();
		cit != servers.end(); ++cit)
	{
		print_server(*cit);
		printf("-----------------------------------------------\r\n");
	}
}

static void print_servers(const std::vector<serv_info_t>& servers)
{
	for (std::vector<serv_info_t>::const_iterator cit = servers.begin();
		cit != servers.end(); ++cit)
	{
		print_server(*cit);
		printf("-----------------------------------------------\r\n");
	}
}

static bool do_list(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, "list", req, res))
		return false;

	print_servers(res.data);
	return true;
}

static bool do_stat(const char* addr, const char* filepath)
{
	if (*filepath == 0)
	{
		printf("filepath null\r\n");
		return false;
	}

	stat_req_t req;
	req.cmd = "stat";
	stat_req_data_t req_data;
	req_data.path = filepath;
	req.data.push_back(req_data);

	stat_res_t res;
	if (!http_request<stat_req_t, stat_res_t>(addr, "stat", req, res))
		return false;

	print_servers(res.data);
	return true;
}

static void print_start_result(const start_res_data_t& data)
{
	printf("status: %d\r\n", data.status);
	printf("name: %s\r\n", data.name.c_str());
	printf("path: %s\r\n", 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 char* addr, const char* filepath)
{
	if (*filepath == 0)
	{
		printf("filepath null\r\n");
		return false;
	}

	start_req_t req;
	start_req_data_t req_data;
	req_data.path = filepath;
	req.data.push_back(req_data);

	start_res_t res;
	if (!http_request<start_req_t, start_res_t>(addr, "start", req, res))
		return false;

	print_start_results(res.data);
	return true;
}

static void print_stop_result(const stop_res_data_t& data)
{
	printf("status: %d\r\n", data.status);
	printf("path: %s\r\n", 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 char* addr, const char* filepath)
{
	if (*filepath == 0)
	{
		printf("filepath null\r\n");
		return false;
	}

	stop_req_t req;
	stop_req_data_t req_data;
	req_data.path = filepath;
	req.data.push_back(req_data);

	stop_res_t res;
	if (!http_request<stop_req_t, stop_res_t>(addr, "stop", req, res))
		return false;

	print_stop_results(res.data);
	return true;
}

static void print_kill_result(const kill_res_data_t& data)
{
	printf("status: %d\r\n", data.status);
	printf("path: %s\r\n", 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 char* addr, const char* filepath)
{
	if (*filepath == 0)
	{
		printf("filepath null\r\n");
		return false;
	}

	kill_req_t req;
	kill_req_data_t req_data;
	req_data.path = filepath;
	req.data.push_back(req_data);

	kill_res_t res;
	if (!http_request<kill_req_t, kill_res_t>(addr, "kill", req, res))
		return false;

	print_kill_results(res.data);
	return true;
}

static void print_reload(const reload_res_data_t& data)
{
	printf("status: %d\r\n", data.status);
	printf("proc_count: %d\r\n", data.proc_count);
	printf("proc_signaled: %d\r\n", data.proc_signaled);
	printf("path: %s\r\n", 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 bool do_reload(const char* addr, const char* filepath)
{
	if (*filepath == 0)
	{
		printf("filepath null\r\n");
		return false;
	}

	reload_req_t req;
	reload_req_data_t req_data;
	req_data.path = filepath;
	req.data.push_back(req_data);

	reload_res_t res;
	if (!http_request<reload_req_t, reload_res_t>(addr, "reload", req, res))
		return false;

	print_reload_results(res.data);
	return true;
}

static void getline(acl::string& out)
{
	const char* prompt = "master_ctl> ";
	char* ptr = readline(prompt);
	if (ptr == NULL)
	{
		printf("Bye!\r\n");
		exit(0);
	}
	out = ptr;
	out.trim_right_line();
	if (!out.empty() && !out.equal("y", false) && !out.equal("n", false))
		add_history(out.c_str());
}

static void help(void)
{
	printf("\033[1;33;40mset\033[0m: set the service's configure\r\n");
	printf("\033[1;33;40mclear\033[0m: clear the screan\r\n");
	printf("\033[1;33;40mlist\033[0m: list all the running services\r\n");
	printf("\033[1;33;40mstat\033[0m: show one service's running status\r\n");
	printf("\033[1;33;40mstart\033[0m: start one service\r\n");
	printf("\033[1;33;40mstop\033[0m: stop one service\r\n");
	printf("\033[1;33;40mkill\033[0m: kill one service\r\n");
	printf("\033[1;33;40mreload\033[0m: reload one service\r\n");
}

static struct {
	const char* cmd;
	bool (*func)(const char*, const char*);
} __actions[] = {
	{ "list",	do_list		},
	{ "stat",	do_stat		},
	{ "start",	do_start	},
	{ "stop",	do_stop		},
	{ "kill",	do_kill		},
	{ "reload",	do_reload	},

	{ 0,		0		},
};

static void run(const char* addr, const char* filepath)
{
	acl::string buf, fpath;

	if (filepath && *filepath)
		fpath = filepath;

	while (true)
	{
		getline(buf);
		if (buf.equal("quit" ,false) || buf.equal("exit", false)
			|| buf.equal("q", false))
		{
			printf("Bye!\r\n");
			break;
		}
		if (buf.empty() || buf.equal("help", false))
		{
			help();
			continue;
		}

		std::vector<acl::string>& tokens = buf.split2(" \t", true);
		acl::string cmd = tokens[0];
		cmd.lower();

		if (cmd == "clear")
		{
			rl_clear_screen(0, 0);
			printf("\r\n");
			continue;
		}

		if (cmd == "set")
		{
			if (tokens.size() == 1)
			{
				printf("\033[1;34;40musage\033[0m: "
				  "\033[1;33;40mset\033[0m configure_path\r\n");
				continue;
			}

			fpath = tokens[1];
			printf("set service to %s ok\r\n", fpath.c_str());
			continue;
		}

		if (tokens.size() >= 2)
			fpath = tokens[1];

		bool ret = false;
		int  i;
		for (i = 0; __actions[i].cmd; i++)
		{
			if (cmd == __actions[i].cmd)
			{
				ret = __actions[i].func(addr, fpath);
				break;
			}
		}

		if (__actions[i].cmd == NULL)
			help();
		else
			printf("%s: %s\r\n", cmd.c_str(), ret ? "ok" : "err");
	}
}

static void usage(const char* procname)
{
	printf("usage: %s -h[help]\r\n"
		" -s master_manage_addr[default: 127.0.0.1:8190]\r\n"
		" -f servicde_path\r\n"
		" -a cmd[list|stat|start|stop|reload]\r\n",
		procname);
}

int main(int argc, char* argv[])
{
	acl::string filepath, action, addr("127.0.0.1:8190");
	int  ch;

	while ((ch = getopt(argc, argv, "hs:f:a:")) > 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;
		default:
			usage(argv[0]);
			return 1;
		}
	}

	action.lower();

        acl::log::stdout_open(true);

	int i;
	for (i = 0; __actions[i].cmd; i++)
	{
		if (action == __actions[i].cmd)
		{
			(void) __actions[i].func(addr, filepath);
			break;
		}
	}

	if (__actions[i].cmd == NULL)
		run(addr, filepath);

	return 0;
}