You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
86 lines
2.4 KiB
86 lines
2.4 KiB
package rpcx
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"regexp"
|
|
|
|
"yunyan/lego/core"
|
|
"yunyan/lego/utils/container/id"
|
|
"yunyan/lego/utils/mapstructure"
|
|
|
|
"gopkg.in/yaml.v2"
|
|
)
|
|
|
|
// 匹配 ${VAR} 与 ${VAR:-default} 两种占位符
|
|
var envVarRe = regexp.MustCompile(`\$\{([A-Za-z_][A-Za-z0-9_]*)(?::-([^}]*))?\}`)
|
|
|
|
// expandEnv 展开配置内容里的环境变量占位符，便于多服务共用同一份环境文件（docker compose env_file）：
|
|
// - ${VAR} 取环境变量 VAR，未设置时替换为空串
|
|
// - ${VAR:-default} 取环境变量 VAR，未设置或为空时使用 default
|
|
func expandEnv(content string) string {
|
|
return envVarRe.ReplaceAllStringFunc(content, func(m string) string {
|
|
sub := envVarRe.FindStringSubmatch(m)
|
|
name, def := sub[1], sub[2]
|
|
if v, ok := os.LookupEnv(name); ok && v != "" {
|
|
return v
|
|
}
|
|
return def
|
|
})
|
|
}
|
|
|
|
type RPCXConfig struct {
|
|
Port int32 //监听短裤
|
|
Addr string //对外地址
|
|
ETCDServers []string //etcd 地址
|
|
UpdateInterval int32 //保活间隔
|
|
}
|
|
|
|
type Option func(*Options)
|
|
|
|
type Options struct {
|
|
ConfPath string
|
|
Version string //服务版本
|
|
Setting core.ServiceSttings
|
|
RPCXConfig RPCXConfig
|
|
}
|
|
|
|
func SetConfPath(v string) Option {
|
|
return func(o *Options) {
|
|
o.ConfPath = v
|
|
}
|
|
}
|
|
|
|
func SetVersion(v string) Option {
|
|
return func(o *Options) {
|
|
o.Version = v
|
|
}
|
|
}
|
|
|
|
func newOptions(option ...Option) *Options {
|
|
options := &Options{
|
|
ConfPath: "conf/cluster.yaml",
|
|
}
|
|
for _, o := range option {
|
|
o(options)
|
|
}
|
|
yamlFile, err := os.ReadFile(options.ConfPath)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("读取服务配置【%s】文件失败err:%v:", options.ConfPath, err))
|
|
}
|
|
// 先做环境变量替换，再解析。这样公共配置（如 Redis 地址、key 前缀）只需在环境文件改一处。
|
|
err = yaml.Unmarshal([]byte(expandEnv(string(yamlFile))), &options.Setting)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("读取服务配置【%s】文件失败err:%v:", options.ConfPath, err))
|
|
}
|
|
if len(options.Setting.Type) == 0 || len(options.Setting.Tag) == 0 {
|
|
panic(fmt.Sprintf("[%s] 配置缺少必要配置: %+v", options.ConfPath, options))
|
|
}
|
|
if len(options.Setting.Id) == 0 {
|
|
options.Setting.Id = fmt.Sprintf("%s-%s", options.Setting.Type, id.NewXId())
|
|
}
|
|
if err = mapstructure.Decode(options.Setting.Exte, &options.RPCXConfig); err != nil {
|
|
panic(fmt.Sprintf("[%s] 配置缺少必要配置 Exte: %+v", options.ConfPath, options))
|
|
}
|
|
return options
|
|
}
|
|
|