253 lines
6.6 KiB
Go
253 lines
6.6 KiB
Go
package config
|
||
|
||
import (
|
||
"fmt"
|
||
"os"
|
||
"time"
|
||
|
||
"gopkg.in/yaml.v3"
|
||
)
|
||
|
||
// Config 应用配置
|
||
type Config struct {
|
||
Server ServerConfig `yaml:"server"`
|
||
City CityConfig `yaml:"city"`
|
||
LLM LLMConfig `yaml:"llm"`
|
||
Amap AmapConfig `yaml:"amap"`
|
||
JobAPI JobAPIConfig `yaml:"job_api"`
|
||
OCR OCRConfig `yaml:"ocr"`
|
||
Policy PolicyConfig `yaml:"policy"`
|
||
Logging LoggingConfig `yaml:"logging"`
|
||
Performance PerformanceConfig `yaml:"performance"`
|
||
}
|
||
|
||
// CityConfig 城市配置
|
||
type CityConfig struct {
|
||
Name string `yaml:"name"` // 城市名称，如：青岛
|
||
SystemName string `yaml:"system_name"` // 系统名称，如：青岛岗位匹配系统
|
||
AreaCodes map[string]string `yaml:"area_codes"` // 区域代码映射，如：市南区:0, 市北区:1
|
||
Landmarks []string `yaml:"landmarks"` // 地标示例，如：五四广场、青岛啤酒博物馆
|
||
Abbreviations map[string]string `yaml:"abbreviations"` // 简称映射，如：青啤:青岛啤酒
|
||
}
|
||
|
||
// ServerConfig 服务器配置
|
||
type ServerConfig struct {
|
||
Port int `yaml:"port"`
|
||
Host string `yaml:"host"`
|
||
ReadTimeout time.Duration `yaml:"read_timeout"`
|
||
WriteTimeout time.Duration `yaml:"write_timeout"`
|
||
}
|
||
|
||
// LLMConfig LLM配置
|
||
type LLMConfig struct {
|
||
BaseURL string `yaml:"base_url"`
|
||
APIKey string `yaml:"api_key"`
|
||
Model string `yaml:"model"`
|
||
Timeout time.Duration `yaml:"timeout"`
|
||
MaxRetries int `yaml:"max_retries"`
|
||
}
|
||
|
||
// AmapConfig 高德地图配置
|
||
type AmapConfig struct {
|
||
APIKey string `yaml:"api_key"`
|
||
BaseURL string `yaml:"base_url"`
|
||
Timeout time.Duration `yaml:"timeout"`
|
||
}
|
||
|
||
// JobAPIConfig 岗位API配置
|
||
type JobAPIConfig struct {
|
||
BaseURL string `yaml:"base_url"`
|
||
Timeout time.Duration `yaml:"timeout"`
|
||
}
|
||
|
||
// OCRConfig OCR服务配置
|
||
type OCRConfig struct {
|
||
BaseURL string `yaml:"base_url"`
|
||
Timeout time.Duration `yaml:"timeout"`
|
||
}
|
||
|
||
// PolicyConfig 政策大模型配置
|
||
type PolicyConfig struct {
|
||
BaseURL string `yaml:"base_url"`
|
||
LoginName string `yaml:"login_name"`
|
||
UserKey string `yaml:"user_key"`
|
||
ServiceID string `yaml:"service_id"`
|
||
Timeout time.Duration `yaml:"timeout"`
|
||
}
|
||
|
||
// LoggingConfig 日志配置
|
||
type LoggingConfig struct {
|
||
Level string `yaml:"level"`
|
||
Format string `yaml:"format"`
|
||
}
|
||
|
||
// PerformanceConfig 性能配置
|
||
type PerformanceConfig struct {
|
||
MaxGoroutines int `yaml:"max_goroutines"`
|
||
GoroutinePoolSize int `yaml:"goroutine_pool_size"`
|
||
TaskQueueSize int `yaml:"task_queue_size"`
|
||
EnablePprof *bool `yaml:"enable_pprof"`
|
||
EnableMetrics *bool `yaml:"enable_metrics"`
|
||
GCPercent int `yaml:"gc_percent"`
|
||
}
|
||
|
||
var globalConfig *Config
|
||
|
||
// Load 从文件加载配置
|
||
func Load(configPath string) (*Config, error) {
|
||
data, err := os.ReadFile(configPath)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("读取配置文件失败: %w", err)
|
||
}
|
||
|
||
var cfg Config
|
||
if err := yaml.Unmarshal(data, &cfg); err != nil {
|
||
return nil, fmt.Errorf("解析配置文件失败: %w", err)
|
||
}
|
||
|
||
// 环境变量覆盖（用于生产环境，避免密钥泄露）
|
||
if v := os.Getenv("LLM_API_KEY"); v != "" {
|
||
cfg.LLM.APIKey = v
|
||
}
|
||
if v := os.Getenv("LLM_BASE_URL"); v != "" {
|
||
cfg.LLM.BaseURL = v
|
||
}
|
||
if v := os.Getenv("AMAP_API_KEY"); v != "" {
|
||
cfg.Amap.APIKey = v
|
||
}
|
||
if v := os.Getenv("OCR_BASE_URL"); v != "" {
|
||
cfg.OCR.BaseURL = v
|
||
}
|
||
if v := os.Getenv("POLICY_LOGIN_NAME"); v != "" {
|
||
cfg.Policy.LoginName = v
|
||
}
|
||
if v := os.Getenv("POLICY_USER_KEY"); v != "" {
|
||
cfg.Policy.UserKey = v
|
||
}
|
||
if v := os.Getenv("POLICY_SERVICE_ID"); v != "" {
|
||
cfg.Policy.ServiceID = v
|
||
}
|
||
if v := os.Getenv("SERVER_PORT"); v != "" {
|
||
fmt.Sscanf(v, "%d", &cfg.Server.Port)
|
||
}
|
||
|
||
// 设置默认值
|
||
if cfg.Server.Port == 0 {
|
||
cfg.Server.Port = 8080
|
||
}
|
||
if cfg.Server.Host == "" {
|
||
cfg.Server.Host = "0.0.0.0"
|
||
}
|
||
if cfg.LLM.MaxRetries == 0 {
|
||
cfg.LLM.MaxRetries = 3
|
||
}
|
||
if cfg.Server.ReadTimeout == 0 {
|
||
cfg.Server.ReadTimeout = 30 * time.Second
|
||
}
|
||
if cfg.Server.WriteTimeout == 0 {
|
||
cfg.Server.WriteTimeout = 300 * time.Second
|
||
}
|
||
|
||
// 城市配置默认值
|
||
if cfg.City.Name == "" {
|
||
cfg.City.Name = "青岛"
|
||
}
|
||
if cfg.City.SystemName == "" {
|
||
cfg.City.SystemName = cfg.City.Name + "岗位匹配系统"
|
||
}
|
||
if len(cfg.City.AreaCodes) == 0 {
|
||
cfg.City.AreaCodes = map[string]string{
|
||
"市南区": "0", "市北区": "1", "李沧区": "2", "崂山区": "3", "黄岛区": "4",
|
||
"城阳区": "5", "即墨区": "6", "胶州市": "7", "平度市": "8", "莱西市": "9",
|
||
}
|
||
}
|
||
if len(cfg.City.Landmarks) == 0 {
|
||
cfg.City.Landmarks = []string{"五四广场", "青岛啤酒博物馆"}
|
||
}
|
||
if len(cfg.City.Abbreviations) == 0 {
|
||
cfg.City.Abbreviations = map[string]string{"青啤": "青岛啤酒"}
|
||
}
|
||
|
||
// 性能配置默认值
|
||
if cfg.Performance.MaxGoroutines == 0 {
|
||
cfg.Performance.MaxGoroutines = 10000
|
||
}
|
||
if cfg.Performance.GoroutinePoolSize == 0 {
|
||
cfg.Performance.GoroutinePoolSize = 5000
|
||
}
|
||
if cfg.Performance.TaskQueueSize == 0 {
|
||
cfg.Performance.TaskQueueSize = 10000
|
||
}
|
||
if cfg.Performance.GCPercent == 0 {
|
||
cfg.Performance.GCPercent = 100
|
||
}
|
||
// pprof和metrics默认启用（允许在配置文件中显式关闭）
|
||
if cfg.Performance.EnablePprof == nil {
|
||
v := true
|
||
cfg.Performance.EnablePprof = &v
|
||
}
|
||
if cfg.Performance.EnableMetrics == nil {
|
||
v := true
|
||
cfg.Performance.EnableMetrics = &v
|
||
}
|
||
|
||
globalConfig = &cfg
|
||
return &cfg, nil
|
||
}
|
||
|
||
// Get 获取全局配置
|
||
func Get() *Config {
|
||
return globalConfig
|
||
}
|
||
|
||
// GetAreaCodesDescription 获取区域代码描述字符串
|
||
func (c *CityConfig) GetAreaCodesDescription() string {
|
||
if len(c.AreaCodes) == 0 {
|
||
return ""
|
||
}
|
||
// 按代码排序输出
|
||
result := ""
|
||
for i := 0; i <= 9; i++ {
|
||
for name, code := range c.AreaCodes {
|
||
if code == fmt.Sprintf("%d", i) {
|
||
if result != "" {
|
||
result += ", "
|
||
}
|
||
result += fmt.Sprintf("%s(%s)", name, code)
|
||
break
|
||
}
|
||
}
|
||
}
|
||
return result
|
||
}
|
||
|
||
// GetLandmarksExample 获取地标示例字符串
|
||
func (c *CityConfig) GetLandmarksExample() string {
|
||
if len(c.Landmarks) == 0 {
|
||
return ""
|
||
}
|
||
result := ""
|
||
for i, landmark := range c.Landmarks {
|
||
if i > 0 {
|
||
result += "、"
|
||
}
|
||
result += landmark
|
||
}
|
||
return result
|
||
}
|
||
|
||
// GetAbbreviationsDescription 获取简称映射描述
|
||
func (c *CityConfig) GetAbbreviationsDescription() string {
|
||
if len(c.Abbreviations) == 0 {
|
||
return ""
|
||
}
|
||
result := ""
|
||
for abbr, full := range c.Abbreviations {
|
||
if result != "" {
|
||
result += "、"
|
||
}
|
||
result += fmt.Sprintf("\"%s\"指\"%s\"", abbr, full)
|
||
}
|
||
return result
|
||
}
|