package config import ( "os" yaml "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"` TempPath string `yaml:"tempPath,omitempty"` Logging LogOptions `yaml:"logging"` Node HpdsNode `yaml:"node,omitempty"` Funcs []FuncConfig `yaml:"functions,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"` // 是否是开发模式 } type HpdsNode struct { Host string `yaml:"host"` Port int `yaml:"port"` Token string `yaml:"token,omitempty"` } type FuncConfig struct { Name string `yaml:"name"` DataTag uint8 `yaml:"dataTag"` MqType uint `yaml:"mqType"` //消息类型, 发布,1;订阅;2 } 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) { cfg = new(WebConfig) cfg.Funcs = make([]FuncConfig, 0) //cViper.ReadConfig(bytes.NewBuffer(buf)) err = yaml.Unmarshal(buf, cfg) //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 }