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

feat: full working max_retry logic
All checks were successful
build / build (push) Successful in 2m45s
Details

Browse Source
This commit is contained in:
d3m0k1d
2026-02-22 16:06:51 +03:00
parent 8c0cfcdbe7
commit a602207369
5 changed files with 24 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
internal/storage/requests_db.go
View File

@@ -51,12 +51,13 @@ func NewRequestsRd() (*RequestReader, error) {
}, nil
}
func (r *RequestReader) IsMaxRetryExceeded(ip string, max_retry int) (bool, error) {
func (r *RequestReader) IsMaxRetryExceeded(ip string, maxRetry int) (bool, error) {
var count int
err := r.db.QueryRow("SELECT COUNT(*) FROM requests WHERE ip = ?", ip).Scan(&count)
if err != nil {
r.logger.Error("error query count: " + err.Error())
return false, err
}
return count >= max_retry, nil
r.logger.Info("Current request count for IP", "ip", ip, "count", count, "maxRetry", maxRetry)
return count >= maxRetry, nil
}