Explorar o código

完善中间件

huangrf %!s(int64=5) %!d(string=hai) anos
pai
achega
a971ee66c1
Modificáronse 2 ficheiros con 10 adicións e 1 borrados
  1. 5 0
      engine/apiengine.go
  2. 5 1
      middleware/logger/logger.go

+ 5 - 0
engine/apiengine.go

@@ -2,6 +2,7 @@ package engine
 
 import (
 	"fmt"
+	"git.qianqiusoft.com/qianqiusoft/light-apiengine/middleware/logger"
 	"time"
 
 	//"github.com/gin-contrib/sessions"
@@ -51,6 +52,9 @@ func NewApiEngine(driverName, dataSourceName string, addr string) *ApiEngine {
 	engine.PlatformOrmEngine, err = xorm.NewEngine(driverName, dataSourceName)
 	printError(err)
 
+	engine.PlatformOrmEngine.ShowSQL(true)
+	engine.PlatformOrmEngine.ShowExecTime(true)
+
 	initOrmEngine(engine.PlatformOrmEngine, driverName)
 	if config.AppConfig.SyncDb {
 
@@ -59,6 +63,7 @@ func NewApiEngine(driverName, dataSourceName string, addr string) *ApiEngine {
 	engine.GinEngine = gin.Default()
 
 	engine.GinEngine.Use(gzip.Gzip(gzip.DefaultCompression))
+	engine.GinEngine.Use(logger.LoggerMiddleware())
 
 	//启动session
 	//store := sessions.NewCookieStore([]byte("secret"))

+ 5 - 1
middleware/logger/logger.go

@@ -68,6 +68,7 @@ func init(){
 
 	// 设置app名称
 	__appName = config.AppConfig.GetKey("app_name")
+	__logInfoChan = make(chan *LogInfo, 2000)
 
 	// 日志处理
 	logProcess()
@@ -103,8 +104,11 @@ func LoggerMiddleware()gin.HandlerFunc{
 		logInfo.ReqUa = c.Request.UserAgent()
 		logInfo.ReqReferer = c.Request.Referer()
 		logInfo.ReqPostData = c.Request.PostForm.Encode()
-		logInfo.ReqClientIp = c.ClientIP()
+		if logInfo.ReqPostData == ""{
+			logInfo.ReqPostData =c.Request.Form.Encode()
 
+		}
+		logInfo.ReqClientIp = c.ClientIP()
 
 		token := c.GetHeader("token")
 		if token != "" {