ecoppens/colmena
1
0
Fork 0
forked from DGNum/colmena
Code Pull requests Activity

apply-local: Allow overriding the node name

Browse source
This commit is contained in:
Zhaofeng Li 2021-02-05 02:20:08 -08:00
parent 6e909477ae
commit 406e5a6443
1 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/command/apply_local.rs
View file

@ -23,6 +23,10 @@ pub fn subcommand() -> App<'static, 'static> {
.long("sudo")
.help("Attempt to escalate privileges if not run as root")
.takes_value(false))
.arg(Arg::with_name("node")
.long("node")
.help("Override the node name to use")
.takes_value(true))
.arg(Arg::with_name("we-are-launched-by-sudo")
.long("we-are-launched-by-sudo")
.hidden(true)
@ -60,8 +64,12 @@ pub async fn run(_global_args: &ArgMatches<'_>, local_args: &ArgMatches<'_>) {
}
let hive = util::hive_from_args(local_args).unwrap();
let hostname = hostname::get().expect("Could not get hostname")
.to_string_lossy().into_owned();
let hostname = if local_args.is_present("node") {
local_args.value_of("node").unwrap().to_owned()
} else {
hostname::get().expect("Could not get hostname")
.to_string_lossy().into_owned()
};
let goal = DeploymentGoal::from_str(local_args.value_of("goal").unwrap()).unwrap();
log::info!("Enumerating nodes...");