设置环境

config/config.go

@@ -26,6 +26,10 @@ func GetServerPort() int {
 	return viper.GetInt("port")
 }
 
+func GetEnvironment() string {
+	return viper.GetString("ENVIRONMENT")
+}
+
 // // MustLoadLogger 初始化日志
 // func MustLoadLogger() zlog.Writer {
 // 	var (

websocket.go

@@ -34,6 +34,7 @@ func main() {
 func newApp() *gin.Engine {
 	app := gin.Default()
 	ginpprof.Wrap(app)
+	gin.SetMode(config.GetEnvironment())
 
 	id := serverId()
 	srv := &server.Server{