connect/config/config.go

package config

import (
	"log"

	"github.com/spf13/viper"
	"sikey.com/websocket/pkg/confx"
)

var Websocket websocket

func loadWebsocketConfig() error {
	return viper.UnmarshalKey("websocket", &Websocket)
}

// MustLoadConfig 加载配置
func MustLoadConfig(file string) {
	if err := confx.LoadConfig(file, loadWebsocketConfig); err != nil {
		log.Fatalln(err)
	}
	log.Printf("Load configs [ toml ] from path [%s] \n", file)
}

func GetEnvironment() string {
	return viper.GetString("environment")
}