71 lines
1.6 KiB
Go
71 lines
1.6 KiB
Go
package middleware
|
||
|
||
import (
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/eryajf-world/go-ldap-admin/config"
|
||
"github.com/eryajf-world/go-ldap-admin/model"
|
||
"github.com/eryajf-world/go-ldap-admin/public/tools"
|
||
"github.com/eryajf-world/go-ldap-admin/service/isql"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// 操作日志channel
|
||
var OperationLogChan = make(chan *model.OperationLog, 30)
|
||
|
||
func OperationLogMiddleware() gin.HandlerFunc {
|
||
return func(c *gin.Context) {
|
||
// 开始时间
|
||
startTime := time.Now()
|
||
|
||
// 处理请求
|
||
c.Next()
|
||
|
||
// 结束时间
|
||
endTime := time.Now()
|
||
|
||
// 执行耗时
|
||
timeCost := endTime.Sub(startTime).Milliseconds()
|
||
|
||
// 获取当前登录用户
|
||
var username string
|
||
ctxUser, exists := c.Get("user")
|
||
if !exists {
|
||
username = "未登录"
|
||
}
|
||
user, ok := ctxUser.(model.User)
|
||
if !ok {
|
||
username = "未登录"
|
||
}
|
||
username = user.Username
|
||
|
||
// 获取访问路径
|
||
path := strings.TrimPrefix(c.FullPath(), "/"+config.Conf.System.UrlPathPrefix)
|
||
// 请求方式
|
||
method := c.Request.Method
|
||
|
||
// 获取接口描述
|
||
api := new(model.Api)
|
||
_ = isql.Api.Find(tools.H{"path": path, "method": method}, api)
|
||
|
||
operationLog := model.OperationLog{
|
||
Username: username,
|
||
Ip: c.ClientIP(),
|
||
IpLocation: "",
|
||
Method: method,
|
||
Path: path,
|
||
Remark: api.Remark,
|
||
Status: c.Writer.Status(),
|
||
StartTime: startTime,
|
||
TimeCost: timeCost,
|
||
//UserAgent: c.Request.UserAgent(),
|
||
}
|
||
|
||
// 最好是将日志发送到rabbitmq或者kafka中
|
||
// 这里是发送到channel中，开启3个goroutine处理
|
||
OperationLogChan <- &operationLog
|
||
}
|
||
}
|