Added cleanup

api/internal/core/ports/download_request/repository.go

@@ -10,4 +10,6 @@ type Repository interface {
 	GetById(ctx context.Context, id string) (*entities.Download, error)
 	Update(ctx context.Context, download *entities.Download) error
 	Get(ctx context.Context, active bool) ([]*entities.Download, error)
+	GetOldOrStuck(ctx context.Context) ([]*entities.Download, error)
+	BatchDelete(ctx context.Context, requests []*entities.Download) error
 }

api/internal/core/ports/download_request/sql/repository.go

@@ -3,6 +3,7 @@ package sql
 import (
 	"context"
 	"downloader/internal/core/entities"
+	"downloader/internal/core/ports/download_request"
 	"github.com/uptrace/bun"
 	"go.uber.org/zap"
 )
@@ -12,7 +13,7 @@ type repository struct {
 	logger *zap.SugaredLogger
 }
 
-func NewDownloadRequestSqlRepository(db *bun.DB, logger *zap.SugaredLogger) *repository {
+func NewDownloadRequestSqlRepository(db *bun.DB, logger *zap.SugaredLogger) download_request.Repository {
 	return &repository{
 		db:     db,
 		logger: logger,
@@ -82,6 +83,7 @@ func (r repository) Get(ctx context.Context, active bool) ([]*entities.Download,
 	err := r.db.NewSelect().
 		Model(&downloads).
 		Column("id", "status", "link").
+		Where("status LIKE ?", "in-progress%").
 		Limit(20).
 		Order("created_at ASC").
 		Scan(ctx)
@@ -100,3 +102,53 @@ func (r repository) Get(ctx context.Context, active bool) ([]*entities.Download,
 
 	return responseDownloads, nil
 }
+
+func (r repository) GetOldOrStuck(ctx context.Context) ([]*entities.Download, error) {
+
+	var downloads []Download
+	err := r.db.NewSelect().
+		Model(&downloads).
+		Column("id", "status", "link").
+		Where("status LIKE ?", "in-progress%").
+		Where("updated_at < now() - interval '1 minutes'").
+		Limit(20).
+		Order("created_at ASC").
+		Scan(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	var responseDownloads []*entities.Download
+	for _, download := range downloads {
+		responseDownloads = append(responseDownloads, &entities.Download{
+			ID:     download.ID,
+			Status: download.Status,
+			Link:   download.Link,
+		})
+	}
+	if len(responseDownloads) == 0 {
+		return []*entities.Download{}, nil
+	}
+
+	return responseDownloads, nil
+}
+
+func (r repository) BatchDelete(ctx context.Context, requests []*entities.Download) error {
+	var downloads []*Download
+	for _, request := range requests {
+		downloads = append(downloads, &Download{ID: request.ID})
+	}
+	if len(downloads) == 0 {
+		return nil
+	}
+
+	_, err := r.db.NewDelete().
+		Model(&downloads).
+		WherePK().
+		Exec(ctx)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}