Skip to content

Commit a765f81

Browse files
committed
add function for multiple callouts
1 parent 1a34af9 commit a765f81

1 file changed

Lines changed: 14 additions & 19 deletions

File tree

rust/operator-binary/src/airflow_controller.rs

Lines changed: 14 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -475,16 +475,7 @@ pub async fn reconcile_airflow(
475475
let git_sync_resources = git_sync::v1alpha2::GitSyncResources::new(
476476
&airflow.spec.cluster_config.dags_git_sync,
477477
&validated_cluster.image,
478-
&validated_rg_config
479-
.overrides
480-
.env_overrides
481-
.iter()
482-
.map(|(k, v)| EnvVar {
483-
name: k.clone(),
484-
value: Some(v.clone()),
485-
..EnvVar::default()
486-
})
487-
.collect::<Vec<_>>(),
478+
&env_vars_from_overrides(&validated_rg_config.overrides.env_overrides),
488479
&airflow.volume_mounts(),
489480
LOG_VOLUME_NAME,
490481
&validated_rg_config
@@ -635,15 +626,7 @@ async fn build_executor_template(
635626
let git_sync_resources = git_sync::v1alpha2::GitSyncResources::new(
636627
&airflow.spec.cluster_config.dags_git_sync,
637628
&validated_cluster.image,
638-
&common_config
639-
.env_overrides
640-
.iter()
641-
.map(|(k, v)| EnvVar {
642-
name: k.clone(),
643-
value: Some(v.clone()),
644-
..EnvVar::default()
645-
})
646-
.collect::<Vec<_>>(),
629+
&env_vars_from_overrides(&common_config.env_overrides),
647630
&airflow.volume_mounts(),
648631
LOG_VOLUME_NAME,
649632
&merged_executor_config
@@ -1292,3 +1275,15 @@ fn add_git_sync_resources(
12921275

12931276
Ok(())
12941277
}
1278+
1279+
/// Convert user-supplied `envOverrides` into a list of [`EnvVar`]s.
1280+
fn env_vars_from_overrides(env_overrides: &HashMap<String, String>) -> Vec<EnvVar> {
1281+
env_overrides
1282+
.iter()
1283+
.map(|(k, v)| EnvVar {
1284+
name: k.clone(),
1285+
value: Some(v.clone()),
1286+
..EnvVar::default()
1287+
})
1288+
.collect()
1289+
}

0 commit comments

Comments
 (0)