likeadmin/server/middleware/log.go

package middleware

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/gin-gonic/gin/binding"
	"likeadmin/config"
	"likeadmin/core"
	"likeadmin/core/response"
	"likeadmin/model/system"
	"likeadmin/util"
	"net/url"
	"strings"
	"time"
)

//requestType 请求参数类
type requestType string

const (
	RequestFile    requestType = "file"    // 文件类型
	RequestDefault requestType = "default" // 默认数据类型
)

//RecordLog 记录系统日志信息中间件
func RecordLog(title string, reqTypes ...requestType) gin.HandlerFunc {
	reqType := RequestDefault
	if len(reqTypes) > 0 {
		reqType = reqTypes[0]
	}
	return func(c *gin.Context) {
		// 开始时间
		startTime := time.Now().UnixMilli()
		// 异常信息
		errStr := ""
		var status uint8 = 1 // 1=成功, 2=失败
		args := ""
		// 请求方式
		reqMethod := c.Request.Method
		// 获取请求参数
		if reqMethod == "POST" {
			// POST请求
			if reqType == RequestFile {
				// 文件类型
				var filenames []string
				form, _ := c.MultipartForm()
				for _, files := range form.File {
					for _, file := range files {
						filenames = append(filenames, file.Filename)
					}
				}
				args = strings.Join(filenames, ",")
			} else {
				//默认类型
				var formParams map[string]interface{}
				err := c.ShouldBindBodyWith(&formParams, binding.JSON)
				if err == nil {
					val, err := util.ToolsUtil.ObjToJson(&formParams)
					if err != nil {
						core.Logger.Errorf("RecordLog POST Marshal err: err=[%+v]", err)
						panic(response.SystemError)
					}
					args = val
				}
			}
		} else if reqMethod == "GET" {
			// GET请求
			query := c.Request.URL.RawQuery
			if query != "" {
				args, _ = url.QueryUnescape(query)
			}
		}
		// 处理异常
		defer func() {
			if r := recover(); r != nil {
				errStr = fmt.Sprintf("%+v", r)
				status = 2
				// 结束时间
				endTime := time.Now().UnixMilli()
				// 执行时间(毫秒)
				taskTime := endTime - startTime
				// 获取当前的用户
				adminId := config.AdminConfig.GetAdminId(c)
				url := c.Request.URL.Path
				ip := c.ClientIP()
				method := c.HandlerName()
				err := core.DB.Create(&system.SystemLogOperate{
					AdminId: adminId, Type: reqMethod, Title: title, Ip: ip,
					Url: url, Method: method, Args: args, Error: errStr, Status: status,
					StartTime: startTime / 1000, EndTime: endTime / 1000, TaskTime: taskTime,
				}).Error
				if err != nil {
					core.Logger.Errorf("RecordLog recover Create err: err=[%+v]", err)
					panic(response.SystemError)
				}
				panic(r)
			}
		}()
		// 执行方法
		c.Next()
		// 结束时间
		endTime := time.Now().UnixMilli()
		// 执行时间(毫秒)
		taskTime := endTime - startTime
		// 获取当前的用户
		adminId := config.AdminConfig.GetAdminId(c)
		url := c.Request.URL.Path
		ip := c.ClientIP()
		method := c.HandlerName()
		err := core.DB.Create(&system.SystemLogOperate{
			AdminId: adminId, Type: reqMethod, Title: title, Ip: ip,
			Url: url, Method: method, Args: args, Error: errStr, Status: status,
			StartTime: startTime / 1000, EndTime: endTime / 1000, TaskTime: taskTime,
		}).Error
		if err != nil {
			core.Logger.Errorf("RecordLog Create err: err=[%+v]", err)
			panic(response.SystemError)
		}
	}
}