refactor: provide context to repository to apply timeout

1 files changed, 3 insertions, 3 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)
 		}
 	})