From bd65db24182ede90a8e1849409e635543e5e3116 Mon Sep 17 00:00:00 2001 From: Jonas Kaninda Date: Wed, 5 Feb 2025 07:39:52 +0100 Subject: [PATCH] chore: update helper func to check env with prefix or suffix for multi backups --- pkg/config.go | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/pkg/config.go b/pkg/config.go index 349d6c6..7573c60 100644 --- a/pkg/config.go +++ b/pkg/config.go @@ -145,15 +145,26 @@ func getDatabase(database Database) *dbConfig { // Helper function to get environment variable or use a default value func getEnvOrDefault(currentValue, envKey, suffix, defaultValue string) string { + // Return the current value if it's already set if currentValue != "" { return currentValue } + + // Check for suffixed or prefixed environment variables if a suffix is provided if suffix != "" { - envSuffix := os.Getenv(fmt.Sprintf("%s_%s", envKey, strings.ToUpper(suffix))) + suffixUpper := strings.ToUpper(suffix) + envSuffix := os.Getenv(fmt.Sprintf("%s_%s", envKey, suffixUpper)) if envSuffix != "" { return envSuffix } + + envPrefix := os.Getenv(fmt.Sprintf("%s_%s", suffixUpper, envKey)) + if envPrefix != "" { + return envPrefix + } } + + // Fall back to the default value using a helper function return utils.EnvWithDefault(envKey, defaultValue) }