pass context to db

cenkalti · Oct 2, 2020 · 9cdef80 · 9cdef80
1 parent 0f51ffa
commit 9cdef80
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/internal/table/table.go b/internal/table/table.go
@@ -69,14 +69,14 @@ func (t *Table) Create(ctx context.Context) error {
 
 func (t *Table) Drop(ctx context.Context) error {
 	dropSQL := "DROP TABLE " + t.name
-	_, err := t.db.Exec(dropSQL)
+	_, err := t.db.ExecContext(ctx, dropSQL)
 	if err != nil {
 		if myErr, ok := err.(*mysql.MySQLError); !ok || myErr.Number != 1051 { // Unknown table
 			return err
 		}
 	}
 	dropSQL = "DROP TABLE " + t.name + "_instances"
-	_, err = t.db.Exec(dropSQL)
+	_, err = t.db.ExecContext(ctx, dropSQL)
 	if err != nil {
 		if myErr, ok := err.(*mysql.MySQLError); !ok || myErr.Number != 1051 { // Unknown table
 			return err