diff options
Diffstat (limited to 'shrine/utils')
| -rw-r--r-- | shrine/utils/env/functions.go | 32 | ||||
| -rw-r--r-- | shrine/utils/env/setter.go | 50 | ||||
| -rw-r--r-- | shrine/utils/meta/request.go | 4 |
3 files changed, 8 insertions, 78 deletions
diff --git a/shrine/utils/env/functions.go b/shrine/utils/env/functions.go index de03793..12b57e2 100644 --- a/shrine/utils/env/functions.go +++ b/shrine/utils/env/functions.go @@ -2,7 +2,6 @@ package env import ( "os" - "reflect" "strconv" "strings" "time" @@ -66,32 +65,11 @@ func getEnvStringSlice(key string, defaultVal []string) []string { return defaultVal } -func Defaults[T any](config *T) *T { - v := reflect.ValueOf(config) - if v.Kind() != reflect.Ptr || v.Elem().Kind() != reflect.Struct { - return config - } - - elem := v.Elem() - t := elem.Type() - newStruct := reflect.New(t) - newElem := newStruct.Elem() - - for i := range elem.NumField() { - field := newElem.Field(i) - fieldType := t.Field(i) - - if !field.CanSet() { - continue - } - - defaultVal := fieldType.Tag.Get("default") - if defaultVal == "" { - continue +func getEnvUint(key string, defaultVal uint64) uint64 { + if value := os.Getenv(key); value != "" { + if parsed, err := strconv.ParseUint(value, 10, 64); err == nil { + return parsed } - - setFieldDefault(field, defaultVal) } - - return newStruct.Interface().(*T) + return defaultVal } diff --git a/shrine/utils/env/setter.go b/shrine/utils/env/setter.go index 8b53a1d..4aad059 100644 --- a/shrine/utils/env/setter.go +++ b/shrine/utils/env/setter.go @@ -1,7 +1,6 @@ package env import ( - "os" "reflect" "strconv" "strings" @@ -32,13 +31,7 @@ func setFieldFromEnv(field reflect.Value, envKey, defaultVal string) { } func setUintField(field reflect.Value, envKey string, defaultVal uint64) { - if value := os.Getenv(envKey); value != "" { - if parsed, err := strconv.ParseUint(value, 10, 64); err == nil { - field.SetUint(parsed) - return - } - } - field.SetUint(defaultVal) + field.SetUint(getEnvUint(envKey, defaultVal)) } func setDurationField(field reflect.Value, envKey, defaultVal string) { @@ -64,44 +57,3 @@ func setSliceField(field reflect.Value, envKey, defaultVal string) { field.Set(reflect.ValueOf(result)) } } - -func setFieldDefault(field reflect.Value, defaultVal string) { - switch field.Kind() { - case reflect.String: - field.SetString(defaultVal) - case reflect.Bool: - if defaultBool, err := strconv.ParseBool(defaultVal); err == nil { - field.SetBool(defaultBool) - } - case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64: - if defaultInt, err := strconv.ParseInt(defaultVal, 10, 64); err == nil { - field.SetInt(defaultInt) - } - case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64: - if defaultUint, err := strconv.ParseUint(defaultVal, 10, 64); err == nil { - field.SetUint(defaultUint) - } - case reflect.Float32, reflect.Float64: - if defaultFloat, err := strconv.ParseFloat(defaultVal, 64); err == nil { - field.SetFloat(defaultFloat) - } - case reflect.Slice: - if field.Type().Elem().Kind() == reflect.String && defaultVal != "" { - parts := strings.Split(defaultVal, ",") - result := make([]string, 0, len(parts)) - for _, part := range parts { - trimmed := strings.TrimSpace(part) - if trimmed != "" { - result = append(result, trimmed) - } - } - field.Set(reflect.ValueOf(result)) - } - default: - if field.Type() == reflect.TypeFor[time.Duration]() { - if defaultDuration, err := time.ParseDuration(defaultVal); err == nil { - field.Set(reflect.ValueOf(defaultDuration)) - } - } - } -} diff --git a/shrine/utils/meta/request.go b/shrine/utils/meta/request.go index d2ae242..7e499e7 100644 --- a/shrine/utils/meta/request.go +++ b/shrine/utils/meta/request.go @@ -7,10 +7,10 @@ import ( "github.com/gofiber/fiber/v2" ) -const requestKey = "__request_ctx" +const RequestKey = "__request_ctx" func Request(c *fiber.Ctx) facade { - req, ok := c.Locals(requestKey).(types.Request) + req, ok := c.Locals(RequestKey).(types.Request) if !ok { logger.Errorf("META", "RequestContext missing in fiber locals") return facade{} |
