From 5bb7467b566437241aee92f548a2ecd8f580625c Mon Sep 17 00:00:00 2001 From: Zhenhui Xie Date: Thu, 7 Nov 2019 11:38:30 +0800 Subject: refactor: Cleanup after module config refactor (#630) --- src/modules/env_var.rs | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) (limited to 'src/modules/env_var.rs') diff --git a/src/modules/env_var.rs b/src/modules/env_var.rs index f4905fda3..733f1e169 100644 --- a/src/modules/env_var.rs +++ b/src/modules/env_var.rs @@ -1,6 +1,6 @@ use std::env; -use super::{Context, Module}; +use super::{Context, Module, SegmentConfig}; use crate::config::RootModuleConfig; use crate::configs::env_var::EnvVarConfig; @@ -25,10 +25,8 @@ pub fn module<'a>(context: &'a Context) -> Option> { } // TODO: Use native prefix and suffix instead of stacking custom ones together with env_value. - module.new_segment( - "env_var", - &format!("{}{}{}", config.prefix, env_value, config.suffix), - ); + let env_var_stacked = format!("{}{}{}", config.prefix, env_value, config.suffix); + module.create_segment("env_var", &SegmentConfig::new(&env_var_stacked)); Some(module) } -- cgit v1.2.3