yanweidong
cef8b55fba
在`New`函数中增加了对配置文件不存在时的处理逻辑，如果指定的服务配置文件不存在，则尝试读取基于工作空间的配置文件。此外，在加载YAML配置之前，新增了对环境变量的替换步骤，确保配置中的环境变量能够被正确解析。 这些改动提高了配置加载过程的灵活性与适应性，使得应用能够在缺少特定服务配置的情况下也能正常启动，并且支持通过环境变量动态调整配置内容。
92 lines
2.1 KiB
Go
92 lines
2.1 KiB
Go
package conf
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"math/rand/v2"
|
|
|
|
"git.apinb.com/bsm-sdk/core/env"
|
|
"git.apinb.com/bsm-sdk/core/print"
|
|
"git.apinb.com/bsm-sdk/core/utils"
|
|
"git.apinb.com/bsm-sdk/core/vars"
|
|
yaml "gopkg.in/yaml.v3"
|
|
)
|
|
|
|
func New(srvKey string, cfg any) {
|
|
env.NewEnv()
|
|
|
|
// 设置服务键
|
|
vars.ServiceKey = srvKey
|
|
|
|
// 获取主机名
|
|
vars.HostName, _ = os.Hostname()
|
|
|
|
// 构造配置文件路径,输出配置文件信息
|
|
cfp := fmt.Sprintf("%s_%s.yaml", strings.ToLower(srvKey), env.Runtime.Mode)
|
|
cfp = filepath.Join(env.Runtime.Prefix, "etc", cfp)
|
|
|
|
// 配置文件不存在则读取Workspace配置文件
|
|
if !utils.PathExists(cfp) {
|
|
cfp = fmt.Sprintf("workspace_%s_%s.yaml", strings.ToLower(env.Runtime.Workspace), env.Runtime.Mode)
|
|
cfp = filepath.Join(env.Runtime.Prefix, "etc", cfp)
|
|
}
|
|
|
|
print.Info("[BSM - %s] Config File: %s", srvKey, cfp)
|
|
print.Info("[BSM - %s] Check Configure ...", vars.ServiceKey)
|
|
|
|
// 读取配置文件内容
|
|
yamlFile, err := os.ReadFile(cfp)
|
|
if err != nil {
|
|
log.Fatalf("ERROR: %v", err)
|
|
}
|
|
|
|
// 替换环境变量
|
|
yamlString := os.ExpandEnv(string(yamlFile))
|
|
|
|
// 检查配置文件中是否存在Service和Port字段
|
|
if !strings.Contains(yamlString, "Service:") {
|
|
log.Fatalln("ERROR: Service Not Nil", cfp)
|
|
}
|
|
|
|
// 解析YAML
|
|
err = yaml.Unmarshal(yamlFile, cfg)
|
|
if err != nil {
|
|
log.Fatalf("ERROR: %v", err)
|
|
}
|
|
}
|
|
|
|
func NotNil(values ...string) {
|
|
for _, value := range values {
|
|
if strings.TrimSpace(value) == "" {
|
|
log.Fatalln("ERROR:Must config key not nil")
|
|
}
|
|
}
|
|
}
|
|
|
|
func PrintInfo(addr string) {
|
|
print.Success("[BSM - %s] Config Check Success.", vars.ServiceKey)
|
|
print.Info("[BSM - %s] Service Name: %s", vars.ServiceKey, vars.ServiceKey)
|
|
print.Info("[BSM - %s] Runtime Mode: %s", vars.ServiceKey, env.Runtime.Mode)
|
|
}
|
|
|
|
func CheckPort(port string) string {
|
|
if port == "" {
|
|
r := rand.New(rand.NewPCG(1000, uint64(time.Now().UnixNano())))
|
|
p := r.IntN(65535-1024) + 1024 // 生成1024到65535之间的随机端口
|
|
return utils.Int2String(p)
|
|
}
|
|
return port
|
|
}
|
|
|
|
func CheckIP(ip string) string {
|
|
if ip == "" {
|
|
return utils.GetLocationIP()
|
|
}
|
|
return ip
|
|
}
|