43 lines
849 B
Go
43 lines
849 B
Go
|
package lib
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/json"
|
|||
|
|
"io/ioutil"
|
|||
|
|
"path"
|
|||
|
|
|
|||
|
|
"gopkg.in/yaml.v2"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type callback func(interface{})
|
|||
|
|
|
|||
|
|
var loaders = map[string]func([]byte, interface{}) error{
|
|||
|
|
".json": LoadConfigFormJsonBytes,
|
|||
|
|
".yaml": LoadConfigFromYamlBytes,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func LoadConfigFormJsonBytes(content []byte, obj interface{}) error {
|
|||
|
|
return json.Unmarshal(content, obj)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func LoadConfigFromYamlBytes(content []byte, obj interface{}) error {
|
|||
|
|
return yaml.Unmarshal(content, obj)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func LoadConfig(file string, v interface{}, callback ...callback) {
|
|||
|
|
content, err := ioutil.ReadFile(file)
|
|||
|
|
|
|||
|
|
if err != nil {
|
|||
|
|
panic("Load Config Error " + err.Error())
|
|||
|
|
}
|
|||
|
|
loader, ok := loaders[path.Ext(file)]
|
|||
|
|
|
|||
|
|
if !ok {
|
|||
|
|
panic("Unknown File Type：" + path.Ext(file))
|
|||
|
|
}
|
|||
|
|
if err = loader(content, v); err == nil {
|
|||
|
|
for _, _callback := range callback {
|
|||
|
|
_callback(v)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|