55 lines
1.7 KiB
Go
55 lines
1.7 KiB
Go
package config
|
||
|
||
import (
|
||
"fmt"
|
||
|
||
"github.com/spf13/viper"
|
||
)
|
||
|
||
// GlobalConfig 全局配置实例
|
||
var GlobalConfig Config
|
||
|
||
type Config struct {
|
||
App AppConfig `mapstructure:"app"`
|
||
Ingest IngestConfig `mapstructure:"ingest"`
|
||
}
|
||
|
||
// IngestConfig 入库管线配置：爬虫 → 后台 ingest 接口（HMAC 签名）。
|
||
// spider 不直连数据库：品牌任务经 GET /admin/internal/crawl/brands 拉取，
|
||
// 抓取结果经 POST /admin/internal/ingest 上报，由后台 worker 异步落库。
|
||
type IngestConfig struct {
|
||
Enabled bool `mapstructure:"enabled"` // 是否启用签名上报（替代直写）
|
||
Endpoint string `mapstructure:"endpoint"` // 后台 ingest 地址，如 http://localhost:8092/admin/internal/ingest
|
||
Secret string `mapstructure:"secret"` // 必须与后台 INGEST_SECRET 一致
|
||
HashIDSecret string `mapstructure:"hashid_secret"` // 必须与后台 HASHID_SECRET 一致（用于编码 brand_uid）
|
||
}
|
||
|
||
type AppConfig struct {
|
||
MaxCo int `mapstructure:"max_co"`
|
||
Debug bool `mapstructure:"debug"` // 新增 Debug 字段
|
||
}
|
||
|
||
// InitConfig 初始化并读取 YAML 配置文件
|
||
func InitConfig(cfgFile string) error {
|
||
if cfgFile != "" {
|
||
// 如果命令行指定了配置文件路径
|
||
viper.SetConfigFile(cfgFile)
|
||
} else {
|
||
// 默认在当前目录下查找 config.yaml
|
||
viper.AddConfigPath(".")
|
||
viper.SetConfigType("yaml")
|
||
viper.SetConfigName("config")
|
||
}
|
||
|
||
if err := viper.ReadInConfig(); err != nil {
|
||
return fmt.Errorf("读取配置文件失败: %w", err)
|
||
}
|
||
|
||
// 解析到结构体
|
||
if err := viper.Unmarshal(&GlobalConfig); err != nil {
|
||
return fmt.Errorf("解析配置文件结构失败: %w", err)
|
||
}
|
||
|
||
return nil
|
||
}
|