hpds_jkw_web/config/config.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)
}