feat(training): add TrainingDate

pkg/training/repository.go

@@ -242,3 +242,104 @@ func (r *InMemoryTrainingRepository) Delete(id TrainingID) error {
 	delete(r.trainings, id)
 	return nil
 }
+
+type TrainingDateRepository interface {
+	Create(trainingID TrainingID, trainingDate *TrainingDate) error
+	FindByID(id TrainingDateID) (*TrainingDate, error)
+	FindAll() ([]TrainingDate, error)
+	FindAllByTrainingID(trainingID TrainingID) ([]TrainingDate, error)
+	Update(trainingDate *TrainingDate) error
+	Delete(id TrainingDateID) error
+}
+
+type InMemoryTrainingDateRepository struct {
+	trainingDates   map[TrainingDateID]TrainingDate
+	trainingToDates map[TrainingID][]TrainingDateID
+
+	lock sync.RWMutex
+}
+
+func NewInMemoryTrainingDateRepository() *InMemoryTrainingDateRepository {
+	return &InMemoryTrainingDateRepository{
+		trainingDates:   make(map[TrainingDateID]TrainingDate),
+		trainingToDates: make(map[TrainingID][]TrainingDateID),
+	}
+}
+
+func (r *InMemoryTrainingDateRepository) Create(trainingID TrainingID, trainingDate *TrainingDate) error {
+	r.lock.Lock()
+	defer r.lock.Unlock()
+
+	trainingDate.ID = NewTrainingDateID()
+	trainingDate.trainingID = trainingID
+
+	r.trainingDates[trainingDate.ID] = *trainingDate
+	r.trainingToDates[trainingID] = append(r.trainingToDates[trainingID], trainingDate.ID)
+	return nil
+}
+
+func (r *InMemoryTrainingDateRepository) FindByID(id TrainingDateID) (*TrainingDate, error) {
+	r.lock.RLock()
+	defer r.lock.RUnlock()
+
+	date, ok := r.trainingDates[id]
+	if !ok {
+		return nil, ErrTrainingDateNotFound
+	}
+	return &date, nil
+}
+
+func (r *InMemoryTrainingDateRepository) FindAll() ([]TrainingDate, error) {
+	r.lock.RLock()
+	defer r.lock.RUnlock()
+
+	dates := make([]TrainingDate, len(r.trainingDates))
+	for _, date := range r.trainingDates {
+		dates = append(dates, date)
+	}
+
+	return dates, nil
+}
+
+func (r *InMemoryTrainingDateRepository) FindAllByTrainingID(trainingID TrainingID) ([]TrainingDate, error) {
+	r.lock.RLock()
+	defer r.lock.RUnlock()
+
+	dates := make([]TrainingDate, 0)
+	for _, id := range r.trainingToDates[trainingID] {
+		dates = append(dates, r.trainingDates[id])
+	}
+
+	return dates, nil
+}
+
+func (r *InMemoryTrainingDateRepository) Update(trainingDate *TrainingDate) error {
+	r.lock.Lock()
+	defer r.lock.Unlock()
+
+	oldDate := r.trainingDates[trainingDate.ID]
+	trainingDate.trainingID = oldDate.trainingID
+	r.trainingDates[trainingDate.ID] = *trainingDate
+	return nil
+}
+
+func (r *InMemoryTrainingDateRepository) Delete(id TrainingDateID) error {
+	r.lock.Lock()
+	defer r.lock.Unlock()
+
+	date, ok := r.trainingDates[id]
+	if !ok {
+		return ErrTrainingDateNotFound
+	}
+
+	delete(r.trainingDates, id)
+	dates := r.trainingToDates[date.trainingID]
+	for idx, d := range dates {
+		if d == id {
+			r.trainingToDates[date.trainingID] = append(dates[:idx], dates[idx+1:]...)
+			break
+		}
+	}
+
+	return nil
+}