FIX[BE]: period and adjustment helper to function

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

@@ -7,6 +7,7 @@ import (
 	"gitlab.com/mbugroup/lti-api.git/internal/common/repository"
 	entity "gitlab.com/mbugroup/lti-api.git/internal/entities"
 	"gorm.io/gorm"
+	"gorm.io/gorm/clause"
 )
 
 type ProjectflockRepository interface {
@@ -14,6 +15,7 @@ type ProjectflockRepository interface {
 	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)
+	GetNextPeriodForFlock(ctx context.Context, flockID uint) (int, error)
 }
 
 type ProjectflockRepositoryImpl struct {
@@ -64,3 +66,23 @@ func (r *ProjectflockRepositoryImpl) GetMaxPeriodByFlock(ctx context.Context, fl
 	}
 	return max, nil
 }
+
+func (r *ProjectflockRepositoryImpl) GetNextPeriodForFlock(ctx context.Context, flockID uint) (int, error) {
+	var payload struct {
+		Period int
+	}
+	if err := r.DB().WithContext(ctx).
+		Model(&entity.ProjectFlock{}).
+		Where("flock_id = ?", flockID).
+		Clauses(clause.Locking{Strength: "UPDATE"}).
+		Order("period DESC").
+		Limit(1).
+		Select("period").
+		Scan(&payload).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return 1, nil
+		}
+		return 0, err
+	}
+	return payload.Period + 1, nil
+}