feat: add new method and for db req and add to template max retry

internal/storage/requests_db.go

@@ -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
+}

internal/storage/writer_test.go

@@ -299,21 +299,3 @@ func (w *RequestWriter) CreateTable() error {
 	w.logger.Info("Created requests table")
 	return nil
 }
-
-func (w *RequestWriter) Close() error {
-	w.logger.Info("Closing request database connection")
-	err := w.db.Close()
-	if err != nil {
-		return err
-	}
-	return nil
-}
-
-func (w *RequestWriter) GetRequestCount() (int, error) {
-	var count int
-	err := w.db.QueryRow("SELECT COUNT(*) FROM requests").Scan(&count)
-	if err != nil {
-		return 0, err
-	}
-	return count, nil
-}