@@ -0,0 +1,95 @@
|
||||
// Package environment provides strict parsing for runtime configuration.
|
||||
package environment
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
// String returns a trimmed environment value or fallback when it is empty.
|
||||
func String(name, fallback string) string {
|
||||
if value := strings.TrimSpace(os.Getenv(name)); value != "" {
|
||||
return value
|
||||
}
|
||||
return fallback
|
||||
}
|
||||
|
||||
// Required returns a non-empty environment value.
|
||||
func Required(name string) (string, error) {
|
||||
value := strings.TrimSpace(os.Getenv(name))
|
||||
if value == "" {
|
||||
return "", fmt.Errorf("%s must be set", name)
|
||||
}
|
||||
return value, nil
|
||||
}
|
||||
|
||||
// Int parses an integer environment value.
|
||||
func Int(name string, fallback int) (int, error) {
|
||||
value := strings.TrimSpace(os.Getenv(name))
|
||||
if value == "" {
|
||||
return fallback, nil
|
||||
}
|
||||
parsed, err := strconv.Atoi(value)
|
||||
if err != nil {
|
||||
return 0, fmt.Errorf("%s must be an integer: %w", name, err)
|
||||
}
|
||||
return parsed, nil
|
||||
}
|
||||
|
||||
// Float parses a floating-point environment value.
|
||||
func Float(name string, fallback float64) (float64, error) {
|
||||
value := strings.TrimSpace(os.Getenv(name))
|
||||
if value == "" {
|
||||
return fallback, nil
|
||||
}
|
||||
parsed, err := strconv.ParseFloat(value, 64)
|
||||
if err != nil {
|
||||
return 0, fmt.Errorf("%s must be a number: %w", name, err)
|
||||
}
|
||||
return parsed, nil
|
||||
}
|
||||
|
||||
// Bool parses a boolean environment value.
|
||||
func Bool(name string, fallback bool) (bool, error) {
|
||||
value := strings.TrimSpace(os.Getenv(name))
|
||||
if value == "" {
|
||||
return fallback, nil
|
||||
}
|
||||
parsed, err := strconv.ParseBool(value)
|
||||
if err != nil {
|
||||
return false, fmt.Errorf("%s must be a boolean: %w", name, err)
|
||||
}
|
||||
return parsed, nil
|
||||
}
|
||||
|
||||
// Duration parses a Go duration environment value.
|
||||
func Duration(name string, fallback time.Duration) (time.Duration, error) {
|
||||
value := strings.TrimSpace(os.Getenv(name))
|
||||
if value == "" {
|
||||
return fallback, nil
|
||||
}
|
||||
parsed, err := time.ParseDuration(value)
|
||||
if err != nil {
|
||||
return 0, fmt.Errorf("%s must be a duration: %w", name, err)
|
||||
}
|
||||
return parsed, nil
|
||||
}
|
||||
|
||||
// CSV returns non-empty, trimmed comma-separated values.
|
||||
func CSV(name string) []string {
|
||||
value := strings.TrimSpace(os.Getenv(name))
|
||||
if value == "" {
|
||||
return nil
|
||||
}
|
||||
parts := strings.Split(value, ",")
|
||||
result := make([]string, 0, len(parts))
|
||||
for _, part := range parts {
|
||||
if part = strings.TrimSpace(part); part != "" {
|
||||
result = append(result, part)
|
||||
}
|
||||
}
|
||||
return result
|
||||
}
|
||||
@@ -0,0 +1,44 @@
|
||||
package environment
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestParsers(t *testing.T) {
|
||||
t.Setenv("TEST_INT", "42")
|
||||
t.Setenv("TEST_FLOAT", "2.5")
|
||||
t.Setenv("TEST_BOOL", "true")
|
||||
t.Setenv("TEST_DURATION", "15m")
|
||||
t.Setenv("TEST_CSV", " one, two ,,three ")
|
||||
|
||||
if value, err := Int("TEST_INT", 0); err != nil || value != 42 {
|
||||
t.Fatalf("Int() = %d, %v", value, err)
|
||||
}
|
||||
if value, err := Float("TEST_FLOAT", 0); err != nil || value != 2.5 {
|
||||
t.Fatalf("Float() = %f, %v", value, err)
|
||||
}
|
||||
if value, err := Bool("TEST_BOOL", false); err != nil || !value {
|
||||
t.Fatalf("Bool() = %t, %v", value, err)
|
||||
}
|
||||
if value, err := Duration("TEST_DURATION", 0); err != nil || value != 15*time.Minute {
|
||||
t.Fatalf("Duration() = %s, %v", value, err)
|
||||
}
|
||||
values := CSV("TEST_CSV")
|
||||
if len(values) != 3 || values[1] != "two" {
|
||||
t.Fatalf("CSV() = %#v", values)
|
||||
}
|
||||
}
|
||||
|
||||
func TestInvalidValues(t *testing.T) {
|
||||
t.Setenv("TEST_VALUE", "not-valid")
|
||||
if _, err := Int("TEST_VALUE", 0); err == nil {
|
||||
t.Fatal("Int() did not return an error")
|
||||
}
|
||||
if _, err := Bool("TEST_VALUE", false); err == nil {
|
||||
t.Fatal("Bool() did not return an error")
|
||||
}
|
||||
if _, err := Duration("TEST_VALUE", 0); err == nil {
|
||||
t.Fatal("Duration() did not return an error")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user