Append trailing underscore to prefixes if not present

Cloudment · Jan 27, 2025 · 6f2f65d · 6f2f65d
1 parent 8d0b592
commit 6f2f65d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/env/options.go b/env/options.go
@@ -88,8 +88,16 @@ func (opts Options) getRawEnv(s string) string {
 //   - A new Options struct with the prefix set.
 //
 // See: https://pkg.go.dev/reflect#StructField
+//
+// Note: If a trailing underscore is not present, it will append one.
 func (opts Options) withPrefix(sf reflect.StructField) Options {
 	opts.Prefix = opts.Prefix + sf.Tag.Get(PrefixEnv)
+
+	// Append an underscore if it's not already there.
+	if len(opts.Prefix) > 0 && opts.Prefix[len(opts.Prefix)-1] != '_' {
+		opts.Prefix = opts.Prefix + "_"
+	}
+
 	return opts
 }