Skip to content

Commit a1c29e8

Browse files
committed
refactor: move to volume constants
1 parent 440ba50 commit a1c29e8

1 file changed

Lines changed: 20 additions & 18 deletions

File tree

rust/operator-binary/src/controller/build/resource/statefulset.rs

Lines changed: 20 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -54,11 +54,13 @@ use crate::{
5454
};
5555

5656
// volume names
57-
const DRUID_CONFIG_VOLUME_NAME: &str = "config";
58-
const HDFS_CONFIG_VOLUME_NAME: &str = "hdfs";
59-
const LOG_CONFIG_VOLUME_NAME: &str = "log-config";
60-
const LOG_VOLUME_NAME: &str = "log";
61-
const RW_CONFIG_VOLUME_NAME: &str = "rwconfig";
57+
stackable_operator::constant!(DRUID_CONFIG_VOLUME_NAME: VolumeName = "config");
58+
stackable_operator::constant!(HDFS_CONFIG_VOLUME_NAME: VolumeName = "hdfs");
59+
stackable_operator::constant!(LOG_CONFIG_VOLUME_NAME: VolumeName = "log-config");
60+
stackable_operator::constant!(LOG_VOLUME_NAME: VolumeName = "log");
61+
stackable_operator::constant!(RW_CONFIG_VOLUME_NAME: VolumeName = "rwconfig");
62+
63+
// volume mount directory (not a volume name)
6264
const USERDATA_MOUNTPOINT: &str = "/stackable/userdata";
6365

6466
#[derive(Snafu, Debug)]
@@ -321,8 +323,8 @@ pub fn build_rolegroup_statefulset(
321323
resolved_product_image,
322324
vector_log_config,
323325
&resource_names,
324-
&VolumeName::from_str(DRUID_CONFIG_VOLUME_NAME).expect("a valid volume name"),
325-
&VolumeName::from_str(LOG_VOLUME_NAME).expect("a valid volume name"),
326+
&DRUID_CONFIG_VOLUME_NAME,
327+
&LOG_VOLUME_NAME,
326328
EnvVarSet::new(),
327329
));
328330
}
@@ -364,10 +366,10 @@ fn add_hdfs_cm_volume_and_volume_mounts(
364366
// hdfs deep storage mount
365367
if let DeepStorageSpec::Hdfs(hdfs) = deep_storage_spec {
366368
cb_druid
367-
.add_volume_mount(HDFS_CONFIG_VOLUME_NAME, HDFS_CONFIG_DIRECTORY)
369+
.add_volume_mount(&*HDFS_CONFIG_VOLUME_NAME, HDFS_CONFIG_DIRECTORY)
368370
.context(AddVolumeMountSnafu)?;
369371
pb.add_volume(
370-
VolumeBuilder::new(HDFS_CONFIG_VOLUME_NAME)
372+
VolumeBuilder::new(&*HDFS_CONFIG_VOLUME_NAME)
371373
.with_config_map(hdfs.config_map_name.to_string())
372374
.build(),
373375
)
@@ -383,19 +385,19 @@ fn add_config_volume_and_volume_mounts(
383385
pb: &mut PodBuilder,
384386
) -> Result<()> {
385387
cb_druid
386-
.add_volume_mount(DRUID_CONFIG_VOLUME_NAME, DRUID_CONFIG_DIRECTORY)
388+
.add_volume_mount(&*DRUID_CONFIG_VOLUME_NAME, DRUID_CONFIG_DIRECTORY)
387389
.context(AddVolumeMountSnafu)?;
388390
pb.add_volume(
389-
VolumeBuilder::new(DRUID_CONFIG_VOLUME_NAME)
391+
VolumeBuilder::new(&*DRUID_CONFIG_VOLUME_NAME)
390392
.with_config_map(resource_names.role_group_config_map().to_string())
391393
.build(),
392394
)
393395
.context(AddVolumeSnafu)?;
394396
cb_druid
395-
.add_volume_mount(RW_CONFIG_VOLUME_NAME, RW_CONFIG_DIRECTORY)
397+
.add_volume_mount(&*RW_CONFIG_VOLUME_NAME, RW_CONFIG_DIRECTORY)
396398
.context(AddVolumeMountSnafu)?;
397399
pb.add_volume(
398-
VolumeBuilder::new(RW_CONFIG_VOLUME_NAME)
400+
VolumeBuilder::new(&*RW_CONFIG_VOLUME_NAME)
399401
.with_empty_dir(Some(""), None)
400402
.build(),
401403
)
@@ -411,7 +413,7 @@ fn add_log_config_volume_and_volume_mounts(
411413
pb: &mut PodBuilder,
412414
) -> Result<()> {
413415
cb_druid
414-
.add_volume_mount(LOG_CONFIG_VOLUME_NAME, LOG_CONFIG_DIRECTORY)
416+
.add_volume_mount(&*LOG_CONFIG_VOLUME_NAME, LOG_CONFIG_DIRECTORY)
415417
.context(AddVolumeMountSnafu)?;
416418

417419
let config_map = match &merged_rolegroup_config.logging.druid_container {
@@ -422,7 +424,7 @@ fn add_log_config_volume_and_volume_mounts(
422424
};
423425

424426
pb.add_volume(
425-
VolumeBuilder::new(LOG_CONFIG_VOLUME_NAME)
427+
VolumeBuilder::new(&*LOG_CONFIG_VOLUME_NAME)
426428
.with_config_map(config_map)
427429
.build(),
428430
)
@@ -437,13 +439,13 @@ fn add_log_volume_and_volume_mounts(
437439
pb: &mut PodBuilder,
438440
) -> Result<()> {
439441
cb_druid
440-
.add_volume_mount(LOG_VOLUME_NAME, STACKABLE_LOG_DIR)
442+
.add_volume_mount(&*LOG_VOLUME_NAME, STACKABLE_LOG_DIR)
441443
.context(AddVolumeMountSnafu)?;
442444
cb_prepare
443-
.add_volume_mount(LOG_VOLUME_NAME, STACKABLE_LOG_DIR)
445+
.add_volume_mount(&*LOG_VOLUME_NAME, STACKABLE_LOG_DIR)
444446
.context(AddVolumeMountSnafu)?;
445447
pb.add_volume(
446-
VolumeBuilder::new(LOG_VOLUME_NAME)
448+
VolumeBuilder::new(&*LOG_VOLUME_NAME)
447449
.with_empty_dir(
448450
Some(""),
449451
Some(product_logging::framework::calculate_log_volume_size_limit(

0 commit comments

Comments
 (0)