40 lines
906 B
Go
40 lines
906 B
Go
|
package middleware
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/gogf/gf/v2/frame/g"
|
||
|
|
"github.com/gogf/gf/v2/net/ghttp"
|
||
|
|
"github.com/gogf/gf/v2/text/gregex"
|
||
|
|
"strconv"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
// TimestampMiddleware
|
||
|
|
//
|
||
|
|
// 全局中间件
|
||
|
|
// 检查时间戳误差是否在
|
||
|
|
func TimestampMiddleware(r *ghttp.Request) {
|
||
|
|
// 检查时间戳误差是否在
|
||
|
|
timestamp, err := strconv.ParseInt(r.GetHeader("timestamp"), 10, 64)
|
||
|
|
if gregex.IsMatch(`^[0-9]{13,14}$`, []byte(strconv.FormatInt(timestamp, 10))) {
|
||
|
|
if timestamp+int64(2000) > time.Now().UnixMilli() && timestamp-int64(2000) < time.Now().UnixMilli() {
|
||
|
|
r.Middleware.Next()
|
||
|
|
} else {
|
||
|
|
if err != nil {
|
||
|
|
r.Response.WriteJson(g.Map{
|
||
|
|
"code": 40011,
|
||
|
|
"message": "时间戳过期",
|
||
|
|
"data": nil,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
if err != nil {
|
||
|
|
r.Response.WriteJson(g.Map{
|
||
|
|
"code": 40010,
|
||
|
|
"message": "时间戳格式错误",
|
||
|
|
"data": nil,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|