package config import ( "os" "git.hpds.cc/Component/logging" yaml "gopkg.in/yaml.v3" ) type FolderMonitorConfig struct { Name string `yaml:"name" json:"name" toml:"name"` MonitorPath string `json:"monitorPath" yaml:"monitorPath" toml:"monitorPath"` TempPath string `json:"tempPath" yaml:"tempPath" toml:"tempPath"` DataType int `json:"dataType" yaml:"dataType" toml:"dataType"` //1:数据集;2:病害库 DatasetId int `yaml:"datasetId" json:"datasetId" toml:"datasetId"` Logging LogOptions `yaml:"logging" json:"logging" tom:"logging"` Node HpdsNode `yaml:"node,omitempty" json:"node,omitempty" toml:"node,omitempty"` Funcs []FuncConfig `yaml:"functions,omitempty" json:"funcs,omitempty" toml:"funcs,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 *FolderMonitorConfig, err error) { buffer, err := os.ReadFile(path) if err != nil { return nil, err } return load(buffer) } func load(buf []byte) (cfg *FolderMonitorConfig, err error) { //cViper := viper.New() //cViper.SetConfigType("yaml") cfg = new(FolderMonitorConfig) cfg.Funcs = make([]FuncConfig, 0) err = yaml.Unmarshal(buf, cfg) return } func LoadLoggerConfig(opt LogOptions) *logging.Logger { return logging.NewLogger( logging.SetPath(opt.Path), logging.SetPrefix(opt.Prefix), logging.SetDevelopment(opt.Development), logging.SetDebugFileSuffix(opt.DebugFileSuffix), logging.SetWarnFileSuffix(opt.WarnFileSuffix), logging.SetErrorFileSuffix(opt.ErrorFileSuffix), logging.SetInfoFileSuffix(opt.InfoFileSuffix), logging.SetMaxAge(opt.MaxAge), logging.SetMaxBackups(opt.MaxBackups), logging.SetMaxSize(opt.MaxSize), logging.SetLevel(logging.LogLevel["debug"]), ) }