feat: add new method and for db req and add to template max retry
Some checks failed
build / build (push) Failing after 1m48s
Some checks failed
build / build (push) Failing after 1m48s
This commit is contained in:
@@ -28,3 +28,44 @@ func NewRequestsWr() (*RequestWriter, error) {
|
||||
db: db,
|
||||
}, nil
|
||||
}
|
||||
|
||||
type RequestReader struct {
|
||||
logger *logger.Logger
|
||||
db *sql.DB
|
||||
}
|
||||
|
||||
func NewRequestsRd() (*RequestReader, error) {
|
||||
db, err := sql.Open(
|
||||
"sqlite",
|
||||
buildSqliteDsn(ReqDBPath, pragmas),
|
||||
)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
db.SetMaxOpenConns(1)
|
||||
db.SetMaxIdleConns(1)
|
||||
db.SetConnMaxLifetime(0)
|
||||
return &RequestReader{
|
||||
logger: logger.New(false),
|
||||
db: db,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (r *RequestReader) IsMaxRetryExceeded(ip string, max_retry int) (bool, error) {
|
||||
row, err := r.db.Query("SELECT COUNT(*) FROM requests WHERE ip = ?", ip)
|
||||
if err != nil {
|
||||
r.logger.Error("error scan" + err.Error())
|
||||
return false, err
|
||||
}
|
||||
if row.Next() {
|
||||
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
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user