d3m0k1d/BanForge
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 14 Wiki Activity

refactoring: method on reader req db
All checks were successful
build / build (push) Successful in 2m8s
Details

Browse Source
This commit is contained in:
d3m0k1d
2026-02-19 12:36:56 +03:00
parent 35a1a89baf
commit 8c0cfcdbe7
1 changed files with 4 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
internal/storage/requests_db.go
View File

@@ -52,20 +52,11 @@ func NewRequestsRd() (*RequestReader, error) {
} }
func (r *RequestReader) IsMaxRetryExceeded(ip string, max_retry int) (bool, error) { func (r *RequestReader) IsMaxRetryExceeded(ip string, max_retry int) (bool, error) {
row, err := r.db.Query("SELECT COUNT(*) FROM requests WHERE ip = ?", ip) var count int
err := r.db.QueryRow("SELECT COUNT(*) FROM requests WHERE ip = ?", ip).Scan(&count)
if err != nil { if err != nil {
r.logger.Error("error scan" + err.Error()) r.logger.Error("error query count: " + err.Error())
return false, err return false, err
} }
if row.Next() { return count >= max_retry, nil
var count int
if err := row.Scan(&count); err != nil {
r.logger.Error("error scan" + err.Error())
return false, err
}
if count >= max_retry {
return true, nil
}
}
return true, nil
} }