f7589ebbb8
- 创建Repository接口定义（UserRepository、ProfileRepository、TextureRepository等） - 创建Repository接口实现 - 创建依赖注入容器（container.Container） - 改造Handler层使用依赖注入（AuthHandler、UserHandler、TextureHandler） - 创建新的路由注册方式（RegisterRoutesWithDI） - 提供main.go示例文件展示如何使用依赖注入 同时包含之前的安全修复： - CORS配置安全加固 - 头像URL验证安全修复 - JWT algorithm confusion漏洞修复 - Recovery中间件增强 - 敏感错误信息泄露修复 - 类型断言安全修复
65 lines
1.4 KiB
Go
65 lines
1.4 KiB
Go
package config
|
||
|
||
import (
|
||
"fmt"
|
||
"sync"
|
||
)
|
||
|
||
var (
|
||
// configInstance 全局配置实例
|
||
configInstance *Config
|
||
// rustFSConfigInstance 全局RustFS配置实例
|
||
rustFSConfigInstance *RustFSConfig
|
||
// once 确保只初始化一次
|
||
once sync.Once
|
||
// initError 初始化错误
|
||
initError error
|
||
)
|
||
|
||
// Init 初始化配置（线程安全，只会执行一次）
|
||
func Init() error {
|
||
once.Do(func() {
|
||
configInstance, initError = Load()
|
||
if initError != nil {
|
||
return
|
||
}
|
||
rustFSConfigInstance = &configInstance.RustFS
|
||
})
|
||
return initError
|
||
}
|
||
|
||
// GetConfig 获取配置实例（线程安全）
|
||
func GetConfig() (*Config, error) {
|
||
if configInstance == nil {
|
||
return nil, fmt.Errorf("配置未初始化，请先调用 config.Init()")
|
||
}
|
||
return configInstance, nil
|
||
}
|
||
|
||
// MustGetConfig 获取配置实例，如果未初始化则panic
|
||
func MustGetConfig() *Config {
|
||
cfg, err := GetConfig()
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
return cfg
|
||
}
|
||
|
||
// GetRustFSConfig 获取RustFS配置实例（线程安全）
|
||
func GetRustFSConfig() (*RustFSConfig, error) {
|
||
if rustFSConfigInstance == nil {
|
||
return nil, fmt.Errorf("配置未初始化，请先调用 config.Init()")
|
||
}
|
||
return rustFSConfigInstance, nil
|
||
}
|
||
|
||
// MustGetRustFSConfig 获取RustFS配置实例，如果未初始化则panic
|
||
func MustGetRustFSConfig() *RustFSConfig {
|
||
cfg, err := GetRustFSConfig()
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
return cfg
|
||
}
|
||
|