This commit is contained in:
parent
922c39f851
commit
f065bb5262
1 changed files with 20 additions and 2 deletions
|
|
@ -569,8 +569,9 @@ async fn main() {
|
||||||
.arg(
|
.arg(
|
||||||
Arg::new("partitions")
|
Arg::new("partitions")
|
||||||
.help("Partition references to analyze")
|
.help("Partition references to analyze")
|
||||||
.required(true)
|
.required(false)
|
||||||
.num_args(1..)
|
.num_args(0..)
|
||||||
|
.value_name("PARTITIONS")
|
||||||
)
|
)
|
||||||
.get_matches();
|
.get_matches();
|
||||||
|
|
||||||
|
|
@ -579,6 +580,23 @@ async fn main() {
|
||||||
.cloned()
|
.cloned()
|
||||||
.collect();
|
.collect();
|
||||||
|
|
||||||
|
// Validate arguments based on mode
|
||||||
|
match mode.as_str() {
|
||||||
|
"plan" | "mermaid" => {
|
||||||
|
if args.is_empty() {
|
||||||
|
error!("Error: Partition references are required for {} mode", mode);
|
||||||
|
eprintln!("Error: Partition references are required for {} mode", mode);
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"import_test" => {
|
||||||
|
// No partition arguments needed for test mode
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
// Unknown mode, will be handled later
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Get build event log configuration from environment variables
|
// Get build event log configuration from environment variables
|
||||||
let build_event_log_uri = env::var("DATABUILD_BUILD_EVENT_LOG").ok();
|
let build_event_log_uri = env::var("DATABUILD_BUILD_EVENT_LOG").ok();
|
||||||
let build_request_id = env::var("DATABUILD_BUILD_REQUEST_ID")
|
let build_request_id = env::var("DATABUILD_BUILD_REQUEST_ID")
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue