fblook/module/config/config.go
2018-09-21 22:58:41 +08:00

58 lines
825 B
Go

package config
import (
"errors"
"io/ioutil"
"os"
"path"
"git.trj.tw/golang/fblook/module/utils"
yaml "gopkg.in/yaml.v2"
)
// Config -
type Config struct {
APIURL string `yaml:"api_url"`
APIKey string `yaml:"api_key"`
}
var conf *Config
// LoadConfig -
func LoadConfig(p ...string) error {
var fp string
if len(p) > 0 && len(p[0]) > 0 {
fp = p[0]
} else {
wd, err := os.Getwd()
if err != nil {
return err
}
fp = path.Join(wd, "config.yml")
}
fp = utils.ParsePath(fp)
exists := utils.CheckExists(fp, false)
if !exists {
return errors.New("config file not exists")
}
data, err := ioutil.ReadFile(fp)
if err != nil {
return err
}
conf = &Config{}
err = yaml.Unmarshal(data, conf)
if err != nil {
return err
}
return nil
}
// GetConf -
func GetConf() *Config {
return conf
}