28 lines
501 B
Go
28 lines
501 B
Go
|
package options
|
||
|
|
|
||
|
|
import (
|
||
|
|
"gitea.timerzz.com/kedaya_haitao/common/pkg/database"
|
||
|
|
"gopkg.in/yaml.v3"
|
||
|
|
"os"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Config struct {
|
||
|
|
DB database.DBOption `yaml:"db"`
|
||
|
|
Interval time.Duration `yaml:"interval"`
|
||
|
|
}
|
||
|
|
|
||
|
|
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)
|
||
|
|
}
|