mirror of
https://github.com/bytedance/g3.git
synced 2026-05-05 23:41:57 +00:00
initial commit
This commit is contained in:
zhangjingqiang
commit
13716f4923
1425 changed files with 163227 additions and 0 deletions
272
g3proxy/utils/ctl/src/proc.rs
Normal file
272
g3proxy/utils/ctl/src/proc.rs
Normal file
|
|
@ -0,0 +1,272 @@
|
|||
/*
|
||||
* Copyright 2023 ByteDance and/or its affiliates.
|
||||
*
|
||||
* 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.
|
||||
*/
|
||||
|
||||
use clap::ArgMatches;
|
||||
|
||||
use g3proxy_proto::escaper_capnp::escaper_control;
|
||||
use g3proxy_proto::proc_capnp::proc_control;
|
||||
use g3proxy_proto::resolver_capnp::resolver_control;
|
||||
use g3proxy_proto::server_capnp::server_control;
|
||||
use g3proxy_proto::user_group_capnp::user_group_control;
|
||||
|
||||
use super::CommandResult;
|
||||
use crate::common::{parse_fetch_result, parse_operation_result, print_list_text};
|
||||
|
||||
pub const COMMAND_VERSION: &str = "version";
|
||||
pub const COMMAND_OFFLINE: &str = "offline";
|
||||
|
||||
pub const COMMAND_FORCE_QUIT: &str = "force-quit";
|
||||
pub const COMMAND_FORCE_QUIT_ALL: &str = "force-quit-all";
|
||||
|
||||
pub const COMMAND_LIST: &str = "list";
|
||||
|
||||
const COMMAND_LIST_ARG_RESOURCE: &str = "resource";
|
||||
const RESOURCE_VALUE_USER_GROUP: &str = "user-group";
|
||||
const RESOURCE_VALUE_RESOLVER: &str = "resolver";
|
||||
const RESOURCE_VALUE_AUDITOR: &str = "auditor";
|
||||
const RESOURCE_VALUE_ESCAPER: &str = "escaper";
|
||||
const RESOURCE_VALUE_SERVER: &str = "server";
|
||||
|
||||
pub const COMMAND_RELOAD_USER_GROUP: &str = "reload-user-group";
|
||||
pub const COMMAND_RELOAD_RESOLVER: &str = "reload-resolver";
|
||||
pub const COMMAND_RELOAD_AUDITOR: &str = "reload-auditor";
|
||||
pub const COMMAND_RELOAD_ESCAPER: &str = "reload-escaper";
|
||||
pub const COMMAND_RELOAD_SERVER: &str = "reload-server";
|
||||
|
||||
const SUBCOMMAND_ARG_NAME: &str = "name";
|
||||
|
||||
pub mod commands {
|
||||
use super::*;
|
||||
use clap::{Arg, Command};
|
||||
|
||||
pub fn version() -> Command {
|
||||
Command::new(COMMAND_VERSION)
|
||||
}
|
||||
|
||||
pub fn offline() -> Command {
|
||||
Command::new(COMMAND_OFFLINE).about("Put this daemon into offline mode")
|
||||
}
|
||||
|
||||
pub fn force_quit() -> Command {
|
||||
Command::new(COMMAND_FORCE_QUIT)
|
||||
.about("Force quit offline server with the same name")
|
||||
.arg(Arg::new(SUBCOMMAND_ARG_NAME).required(true).num_args(1))
|
||||
}
|
||||
|
||||
pub fn force_quit_all() -> Command {
|
||||
Command::new(COMMAND_FORCE_QUIT_ALL).about("Force quit all offline servers")
|
||||
}
|
||||
|
||||
pub fn list() -> Command {
|
||||
Command::new(COMMAND_LIST).arg(
|
||||
Arg::new(COMMAND_LIST_ARG_RESOURCE)
|
||||
.required(true)
|
||||
.num_args(1)
|
||||
.value_parser([
|
||||
RESOURCE_VALUE_USER_GROUP,
|
||||
RESOURCE_VALUE_RESOLVER,
|
||||
RESOURCE_VALUE_AUDITOR,
|
||||
RESOURCE_VALUE_ESCAPER,
|
||||
RESOURCE_VALUE_SERVER,
|
||||
])
|
||||
.ignore_case(true),
|
||||
)
|
||||
}
|
||||
|
||||
pub fn reload_user_group() -> Command {
|
||||
Command::new(COMMAND_RELOAD_USER_GROUP)
|
||||
.arg(Arg::new(SUBCOMMAND_ARG_NAME).required(true).num_args(1))
|
||||
}
|
||||
|
||||
pub fn reload_resolver() -> Command {
|
||||
Command::new(COMMAND_RELOAD_RESOLVER)
|
||||
.arg(Arg::new(SUBCOMMAND_ARG_NAME).required(true).num_args(1))
|
||||
}
|
||||
|
||||
pub fn reload_auditor() -> Command {
|
||||
Command::new(COMMAND_RELOAD_AUDITOR)
|
||||
.arg(Arg::new(SUBCOMMAND_ARG_NAME).required(true).num_args(1))
|
||||
}
|
||||
|
||||
pub fn reload_escaper() -> Command {
|
||||
Command::new(COMMAND_RELOAD_ESCAPER)
|
||||
.arg(Arg::new(SUBCOMMAND_ARG_NAME).required(true).num_args(1))
|
||||
}
|
||||
|
||||
pub fn reload_server() -> Command {
|
||||
Command::new(COMMAND_RELOAD_SERVER)
|
||||
.arg(Arg::new(SUBCOMMAND_ARG_NAME).required(true).num_args(1))
|
||||
}
|
||||
}
|
||||
|
||||
pub async fn version(client: &proc_control::Client) -> CommandResult<()> {
|
||||
let req = client.version_request();
|
||||
let rsp = req.send().promise.await?;
|
||||
let ver = rsp.get()?.get_version()?;
|
||||
println!("{ver}");
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub async fn offline(client: &proc_control::Client) -> CommandResult<()> {
|
||||
let req = client.offline_request();
|
||||
let rsp = req.send().promise.await?;
|
||||
parse_operation_result(rsp.get()?.get_result()?)
|
||||
}
|
||||
|
||||
pub async fn force_quit(client: &proc_control::Client, args: &ArgMatches) -> CommandResult<()> {
|
||||
let name = args.get_one::<String>(SUBCOMMAND_ARG_NAME).unwrap();
|
||||
let mut req = client.force_quit_offline_server_request();
|
||||
req.get().set_name(name);
|
||||
let rsp = req.send().promise.await?;
|
||||
parse_operation_result(rsp.get()?.get_result()?)
|
||||
}
|
||||
|
||||
pub async fn force_quit_all(client: &proc_control::Client) -> CommandResult<()> {
|
||||
let req = client.force_quit_offline_servers_request();
|
||||
let rsp = req.send().promise.await?;
|
||||
parse_operation_result(rsp.get()?.get_result()?)
|
||||
}
|
||||
|
||||
pub async fn list(client: &proc_control::Client, args: &ArgMatches) -> CommandResult<()> {
|
||||
match args
|
||||
.get_one::<String>(COMMAND_LIST_ARG_RESOURCE)
|
||||
.unwrap()
|
||||
.as_str()
|
||||
{
|
||||
RESOURCE_VALUE_USER_GROUP => list_user_group(client).await,
|
||||
RESOURCE_VALUE_RESOLVER => list_resolver(client).await,
|
||||
RESOURCE_VALUE_AUDITOR => list_auditor(client).await,
|
||||
RESOURCE_VALUE_ESCAPER => list_escaper(client).await,
|
||||
RESOURCE_VALUE_SERVER => list_server(client).await,
|
||||
_ => unreachable!(),
|
||||
}
|
||||
}
|
||||
|
||||
async fn list_user_group(client: &proc_control::Client) -> CommandResult<()> {
|
||||
let req = client.list_user_group_request();
|
||||
let rsp = req.send().promise.await?;
|
||||
print_list_text(rsp.get()?.get_result()?)
|
||||
}
|
||||
|
||||
async fn list_resolver(client: &proc_control::Client) -> CommandResult<()> {
|
||||
let req = client.list_resolver_request();
|
||||
let rsp = req.send().promise.await?;
|
||||
print_list_text(rsp.get()?.get_result()?)
|
||||
}
|
||||
|
||||
async fn list_auditor(client: &proc_control::Client) -> CommandResult<()> {
|
||||
let req = client.list_auditor_request();
|
||||
let rsp = req.send().promise.await?;
|
||||
print_list_text(rsp.get()?.get_result()?)
|
||||
}
|
||||
|
||||
async fn list_escaper(client: &proc_control::Client) -> CommandResult<()> {
|
||||
let req = client.list_escaper_request();
|
||||
let rsp = req.send().promise.await?;
|
||||
print_list_text(rsp.get()?.get_result()?)
|
||||
}
|
||||
|
||||
async fn list_server(client: &proc_control::Client) -> CommandResult<()> {
|
||||
let req = client.list_server_request();
|
||||
let rsp = req.send().promise.await?;
|
||||
print_list_text(rsp.get()?.get_result()?)
|
||||
}
|
||||
|
||||
pub async fn reload_user_group(
|
||||
client: &proc_control::Client,
|
||||
args: &ArgMatches,
|
||||
) -> CommandResult<()> {
|
||||
let name = args.get_one::<String>(SUBCOMMAND_ARG_NAME).unwrap();
|
||||
let mut req = client.reload_user_group_request();
|
||||
req.get().set_name(name);
|
||||
let rsp = req.send().promise.await?;
|
||||
parse_operation_result(rsp.get()?.get_result()?)
|
||||
}
|
||||
|
||||
pub async fn reload_resolver(
|
||||
client: &proc_control::Client,
|
||||
args: &ArgMatches,
|
||||
) -> CommandResult<()> {
|
||||
let name = args.get_one::<String>(SUBCOMMAND_ARG_NAME).unwrap();
|
||||
let mut req = client.reload_resolver_request();
|
||||
req.get().set_name(name);
|
||||
let rsp = req.send().promise.await?;
|
||||
parse_operation_result(rsp.get()?.get_result()?)
|
||||
}
|
||||
|
||||
pub async fn reload_auditor(client: &proc_control::Client, args: &ArgMatches) -> CommandResult<()> {
|
||||
let name = args.get_one::<String>(SUBCOMMAND_ARG_NAME).unwrap();
|
||||
let mut req = client.reload_auditor_request();
|
||||
req.get().set_name(name);
|
||||
let rsp = req.send().promise.await?;
|
||||
parse_operation_result(rsp.get()?.get_result()?)
|
||||
}
|
||||
|
||||
pub async fn reload_escaper(client: &proc_control::Client, args: &ArgMatches) -> CommandResult<()> {
|
||||
let name = args.get_one::<String>(SUBCOMMAND_ARG_NAME).unwrap();
|
||||
let mut req = client.reload_escaper_request();
|
||||
req.get().set_name(name);
|
||||
let rsp = req.send().promise.await?;
|
||||
parse_operation_result(rsp.get()?.get_result()?)
|
||||
}
|
||||
|
||||
pub async fn reload_server(client: &proc_control::Client, args: &ArgMatches) -> CommandResult<()> {
|
||||
let name = args.get_one::<String>(SUBCOMMAND_ARG_NAME).unwrap();
|
||||
let mut req = client.reload_server_request();
|
||||
req.get().set_name(name);
|
||||
let rsp = req.send().promise.await?;
|
||||
parse_operation_result(rsp.get()?.get_result()?)
|
||||
}
|
||||
|
||||
pub(crate) async fn get_user_group(
|
||||
client: &proc_control::Client,
|
||||
name: &str,
|
||||
) -> CommandResult<user_group_control::Client> {
|
||||
let mut req = client.get_user_group_request();
|
||||
req.get().set_name(name);
|
||||
let rsp = req.send().promise.await?;
|
||||
parse_fetch_result(rsp.get()?.get_user_group()?)
|
||||
}
|
||||
|
||||
pub(crate) async fn get_resolver(
|
||||
client: &proc_control::Client,
|
||||
name: &str,
|
||||
) -> CommandResult<resolver_control::Client> {
|
||||
let mut req = client.get_resolver_request();
|
||||
req.get().set_name(name);
|
||||
let rsp = req.send().promise.await?;
|
||||
parse_fetch_result(rsp.get()?.get_resolver()?)
|
||||
}
|
||||
|
||||
pub(crate) async fn get_escaper(
|
||||
client: &proc_control::Client,
|
||||
name: &str,
|
||||
) -> CommandResult<escaper_control::Client> {
|
||||
let mut req = client.get_escaper_request();
|
||||
req.get().set_name(name);
|
||||
let rsp = req.send().promise.await?;
|
||||
parse_fetch_result(rsp.get()?.get_escaper()?)
|
||||
}
|
||||
|
||||
pub(crate) async fn get_server(
|
||||
client: &proc_control::Client,
|
||||
name: &str,
|
||||
) -> CommandResult<server_control::Client> {
|
||||
let mut req = client.get_server_request();
|
||||
req.get().set_name(name);
|
||||
let rsp = req.send().promise.await?;
|
||||
parse_fetch_result(rsp.get()?.get_server()?)
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue