refactor: provide context to repository to apply timeout

2 files changed, 6 insertions, 6 deletions

diff --git a/app/controllers/sparkles/controller.go b/app/controllers/sparkles/controller.go
index 4963950..cd86cd2 100644
--- a/app/controllers/sparkles/controller.go
+++ b/app/controllers/sparkles/controller.go
@@ -31,7 +31,7 @@ func (c *Controller) MountTo(mux *http.ServeMux) {
 }
 
 func (c *Controller) Index(w http.ResponseWriter, r *http.Request) {
-	if err := serde.ToHTTP(w, r, c.db.All()); err != nil {
+	if err := serde.ToHTTP(w, r, c.db.All(r.Context())); err != nil {
 		pls.LogError(r.Context(), err)
 		w.WriteHeader(http.StatusInternalServerError)
 	}
@@ -45,7 +45,7 @@ func (c *Controller) Create(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	if err := c.db.Save(sparkle); err != nil {
+	if err := c.db.Save(r.Context(), sparkle); err != nil {
 		pls.LogError(r.Context(), err)
 		w.WriteHeader(http.StatusBadRequest)
 		return
@@ -66,7 +66,7 @@ func (c *Controller) Restore(w http.ResponseWriter, r *http.Request) {
 	log.WithFields(r.Context(), log.Fields{"sparkles": sparkles})
 
 	x.Each(sparkles, func(sparkle *domain.Sparkle) {
-		if err := c.db.Save(sparkle); err != nil {
+		if err := c.db.Save(r.Context(), sparkle); err != nil {
 			pls.LogError(r.Context(), err)
 		}
 	})
diff --git a/app/controllers/sparkles/controller_test.go b/app/controllers/sparkles/controller_test.go
index b2c7752..8a1717d 100644
--- a/app/controllers/sparkles/controller_test.go
+++ b/app/controllers/sparkles/controller_test.go
@@ -17,7 +17,7 @@ func TestSparkles(t *testing.T) {
 	t.Run("GET /sparkles", func(t *testing.T) {
 		sparkle, _ := domain.NewSparkle("@tanuki for helping me")
 		store := db.NewRepository[*domain.Sparkle]()
-		store.Save(sparkle)
+		store.Save(t.Context(), sparkle)
 
 		mux := http.NewServeMux()
 		controller := New(store)
@@ -75,8 +75,8 @@ func TestSparkles(t *testing.T) {
 				})
 
 				t.Run("saves the sparkle to the db", func(t *testing.T) {
-					assert.Equal(t, 1, len(repository.All()))
-					item := repository.All()[0]
+					assert.Equal(t, 1, len(repository.All(t.Context())))
+					item := repository.All(t.Context())[0]
 
 					assert.Equal(t, "@tanuki", item.Sparklee)
 					assert.Equal(t, "for reviewing my code!", item.Reason)