feat/BE/US-74/pengajuan-flock

internal/modules/production/project_flocks/repositories/projectflock.repository.go

@@ -0,0 +1,67 @@
+package repository
+
+import (
+	"context"
+	"errors"
+
+	"gitlab.com/mbugroup/lti-api.git/internal/common/repository"
+	entity "gitlab.com/mbugroup/lti-api.git/internal/entities"
+	"gorm.io/gorm"
+)
+
+type ProjectflockRepository interface {
+	repository.BaseRepository[entity.ProjectFlock]
+	GetAllByFlock(ctx context.Context, flockID uint) ([]entity.ProjectFlock, error)
+	GetActiveByFlock(ctx context.Context, flockID uint) (*entity.ProjectFlock, error)
+	GetMaxPeriodByFlock(ctx context.Context, flockID uint) (int, error)
+}
+
+type ProjectflockRepositoryImpl struct {
+	*repository.BaseRepositoryImpl[entity.ProjectFlock]
+}
+
+func NewProjectflockRepository(db *gorm.DB) ProjectflockRepository {
+	return &ProjectflockRepositoryImpl{
+		BaseRepositoryImpl: repository.NewBaseRepository[entity.ProjectFlock](db),
+	}
+}
+
+func (r *ProjectflockRepositoryImpl) GetAllByFlock(ctx context.Context, flockID uint) ([]entity.ProjectFlock, error) {
+	var records []entity.ProjectFlock
+	if err := r.DB().WithContext(ctx).
+		Unscoped().
+		Where("flock_id = ?", flockID).
+		Order("period ASC").
+		Find(&records).Error; err != nil {
+		return nil, err
+	}
+	return records, nil
+}
+
+func (r *ProjectflockRepositoryImpl) GetActiveByFlock(ctx context.Context, flockID uint) (*entity.ProjectFlock, error) {
+	var record entity.ProjectFlock
+	err := r.DB().WithContext(ctx).
+		Where("flock_id = ?", flockID).
+		Order("period DESC").
+		First(&record).Error
+	if errors.Is(err, gorm.ErrRecordNotFound) {
+		return nil, err
+	}
+	if err != nil {
+		return nil, err
+	}
+	return &record, nil
+}
+
+func (r *ProjectflockRepositoryImpl) GetMaxPeriodByFlock(ctx context.Context, flockID uint) (int, error) {
+	var max int
+	if err := r.DB().WithContext(ctx).
+		Model(&entity.ProjectFlock{}).
+		Unscoped().
+		Where("flock_id = ?", flockID).
+		Select("COALESCE(MAX(period), 0)").
+		Scan(&max).Error; err != nil {
+		return 0, err
+	}
+	return max, nil
+}