adds context to D and U experimental functions

ctrliq · Mar 27, 2024 · 05aa278 · 05aa278
1 parent 61bf1c6
commit 05aa278
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pika_psql_experimental.go b/pika_psql_experimental.go
@@ -42,14 +42,14 @@ func (b *basePsql[T]) F(keyval ...any) QuerySet[T] {
 	return b.Args(args).Filter(queries...)
 }
 
-func (b *basePsql[T]) D(x *T) error {
+func (b *basePsql[T]) D(ctx context.Context, x *T) error {
 	id := b.findID(x)
 	if id == nil {
 		return fmt.Errorf("id not found")
 	}
 
 	qs := b.F("id", id)
-	return qs.Delete(context.Background())
+	return qs.Delete(ctx)
 }
 
 func (b *basePsql[T]) Transaction(ctx context.Context) (QuerySet[T], error) {
@@ -62,12 +62,12 @@ func (b *basePsql[T]) Transaction(ctx context.Context) (QuerySet[T], error) {
 	return Q[T](ts), nil
 }
 
-func (b *basePsql[T]) U(x *T) error {
+func (b *basePsql[T]) U(ctx context.Context, x *T) error {
 	id := b.findID(x)
 	if id == nil {
 		return fmt.Errorf("id not found")
 	}
 
 	qs := b.F("id", id)
-	return qs.Update(context.Background(), x)
+	return qs.Update(ctx, x)
 }