From 2f6c24652e53a4ced02c2c82ed61db6c296770a2 Mon Sep 17 00:00:00 2001
From: Vojtech Mares <iam@vojtechmares.com>
Date: Sat, 4 May 2024 17:39:38 +0200
Subject: [PATCH] refactor(training): Update should just get the struct

---
 pkg/training/repository.go | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/pkg/training/repository.go b/pkg/training/repository.go
index 986a72e..d3e7974 100644
--- a/pkg/training/repository.go
+++ b/pkg/training/repository.go
@@ -4,7 +4,7 @@ type TrainingRepository interface {
 	Create(training *Training) error
 	FindByID(id TrainingID) (*Training, error)
 	FindAll() ([]Training, error)
-	Update(id TrainingID, training *Training) error
+	Update(training *Training) error
 	Delete(id TrainingID) error
 }
 
@@ -68,8 +68,8 @@ func (r *InMemoryTrainingRepository) FindAll() ([]Training, error) {
 	return trainings, nil
 }
 
-func (r *InMemoryTrainingRepository) Update(id TrainingID, training *Training) error {
-	r.trainings[id] = *training
+func (r *InMemoryTrainingRepository) Update(training *Training) error {
+	r.trainings[training.ID] = *training
 	return nil
 }