mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-30 06:16:37 +08:00
696809899e
实现 core.ConfigSource:按 CONFIG_PATH 或向上查找定位 config.yaml,缺文件 降级为纯环境变量来源,坏文件返回错误而非 log.Fatalf,并把 key 命中与"设为 零值"区分开来。viper 依赖被隔离在此包,内核保持零具体运行时依赖。
127 lines
3.5 KiB
Go
127 lines
3.5 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// Package config adapts viper to the kernel configuration source contract. It is a
|
|
// runtime adapter rather than a core.Plugin: it owns no routes, services or tasks and
|
|
// therefore never appears in app.Use. Keeping viper here preserves the micro-kernel
|
|
// rule against importing concrete runtime dependencies.
|
|
package config
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io/fs"
|
|
"os"
|
|
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
// DefaultFileName is the configuration file looked up when CONFIG_PATH is unset.
|
|
const DefaultFileName = "config.yaml"
|
|
|
|
// EnvOnlyOrigin is reported by Describe when no configuration file was loaded.
|
|
const EnvOnlyOrigin = "<env only>"
|
|
|
|
// maxSearchDepth bounds the upward directory walk so a misconfigured working directory
|
|
// cannot make the loader scan the whole filesystem.
|
|
const maxSearchDepth = 5
|
|
|
|
// Option configures a Source.
|
|
type Option func(*Source)
|
|
|
|
// WithPath pins the configuration file, bypassing CONFIG_PATH and the upward search.
|
|
func WithPath(path string) Option {
|
|
return func(s *Source) {
|
|
s.path = path
|
|
}
|
|
}
|
|
|
|
// Source implements core.ConfigSource over a configuration file plus the process environment.
|
|
type Source struct {
|
|
v *viper.Viper
|
|
path string
|
|
found bool
|
|
}
|
|
|
|
// NewSource loads the configuration file. A missing file is not an error: the source
|
|
// then serves environment values only, matching the behaviour the previous pkg/config
|
|
// loader had for deployments that configure everything through the environment.
|
|
func NewSource(opts ...Option) (*Source, error) {
|
|
s := &Source{}
|
|
for _, opt := range opts {
|
|
opt(s)
|
|
}
|
|
|
|
if s.path == "" {
|
|
s.path = os.Getenv("CONFIG_PATH")
|
|
}
|
|
if s.path == "" {
|
|
s.path = findConfigPath(DefaultFileName)
|
|
}
|
|
|
|
v := viper.New()
|
|
v.SetConfigFile(s.path)
|
|
|
|
err := v.ReadInConfig()
|
|
switch {
|
|
case err == nil:
|
|
s.found = true
|
|
case isNotFound(err):
|
|
// No file: fall through to environment-only lookups.
|
|
default:
|
|
if _, statErr := os.Stat(s.path); statErr == nil { //nolint:gosec // s.path comes from CONFIG_PATH or a bounded upward search
|
|
return nil, fmt.Errorf("infra/config: read %s: %w", s.path, err)
|
|
}
|
|
}
|
|
|
|
s.v = v
|
|
return s, nil
|
|
}
|
|
|
|
// isNotFound reports whether the loader failed only because the file is absent.
|
|
func isNotFound(err error) bool {
|
|
var notFound viper.ConfigFileNotFoundError
|
|
return errors.As(err, ¬Found) || errors.Is(err, fs.ErrNotExist)
|
|
}
|
|
|
|
// Lookup returns the raw value stored at a dotted path, or false when the file was not
|
|
// loaded or the path is absent. Declared defaults therefore stay distinguishable from
|
|
// values explicitly set to a zero.
|
|
func (s *Source) Lookup(path string) (any, bool) {
|
|
if !s.found || !s.v.IsSet(path) {
|
|
return nil, false
|
|
}
|
|
return s.v.Get(path), true
|
|
}
|
|
|
|
// LookupEnv reads a process environment variable.
|
|
func (s *Source) LookupEnv(name string) (string, bool) {
|
|
return os.LookupEnv(name)
|
|
}
|
|
|
|
// Describe returns the loaded file path, or EnvOnlyOrigin when running on environment values.
|
|
func (s *Source) Describe() string {
|
|
if !s.found {
|
|
return EnvOnlyOrigin
|
|
}
|
|
return s.path
|
|
}
|
|
|
|
// findConfigPath searches upward from the working directory so tests and binaries run
|
|
// from backend/ still find the repository-root configuration file.
|
|
func findConfigPath(configPath string) string {
|
|
if _, err := os.Stat(configPath); err == nil {
|
|
return configPath
|
|
}
|
|
|
|
dir := "."
|
|
for range maxSearchDepth {
|
|
dir += "/.."
|
|
path := dir + "/" + configPath
|
|
if _, err := os.Stat(path); err == nil {
|
|
return path
|
|
}
|
|
}
|
|
return configPath
|
|
}
|