78 lines
3.0 KiB
Go
78 lines
3.0 KiB
Go
|
package config
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"net"
|
|||
|
|
|
|||
|
|
"git.apinb.com/bsm-sdk/core/conf"
|
|||
|
|
"git.apinb.com/bsm-sdk/core/crypto/encipher"
|
|||
|
|
"git.apinb.com/bsm-sdk/core/env"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var (
|
|||
|
|
// Spec 全局配置实例，包含所有服务配置信息
|
|||
|
|
Spec SrvConfig
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// SrvConfig 服务配置结构体，包含所有必要的配置项
|
|||
|
|
type SrvConfig struct {
|
|||
|
|
conf.Base `yaml:",inline"` // 基础配置（端口、IP等）
|
|||
|
|
Databases *conf.DBConf `yaml:"Databases"` // 数据库配置
|
|||
|
|
MicroService *conf.MicroServiceConf `yaml:"MicroService"` // 微服务配置
|
|||
|
|
Rpc map[string]conf.RpcConf `yaml:"Rpc"` // RPC服务配置
|
|||
|
|
Gateway *conf.GatewayConf `yaml:"Gateway"` // HTTP网关配置
|
|||
|
|
Apm *conf.ApmConf `yaml:"APM"` // 应用性能监控配置
|
|||
|
|
Etcd *conf.EtcdConf `yaml:"Etcd"` // Etcd配置
|
|||
|
|
SMS map[string]*SmsConf `yaml:"SMS"` // 短信服务配置
|
|||
|
|
SMTP map[string]*SmtpConf `yaml:"SMTP"` // 邮件服务配置
|
|||
|
|
Code *codeConf `yaml:"Code"` // 验证码配置
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SmtpConf SMTP邮件服务配置
|
|||
|
|
type SmtpConf struct {
|
|||
|
|
Endpoint string `yaml:"Endpoint"` // SMTP服务器地址
|
|||
|
|
Port int `yaml:"Port"` // SMTP服务器端口
|
|||
|
|
Username string `yaml:"Username"` // 用户名
|
|||
|
|
Password string `yaml:"Password"` // 密码
|
|||
|
|
FromAddress string `yaml:"FromAddress"` // 发件人邮箱地址
|
|||
|
|
FromName string `yaml:"FromName"` // 发件人显示名称
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SmsConf 短信服务配置
|
|||
|
|
type SmsConf struct {
|
|||
|
|
Endpoint string `yaml:"Endpoint"` // 短信服务端点
|
|||
|
|
AccessKeyId string `yaml:"AccessKeyId"` // 访问密钥ID
|
|||
|
|
AccessKeySecret string `yaml:"AccessKeySecret"` // 访问密钥Secret
|
|||
|
|
Region string `yaml:"Region"` // 服务区域
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// codeConf 验证码相关配置
|
|||
|
|
type codeConf struct {
|
|||
|
|
Length int64 `yaml:"Length"` // 验证码长度
|
|||
|
|
Expire int `yaml:"Expire"` // 验证码过期时间（秒）
|
|||
|
|
MaxSentLimit int `yaml:"MaxSentLimit"` // 最大发送次数限制
|
|||
|
|
GenerateCode bool `yaml:"GenerateCode"` // 是否生成验证码
|
|||
|
|
CokeyKey string `yaml:"CokeyKey"` // 验证码密钥
|
|||
|
|
BlackListFilter []string `yaml:"BlackListFilter"` // 黑名单过滤器
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// New 初始化配置文件并进行必要的校验
|
|||
|
|
// srvKey: 服务标识符，用于加载对应的配置文件
|
|||
|
|
func New(srvKey string) {
|
|||
|
|
// 初始化配置，创建一个新的配置实例，用于服务配置
|
|||
|
|
conf.New(srvKey, &Spec)
|
|||
|
|
|
|||
|
|
// 配置校验：服务IP和端口，如果端口不合规则随机分配端口
|
|||
|
|
Spec.Port = conf.CheckPort(Spec.Port)
|
|||
|
|
Spec.BindIP = conf.CheckIP(Spec.BindIP)
|
|||
|
|
Spec.Addr = net.JoinHostPort(Spec.BindIP, Spec.Port)
|
|||
|
|
|
|||
|
|
// 配置校验：服务名称地址及监听地址不能为空
|
|||
|
|
conf.NotNil(Spec.Service, Spec.Cache)
|
|||
|
|
|
|||
|
|
// 初始化JWT加密密钥
|
|||
|
|
encipher.New(env.Runtime.JwtSecretKey)
|
|||
|
|
|
|||
|
|
// 打印服务启动信息
|
|||
|
|
conf.PrintInfo(Spec.Addr)
|
|||
|
|
}
|