83 lines
1.5 KiB
Go
83 lines
1.5 KiB
Go
|
package tools
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"gopkg.in/yaml.v2"
|
|||
|
|
"io"
|
|||
|
|
"log"
|
|||
|
|
"os"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
//自定义选项
|
|||
|
|
|
|||
|
|
type Option struct {
|
|||
|
|
Filters []struct {
|
|||
|
|
Filter7D string `yaml:"filter7d"`
|
|||
|
|
Filter14D string `yaml:"filter14d"`
|
|||
|
|
} `yaml:"filters"`
|
|||
|
|
Macros []string `yaml:"macros"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//自定义文件服务器地址
|
|||
|
|
|
|||
|
|
type ServerConf struct {
|
|||
|
|
Server []string `yaml:"server"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewOptionObj() *Option {
|
|||
|
|
return &Option{}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewServerConfObj() *ServerConf {
|
|||
|
|
return &ServerConf{}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (ServerConf) ReadYaml(yamlFile string) *ServerConf {
|
|||
|
|
file, err := os.Open(yamlFile)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Printf("打开yaml文件错误，错误是：%v", err)
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
defer file.Close()
|
|||
|
|
|
|||
|
|
data, err := io.ReadAll(file)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Printf("读取yaml文件错误，错误是：%v", err)
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
var serverconf ServerConf
|
|||
|
|
|
|||
|
|
err = yaml.Unmarshal(data, &serverconf)
|
|||
|
|
|
|||
|
|
if err != nil {
|
|||
|
|
log.Printf("yaml文件内容反序列化失败，错误是：%v", err)
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return &serverconf
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (Option) ReadYaml(yamlFile string) *Option {
|
|||
|
|
file, err := os.Open(yamlFile)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Printf("打开yaml文件错误，错误是：%v", err)
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
defer file.Close()
|
|||
|
|
|
|||
|
|
data, err := io.ReadAll(file)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Printf("读取yaml文件错误，错误是：%v", err)
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
var option Option
|
|||
|
|
|
|||
|
|
err = yaml.Unmarshal(data, &option)
|
|||
|
|
|
|||
|
|
if err != nil {
|
|||
|
|
log.Printf("yaml文件内容反序列化失败，错误是：%v", err)
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return &option
|
|||
|
|
}
|