tcjlj/nezha
1
0
Fork 0
mirror of https://github.com/nezhahq/nezha.git synced 2025-02-08 12:38:13 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

update waf api

Browse Source
This commit is contained in:
uubulb 2024-12-21 19:16:28 +08:00
parent a839056d69
commit f86b4f961b
2 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
cmd/dashboard/controller/waf.go
View File

@ -15,21 +15,24 @@ import (
// @Schemes // @Schemes
// @Description List server // @Description List server
// @Tags auth required // @Tags auth required
// @Param page query uint false "Page number" // @Param limit query uint false "Page limit"
// @Param offset query uint false "Page offset"
// @Produce json // @Produce json
// @Success 200 {object} model.CommonResponse[[]model.WAFApiMock] // @Success 200 {object} model.CommonResponse[[]model.WAFApiMock]
// @Router /waf [get] // @Router /waf [get]
func listBlockedAddress(c *gin.Context) ([]*model.WAF, error) { func listBlockedAddress(c *gin.Context) ([]*model.WAF, error) {
const pageSize = 25 limit, err := strconv.ParseUint(c.Query("limit"), 10, 64)
if err != nil || limit < 1 {
page, err := strconv.ParseUint(c.Query("page"), 10, 64) limit = 25
if err != nil || page < 1 { }
page = 1
offset, err := strconv.ParseUint(c.Query("offset"), 10, 64)
if err != nil || offset < 1 {
offset = 1
} }
offset := (page - 1) * pageSize
var waf []*model.WAF var waf []*model.WAF
if err := singleton.DB.Limit(pageSize).Offset(int(offset)).Find(&waf).Error; err != nil { if err := singleton.DB.Limit(int(limit)).Offset(int(offset)).Find(&waf).Error; err != nil {
return nil, err return nil, err
} }

4
model/waf.go
View File

@ -52,8 +52,8 @@ func CheckIP(db *gorm.DB, ip string) error {
} }
var blockTimestamp uint64 var blockTimestamp uint64
result := db.Model(&WAF{}).Select("block_timestamp").Order("id desc").Where("ip = ?", ipBinary).First(&blockTimestamp) result := db.Model(&WAF{}).Select("block_timestamp").Order("id desc").Where("ip = ?", ipBinary).Limit(1).Find(&blockTimestamp)
if result.Error != nil && result.Error != gorm.ErrRecordNotFound { if result.Error != nil {
return result.Error return result.Error
} }