114 lines
3.8 KiB
Go
114 lines
3.8 KiB
Go
package config
|
||
|
||
import (
|
||
"bytes"
|
||
"fmt"
|
||
"github.com/spf13/viper"
|
||
"os"
|
||
|
||
consulapi "github.com/hashicorp/consul/api"
|
||
_ "github.com/spf13/viper/remote"
|
||
"gopkg.in/yaml.v3"
|
||
)
|
||
|
||
type WebConfig struct {
|
||
Name string `yaml:"name,omitempty"`
|
||
Host string `yaml:"host,omitempty"`
|
||
Port int `yaml:"port,omitempty"`
|
||
Mode string `yaml:"mode,omitempty"`
|
||
Consul ConsulConfig `yaml:"consul,omitempty"`
|
||
Db DbConfig `yaml:"db"`
|
||
Cache CacheConfig `yaml:"cache"`
|
||
Logging LogOptions `yaml:"logging"`
|
||
}
|
||
type ConsulConfig struct {
|
||
Host string `yaml:"host,omitempty"`
|
||
Port int `yaml:"port,omitempty"`
|
||
Interval int `yaml:"interval,omitempty"`
|
||
Timeout int `yaml:"timeout,omitempty"`
|
||
Deregister int `yaml:"deregister,omitempty"`
|
||
Tags []string `yaml:"tags,omitempty"`
|
||
}
|
||
type DbConfig struct {
|
||
Conn string `yaml:"conn"`
|
||
DriveName string `yaml:"drive_name"`
|
||
}
|
||
type CacheConfig struct {
|
||
Host string `yaml:"host,omitempty"`
|
||
Port int `yaml:"port,omitempty"`
|
||
Pass string `yaml:"pass,omitempty"`
|
||
DB int `yaml:"db,omitempty"`
|
||
PoolSize int `yaml:"pool_size,omitempty"`
|
||
}
|
||
|
||
type LogOptions struct {
|
||
Path string `yaml:"path" json:"path" toml:"path"` // 文件保存地方
|
||
Prefix string `yaml:"prefix" json:"prefix" toml:"prefix"` // 日志文件前缀
|
||
ErrorFileSuffix string `yaml:"errorFileSuffix" json:"errorFileSuffix" toml:"errorFileSuffix"` // error日志文件后缀
|
||
WarnFileSuffix string `yaml:"warnFileSuffix" json:"warnFileSuffix" toml:"warnFileSuffix"` // warn日志文件后缀
|
||
InfoFileSuffix string `yaml:"infoFileSuffix" json:"infoFileSuffix" toml:"infoFileSuffix"` // info日志文件后缀
|
||
DebugFileSuffix string `yaml:"debugFileSuffix" json:"debugFileSuffix" toml:"debugFileSuffix"` // debug日志文件后缀
|
||
Level string `yaml:"level" json:"level" toml:"level"` // 日志等级
|
||
MaxSize int `yaml:"maxSize" json:"maxSize" toml:"maxSize"` // 日志文件大小（M）
|
||
MaxBackups int `yaml:"maxBackups" json:"maxBackups" toml:"maxBackups"` // 最多存在多少个切片文件
|
||
MaxAge int `yaml:"maxAge" json:"maxAge" toml:"maxAge"` // 保存的最大天数
|
||
Development bool `yaml:"development" json:"development" toml:"development"` // 是否是开发模式
|
||
}
|
||
|
||
func ParseConfigByFile(path string) (cfg *WebConfig, err error) {
|
||
buffer, err := os.ReadFile(path)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return load(buffer)
|
||
}
|
||
|
||
func load(buf []byte) (cfg *WebConfig, err error) {
|
||
cViper := viper.New()
|
||
cViper.SetConfigType("yaml")
|
||
cfg = new(WebConfig)
|
||
cViper.ReadConfig(bytes.NewBuffer(buf))
|
||
err = cViper.Unmarshal(cfg)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return
|
||
}
|
||
|
||
func UpdateLocalConfig(cfg *WebConfig, fn string) error {
|
||
data, err := yaml.Marshal(cfg)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
err = os.WriteFile(fn, data, 0600)
|
||
return err
|
||
}
|
||
|
||
func UpdateRemoteConfig(cfg *WebConfig) error {
|
||
consulClient, err := consulapi.NewClient(&consulapi.Config{Address: fmt.Sprintf("%s:%d", cfg.Consul.Host, cfg.Consul.Port)})
|
||
if err != nil {
|
||
return err
|
||
}
|
||
val, err := yaml.Marshal(cfg)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
p := &consulapi.KVPair{Key: fmt.Sprintf("hpds-pavement/hpds_web/%s/%s", cfg.Mode, cfg.Name), Value: val}
|
||
if _, err = consulClient.KV().Put(p, nil); err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func GetRemoteConfig(remoteAddr, path string) (cfg *WebConfig, err error) {
|
||
consulClient, err := consulapi.NewClient(&consulapi.Config{Address: remoteAddr})
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
kv, _, err := consulClient.KV().Get(path, nil)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return load(kv.Value)
|
||
}
|