29 lines
552 B
Go
29 lines
552 B
Go
package dtmsvr
|
|
|
|
import (
|
|
"path/filepath"
|
|
"runtime"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
type Config struct {
|
|
Server string `json:"server"`
|
|
Rabbitmq RabbitmqConfig `json:"rabbitmq"`
|
|
}
|
|
|
|
var ServerConfig Config = Config{}
|
|
|
|
func LoadConfig() {
|
|
_, file, _, _ := runtime.Caller(0)
|
|
viper.SetConfigFile(filepath.Dir(file) + "/dtmsvr.yml")
|
|
if err := viper.ReadInConfig(); err != nil {
|
|
panic(err)
|
|
}
|
|
if err := viper.Unmarshal(&ServerConfig); err != nil {
|
|
panic(err)
|
|
}
|
|
logrus.Printf("config is: %v", ServerConfig)
|
|
}
|