us-coach-spider/pkg/options/init.go

28 lines
536 B
Go
Raw Normal View History

2024-05-14 15:27:40 +08:00
package options
import (
"gitea.timerzz.com/kedaya_haitao/common/pkg/database"
"gitea.timerzz.com/kedaya_haitao/common/pkg/proxy"
"gopkg.in/yaml.v3"
"os"
)
type Config struct {
2024-05-14 16:17:04 +08:00
DB database.DBOption `yaml:"db"`
Proxy proxy.Option `yaml:"proxy"`
2024-05-14 15:27:40 +08:00
}
func LoadConfig() (*Config, error) {
cfgPath := os.Getenv("CONFIG_PATH")
if cfgPath == "" {
cfgPath = "/data/cfg.yaml"
}
var opt Config
f, err := os.Open(cfgPath)
if err != nil {
return nil, err
}
defer f.Close()
return &opt, yaml.NewDecoder(f).Decode(&opt)
}