签入版本

This commit is contained in:
小蚂蚁云
2026-10-20 18:32:22 +08:00
parent 0427c603b1
commit 0f7bf2d59c
4 changed files with 0 additions and 127 deletions
-2
View File
@@ -1,2 +0,0 @@
*
!.gitignore
Binary file not shown.
BIN
View File
Binary file not shown.
-125
View File
@@ -1,125 +0,0 @@
package config
import (
"context"
"os"
"testing"
"github.com/gogf/gf/v2/os/gcfg"
)
// fakeBase 模拟文件适配器,返回固定配置树
type fakeBase struct {
data map[string]any
}
func (f *fakeBase) Available(ctx context.Context, resource ...string) bool { return true }
func (f *fakeBase) Get(ctx context.Context, pattern string) (any, error) {
if pattern == "" || pattern == "." {
return f.data, nil
}
var cur any = f.data
for _, part := range splitPattern(pattern) {
m, ok := cur.(map[string]any)
if !ok {
return nil, nil
}
cur = m[part]
}
return cur, nil
}
func (f *fakeBase) Data(ctx context.Context) (map[string]any, error) { return f.data, nil }
func splitPattern(pattern string) []string {
var parts []string
start := 0
for i := 0; i < len(pattern); i++ {
if pattern[i] == '.' {
parts = append(parts, pattern[start:i])
start = i + 1
}
}
return append(parts, pattern[start:])
}
// TestAdapterEnvFirst 覆盖优先级核心行为:
// - 设置 GF_ 环境变量 → 覆盖文件值(含 map 整节点逐叶叠加)
// - 未设置 → 回落文件值
// - 空值环境变量 → 视为未设置,不清空文件值
func TestAdapterEnvFirst(t *testing.T) {
ctx := context.Background()
base := &fakeBase{data: map[string]any{
"database": map[string]any{
"default": map[string]any{
"type": "mysql",
"link": "file-link",
},
},
"jwt": map[string]any{"secret": "file-secret"},
}}
a := &adapterEnvFirst{base: base}
// 1. 未设置环境变量:文件值原样返回
if v, _ := a.Get(ctx, "database.default.link"); v != "file-link" {
t.Fatalf("未设环境变量时应返回文件值,得到: %v", v)
}
// 2. 标量覆盖
os.Setenv("GF_JWT_SECRET", "env-secret")
defer os.Unsetenv("GF_JWT_SECRET")
if v, _ := a.Get(ctx, "jwt.secret"); v != "env-secret" {
t.Fatalf("环境变量应覆盖文件值,得到: %v", v)
}
// 3. map 整节点逐叶叠加(gdb 读 database 整节点的场景)
os.Setenv("GF_DATABASE_DEFAULT_LINK", "env-link")
defer os.Unsetenv("GF_DATABASE_DEFAULT_LINK")
v, _ := a.Get(ctx, "database")
m, ok := v.(map[string]any)
if !ok {
t.Fatalf("database 节点应返回 map,得到: %T", v)
}
def := m["default"].(map[string]any)
if def["link"] != "env-link" {
t.Fatalf("整节点读取时 link 应被环境变量覆盖,得到: %v", def["link"])
}
if def["type"] != "mysql" {
t.Fatalf("未覆盖的键应保持文件值,得到: %v", def["type"])
}
// 4. 空值环境变量视为未设置(不清空文件值)
os.Setenv("GF_DATABASE_DEFAULT_TYPE", "")
defer os.Unsetenv("GF_DATABASE_DEFAULT_TYPE")
if v, _ := a.Get(ctx, "database.default.type"); v != "mysql" {
t.Fatalf("空环境变量应视为未设置,得到: %v", v)
}
// 5. Data 全量叠加
data, _ := a.Data(ctx)
defData := data["database"].(map[string]any)["default"].(map[string]any)
if defData["link"] != "env-link" {
t.Fatalf("Data 应叠加环境变量覆盖,得到: %v", defData["link"])
}
}
// TestRegisterEnvOverride 注册后 g.Cfg() 读取链路获得覆盖能力
func TestRegisterEnvOverride(t *testing.T) {
adapter, err := gcfg.NewAdapterContent(`database:
default:
link: file-link
`)
if err != nil {
t.Fatalf("构造内容适配器失败: %v", err)
}
gcfg.Instance().SetAdapter(adapter)
RegisterEnvOverride()
ctx := context.Background()
os.Setenv("GF_DATABASE_DEFAULT_LINK", "env-link")
defer os.Unsetenv("GF_DATABASE_DEFAULT_LINK")
if v := gcfg.Instance().MustGet(ctx, "database.default.link").String(); v != "env-link" {
t.Fatalf("注册后 g.Cfg 读取应获得环境变量覆盖,得到: %v", v)
}
}