refactor: provide context to repository to apply timeout

1 files changed, 5 insertions, 4 deletions

diff --git a/app/db/in_memory_repository.go b/app/db/in_memory_repository.go
index 5d8628d..ba9ebad 100644
--- a/app/db/in_memory_repository.go
+++ b/app/db/in_memory_repository.go
@@ -1,6 +1,7 @@
 package db
 
 import (
+	"context"
 	"sort"
 
 	"github.com/xlgmokha/x/pkg/x"
@@ -18,17 +19,17 @@ func NewRepository[T domain.Entity]() domain.Repository[T] {
 	}
 }
 
-func (r *inMemoryRepository[T]) All() []T {
+func (r *inMemoryRepository[T]) All(ctx context.Context) []T {
 	return r.items
 }
 
-func (r *inMemoryRepository[T]) Find(id domain.ID) T {
-	return x.Find(r.All(), func(item T) bool {
+func (r *inMemoryRepository[T]) Find(ctx context.Context, id domain.ID) T {
+	return x.Find(r.All(ctx), func(item T) bool {
 		return item.GetID() == id
 	})
 }
 
-func (r *inMemoryRepository[T]) Save(item T) error {
+func (r *inMemoryRepository[T]) Save(ctx context.Context, item T) error {
 	if err := item.Validate(); err != nil {
 		return err
 	}