126 lines
3.5 KiB
Go
126 lines
3.5 KiB
Go
package config
|
|
|
|
import (
|
|
"strings"
|
|
"sync/atomic"
|
|
)
|
|
|
|
// publishedConfig 是旧配置模型的只读兼容快照。
|
|
// 新业务应通过 CurrentConfig 读取，禁止在请求处理中直接持有并修改 Instance。
|
|
var publishedConfig atomic.Pointer[Config]
|
|
var hasPublishedConfig atomic.Bool
|
|
var setupPort atomic.Int64
|
|
|
|
func IsSetupReady() bool {
|
|
value := CurrentConfig()
|
|
if value != nil && value.SetupState.Valid() {
|
|
return value.SetupState == SetupStateReady
|
|
}
|
|
return IsInit
|
|
}
|
|
|
|
// PublishConfigSnapshot 发布一份深拷贝，供运行时只读访问。
|
|
// 调用方仍可继续使用旧的 Instance 兼容路径，但不能把返回值当作可变全局状态。
|
|
func PublishConfigSnapshot(value *Config) {
|
|
hasPublishedConfig.Store(true)
|
|
if value == nil {
|
|
publishedConfig.Store(nil)
|
|
return
|
|
}
|
|
clone := cloneConfig(value)
|
|
publishedConfig.Store(clone)
|
|
}
|
|
|
|
// CurrentConfig 返回当前运行时配置的独立副本。
|
|
// 返回值可以安全地在业务代码中读取或局部修改，不会改变全局配置快照。
|
|
func CurrentConfig() *Config {
|
|
if value := publishedConfig.Load(); value != nil {
|
|
return cloneConfig(value)
|
|
}
|
|
if hasPublishedConfig.Load() {
|
|
return nil
|
|
}
|
|
if Instance == nil {
|
|
return nil
|
|
}
|
|
return cloneConfig(Instance)
|
|
}
|
|
|
|
// CurrentHTTPConfig 返回当前进程实际使用的 HTTP 配置。
|
|
// 分区配置存在时优先读取分区配置，旧版配置仅作为兼容回退。
|
|
func CurrentHTTPConfig() HTTPConfig {
|
|
if manager := CurrentManager(); manager != nil {
|
|
runtime := manager.Snapshot()
|
|
if runtime.Generation != 0 {
|
|
return cloneHTTPConfig(runtime.Root.HTTP)
|
|
}
|
|
}
|
|
cfg := CurrentConfig()
|
|
if cfg == nil {
|
|
return HTTPConfig{}
|
|
}
|
|
mode := strings.ToLower(strings.TrimSpace(cfg.TLSMode))
|
|
return HTTPConfig{
|
|
Port: cfg.HttpPort,
|
|
HTTPSPort: cfg.HttpsPort,
|
|
HTTPSEnabled: cfg.HttpsEnabled != 2,
|
|
TLSMode: mode,
|
|
TrustedProxyCIDRs: append([]string(nil), cfg.TrustedProxyCIDRs...),
|
|
}
|
|
}
|
|
|
|
// CurrentWebDomain 返回当前进程实际使用的 Web 域名。
|
|
// 分区配置存在时优先读取分区配置，旧版配置仅作为兼容回退。
|
|
func CurrentWebDomain() string {
|
|
if manager := CurrentManager(); manager != nil {
|
|
runtime := manager.Snapshot()
|
|
if runtime.Generation != 0 {
|
|
return runtime.Root.Domains.Web
|
|
}
|
|
}
|
|
cfg := CurrentConfig()
|
|
if cfg == nil {
|
|
return ""
|
|
}
|
|
return cfg.WebDomain
|
|
}
|
|
|
|
// SetRuntimeSetupPort stores the temporary setup listener port outside the persisted configuration.
|
|
func SetRuntimeSetupPort(port int) {
|
|
setupPort.Store(int64(port))
|
|
}
|
|
|
|
func RuntimeSetupPort() int {
|
|
return int(setupPort.Load())
|
|
}
|
|
|
|
func cloneConfig(source *Config) *Config {
|
|
if source == nil {
|
|
return nil
|
|
}
|
|
clone := *source
|
|
clone.Domains = append([]string(nil), source.Domains...)
|
|
clone.AccessHosts = append([]string(nil), source.AccessHosts...)
|
|
clone.TrustedProxyCIDRs = append([]string(nil), source.TrustedProxyCIDRs...)
|
|
clone.PluginAllowList = append([]string(nil), source.PluginAllowList...)
|
|
if source.Tables != nil {
|
|
clone.Tables = make(map[string]string, len(source.Tables))
|
|
for key, value := range source.Tables {
|
|
clone.Tables[key] = value
|
|
}
|
|
}
|
|
if source.TablesInitData != nil {
|
|
clone.TablesInitData = make(map[string]string, len(source.TablesInitData))
|
|
for key, value := range source.TablesInitData {
|
|
clone.TablesInitData[key] = value
|
|
}
|
|
}
|
|
return &clone
|
|
}
|
|
|
|
func cloneHTTPConfig(source HTTPConfig) HTTPConfig {
|
|
clone := source
|
|
clone.TrustedProxyCIDRs = append([]string(nil), source.TrustedProxyCIDRs...)
|
|
return clone
|
|
}
|