first

backend/internal/modules/media/repository.go

@@ -0,0 +1,66 @@
+package media
+
+import (
+	"context"
+	"errors"
+
+	"github.com/google/uuid"
+	"gorm.io/gorm"
+)
+
+var ErrNotFound = errors.New("media not found")
+
+type Repository interface {
+	Create(ctx context.Context, m *Media) error
+	FindByID(ctx context.Context, id uuid.UUID) (*Media, error)
+	List(ctx context.Context) ([]*Media, error)
+	Update(ctx context.Context, m *Media) error
+	Delete(ctx context.Context, id uuid.UUID) error
+}
+
+type gormRepository struct {
+	db *gorm.DB
+}
+
+func NewRepository(db *gorm.DB) Repository {
+	return &gormRepository{db: db}
+}
+
+func (r *gormRepository) Create(ctx context.Context, m *Media) error {
+	return r.db.WithContext(ctx).Create(m).Error
+}
+
+func (r *gormRepository) FindByID(ctx context.Context, id uuid.UUID) (*Media, error) {
+	var m Media
+	err := r.db.WithContext(ctx).Where("id = ?", id).First(&m).Error
+	if errors.Is(err, gorm.ErrRecordNotFound) {
+		return nil, ErrNotFound
+	}
+	if err != nil {
+		return nil, err
+	}
+	return &m, nil
+}
+
+func (r *gormRepository) List(ctx context.Context) ([]*Media, error) {
+	var list []*Media
+	if err := r.db.WithContext(ctx).Order("created_at desc").Find(&list).Error; err != nil {
+		return nil, err
+	}
+	return list, nil
+}
+
+func (r *gormRepository) Update(ctx context.Context, m *Media) error {
+	return r.db.WithContext(ctx).Save(m).Error
+}
+
+func (r *gormRepository) Delete(ctx context.Context, id uuid.UUID) error {
+	res := r.db.WithContext(ctx).Delete(&Media{}, "id = ?", id)
+	if res.Error != nil {
+		return res.Error
+	}
+	if res.RowsAffected == 0 {
+		return ErrNotFound
+	}
+	return nil
+}