goweb/middleware/request.go

package middleware

import (
	"bytes"
	"github.com/gin-gonic/gin"
	"goweb/core"
	"goweb/model"
	"goweb/repository"
	"io/ioutil"
	"net/http"
	"time"
)

// CheckDeviceId 检查是否存在设备指纹
func CheckDeviceId() gin.HandlerFunc {
	return func(ctx *gin.Context) {
		p := ctx.Request.URL.Path
		if p == "/" {
			ctx.Next()
			return
		}
		// 取出设备指纹
		deviceId := ctx.Request.Header.Get("di")
		if deviceId == "" {
			// 设备指纹不存在
			core.R(ctx).FailWithMessageAndCode("请求不合法", http.StatusBadRequest)
			ctx.Abort()
			return
		}
		ctx.Next()
	}
}

// SaveRequestLog 保存所有请求接口日志
func SaveRequestLog() gin.HandlerFunc {
	return func(ctx *gin.Context) {
		// 开始时间
		start := time.Now()
		// 取出接口
		path := ctx.Request.URL.Path
		// 取出参数
		query := ctx.Request.URL.RawQuery
		// body参数
		body, err := ctx.GetRawData()
		bodyStr := ""
		if err == nil {
			bodyStr = string(body)
			ctx.Request.Body = ioutil.NopCloser(bytes.NewBuffer(body))
		}
		// form参数
		form := ctx.Request.Form.Encode()
		// 取出请求方式
		method := ctx.Request.Method
		// 取出设备指纹
		deviceId := ctx.Request.Header.Get("di")
		// 取出IP
		ip := ctx.ClientIP()
		// UA
		ua := ctx.Request.UserAgent()
		ctx.Next()
		// 计算耗时
		cost := time.Since(start).Milliseconds()
		// 组装实体
		rlog := model.RequestLog{
			Method:     method,
			Path:       path,
			Param:      query,
			Body:       bodyStr,
			Form:       form,
			Cost:       cost,
			StatusCode: ctx.Writer.Status(),
			DeviceId:   deviceId,
			Ip:         ip,
			UserAgent:  ua,
		}

		//global.Log.Debugf("接口请求结果: %v", rlog)
		rlr := repository.NewRequestLogRepository()
		rlr.SaveLog(&rlog)
	}
}