mirror of https://gitee.com/godoos/godoos.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
27 lines
654 B
27 lines
654 B
package middleware
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/didip/tollbooth"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func LimitHandler() gin.HandlerFunc {
|
|
lmt := tollbooth.NewLimiter(100, nil)
|
|
lmt.SetMessage("您访问过于频繁,系统安全检查认为恶意攻击。")
|
|
return func(c *gin.Context) {
|
|
httpError := tollbooth.LimitByRequest(lmt, c.Writer, c.Request)
|
|
if httpError != nil {
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 1,
|
|
"message": "您的操作太频繁,请稍后再试!",
|
|
"result": nil,
|
|
})
|
|
c.Data(httpError.StatusCode, lmt.GetMessageContentType(), []byte(httpError.Message))
|
|
c.Abort()
|
|
} else {
|
|
c.Next()
|
|
}
|
|
}
|
|
}
|
|
|