package configcenter import ( "fmt" "io/ioutil" "github.com/pkg/errors" "gopkg.in/yaml.v3" ) // 默认配置 type DefaultConfig struct { Log LoggerConfig `json:"log" yaml:"log"` // 日志配置 App AppConfig `json:"app" yaml:"app"` // 服务配置 } // 日志配置 type LoggerConfig struct { StdoutEnable bool `json:"stdout_enable" yaml:"stdout_enable"` // 是否允许控制台输出 StdoutLevel string `json:"stdout_level" yaml:"stdout_level"` // 控制台输出日志等级 FileEnable bool `json:"file_enable" yaml:"file_enable"` // 是否允许写文件 FileLevel string `json:"file_level" yaml:"file_level"` // 文件输出日志等级 FileAbsPath string `json:"file_abs_path" yaml:"file_abs_path"` // 日志文件绝对路径 } // 服务配置 type AppConfig struct { AppName string `json:"app_name" yaml:"app_name"` // 服务名 Ip string `json:"ip" yaml:"ip"` // ip Port int `json:"port" yaml:"port"` // 端口 } // mongo配置 type MongoConfig struct { Uri string `json:"uri" yaml:"uri"` // 实例地址 Username string `json:"username" yaml:"username"` // 用户名 Password string `json:"password" yaml:"password"` // 密码 MaxPoolSize uint64 `json:"max_pool_size" yaml:"max_pool_size"` // 最大连接池 ConnectionTimeoutMs int64 `json:"connection_timeout_ms" yaml:"connection_timeout_ms"` // 连接超时ms SocketTimeoutMs int64 `json:"socket_timeout_ms" yaml:"socket_timeout_ms"` // socket连接超时ms } // mysql配置 type MysqlConfig struct { Uri string `json:"uri" yaml:"uri"` // 实例地址,ip+端口 Username string `json:"username" yaml:"username"` // 用户名 Password string `json:"password" yaml:"password"` // 密码 TimeoutS int `json:"timeout" yaml:"timeout"` // 超时,秒 ReadTimeoutS int `json:"read_timeout_s" yaml:"read_timeout_s"` // 读超时,秒 WriteTimeoutS int `json:"write_timeout_s" yaml:"write_timeout_s"` // 写超时,秒 } // redis配置 type RedisConfig struct { Uri string `json:"uri" yaml:"uri"` // 实例地址,ip+端口 Password string `json:"password" yaml:"password"` // 鉴权密码 MaxActive int `json:"max_active" yaml:"max_active"` // 最大活动连接数,值为0时表示不限制 MaxIdle int `json:"max_idle" yaml:"max_idle"` // 最大空闲连接数 IdleTimeout int `json:"idle_timeout" yaml:"idle_timeout"` // 空闲连接的超时时间,超过该时间则关闭连接。单位为秒。默认值是5分钟。值为0时表示不关闭空闲连接。此值应该总是大于redis服务的超时时间。 Prefix string `json:"prefix" yaml:"prefix"` // key的前缀 } // 微信支付客户端配置 type WxpayClientConfig struct { MchId string `json:"mchid" yaml:"mchid"` // 商户id AppId string `json:"appid" yaml:"appid"` // appid SerialNo string `json:"serial_no" yaml:"serial_no"` // 商户API证书的证书序列号 ApiV3Key string `json:"apiv3_key" yaml:"apiv3_key"` // APIv3Key,商户平台获取 PrivateKeyPath string `json:"private_key_path" yaml:"private_key_path"` // 商户API证书下载后,私钥 apiclient_key.pem 读取后的字符串内容 NotifyUrl string `json:"notify_url" yaml:"notify_url"` // 回调地址 AppSecret string `json:"app_secret" yaml:"app_secret"` // } // 账号相关验密配置 type CryptoConfig struct { *AESConfig `json:"aes" yaml:"aes"` //AES *RSAConfig `json:"rsa" yaml:"rsa"` //RSA *TokenConfig `json:"token" yaml:"token"` //token签名 *SHA256Config `json:"sha256" yaml:"sha256"` //SHA256 } // 支付宝客户端配置 type AlipayClientConfig struct { Appid string `json:"appid" yaml:"appid"` // 商户id PrivateKey string `json:"private_key" yaml:"private_key"` // 私钥 NotifyUrl string `json:"notify_url" yaml:"notify_url"` // 回调地址 AppCertPath string `json:"app_cert_path" yaml:"app_cert_path"` AlipayRootCertPath string `json:"alipay_root_cert_path" yaml:"alipay_root_cert_path"` AlipayPublicCertPath string `json:"alipay_public_cert_path" yaml:"alipay_public_cert_path"` } // Apollo配置 type ApolloConfig struct { AppID string `json:"app_id" yaml:"app_id"` Cluster string `json:"cluster" yaml:"cluster"` IP string `json:"ip" yaml:"ip"` NamespaceName string `json:"namespace_name" yaml:"namespace_name"` Secret string `json:"secret" yaml:"secret"` IsBackUpConfig bool `json:"is_back_up_config" yaml:"is_back_up_config"` SyncServerTimeout int `json:"sync_server_timeout" yaml:"sync_server_timeout"` } // 媒体服务配置 type MediaConfig struct { AccessKey string `json:"access_key" yaml:"access_key"` AccessSecret string `json:"access_secret" yaml:"access_secret"` Endpoint string `json:"endpoint" yaml:"endpoint"` Bucket string `json:"bucket" yaml:"bucket"` CallbackUrl string `json:"callback_url" yaml:"callback_url"` ExpireTime int64 `json:"expire_time" yaml:"expire_time"` } // 阿里云短信服务配置 type DysmsapiConfig struct { AccessKeyId string `json:"access_key_id" yaml:"access_key_id"` AccessKeySecret string `json:"access_key_secret" yaml:"access_key_secret"` SignName string `json:"sign_name" yaml:"sign_name"` TemplateCode string `json:"template_code" yaml:"template_code"` } // 图片内容审核服务配置 type ImageAuditConfig struct { AccessKeyId string `json:"access_key_id" yaml:"access_key_id"` AccessKeySecret string `json:"access_key_secret" yaml:"access_key_secret"` Scenes string `json:"scenes" yaml:"scenes"` TaskBufferSize int `json:"task_buffer_size" yaml:"task_buffer_size"` TaskPacketSize int `json:"task_packet_size" yaml:"task_packet_size"` } // 文字内容审核服务配置 type TextAuditConfig struct { Labels string `json:"labels" yaml:"labels"` TaskBufferSize int `json:"task_buffer_size" yaml:"task_buffer_size"` TaskPacketSize int `json:"task_packet_size" yaml:"task_packet_size"` } // 视频内容审核服务配置 type VideoModerationConfig struct { AccessKeyId string `json:"access_key_id" yaml:"access_key_id"` AccessKeySecret string `json:"access_key_secret" yaml:"access_key_secret"` NotifyUrl string `json:"notify_url" yaml:"notify_url"` AliAcctId string `json:"ali_acct_id" yaml:"ali_acct_id"` Seed string `json:"seed" bson:"seed"` } // 服务器信息配置 type ServerInfoConfig struct { FileServerDomainName string `json:"file_server_domain_name" yaml:"file_server_domain_name"` } // 钉钉机器人信息配置 type DingTalkRobotConfig struct { AccessToken string `json:"access_token" yaml:"access_token"` Secret string `json:"secret" yaml:"secret"` } // xxl-job服务配置 type XxlJobConfig struct { ServerAddr string `json:"server_addr" yaml:"server_addr"` AccessToken string `json:"access_token" yaml:"access_token"` ExecutorIp string `json:"executor_ip" yaml:"executor_ip"` ExecutorPort string `json:"executor_port" yaml:"executor_port"` RegistryKey string `json:"registry_key" yaml:"registry_key"` LogPath string `json:"log_path" yaml:"log_path"` } // elasticsearch服务配置 type ElasticSearchConfig struct { Uri string `json:"uri" yaml:"uri"` // 实例地址 Username string `json:"username" yaml:"username"` // 用户名 Password string `json:"password" yaml:"password"` // 密码 Sniff bool `json:"sniff" yaml:"sniff"` // sniffer } // 服务发现 type ConsulServiceDiscovery struct { Host string `json:"host" yaml:"host"` ServiceName string `json:"service_name" yaml:"service_name"` } func LoadConfig(configFilePath string, cfg interface{}) error { cfgStr, err := ioutil.ReadFile(configFilePath) if err != nil { msg := fmt.Sprintf("LoadConfig fail, path: %v, err: %v", configFilePath, err) err = errors.New(msg) return err } err = yaml.Unmarshal(cfgStr, cfg) if err != nil { msg := fmt.Sprintf("yaml Unmarshal fail, path: %v, err: %v", configFilePath, err) err = errors.New(msg) return err } return nil }