123

lxrun/run.go

@@ -14,8 +14,12 @@ import (
 	"time"
 )
 
-func Run(env string, port string, fun func(), r func(router *gin.Engine)) {
+func Run(env string, port string, fun func(), r func(router *gin.Engine), zapConfig lxzap.LogConfig) {
 	_ = lxlog.InitLog()
+	err := lxzap.InitLogger(zapConfig)
+	if err != nil {
+		panic(err.Error())
+	}
 	app := gin.New()
 	// 注册zap相关中间件
 	if env == "dev" {
@@ -46,7 +50,7 @@ func Run(env string, port string, fun func(), r func(router *gin.Engine)) {
 		processed <- struct{}{}
 	}()
 	// serve
-	err := srv.ListenAndServe()
+	err = srv.ListenAndServe()
 	if http.ErrServerClosed != err {
 		log.Fatalf("server not gracefully shutdown, err :%v\n", err)
 	}