Quote PK literal in backfill operation

xataio · Dec 5, 2023 · 53cdf05 · 53cdf05
1 parent b023378
commit 53cdf05
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/migrations/backfill.go b/pkg/migrations/backfill.go
@@ -54,7 +54,7 @@ func backfill(ctx context.Context, conn *sql.DB, table *schema.Table, cbs ...Cal
 type batcher struct {
 	table     *schema.Table
 	pkColumn  *schema.Column
-	lastPK    interface{}
+	lastPK    *string
 	batchSize int
 }
 
@@ -85,7 +85,7 @@ func (b *batcher) updateBatch(ctx context.Context, conn *sql.DB) error {
 func (b *batcher) buildQuery() string {
 	whereClause := ""
 	if b.lastPK != nil {
-		whereClause = fmt.Sprintf("WHERE %s > %v", pq.QuoteIdentifier(b.pkColumn.Name), b.lastPK)
+		whereClause = fmt.Sprintf("WHERE %s > %v", pq.QuoteIdentifier(b.pkColumn.Name), pq.QuoteLiteral(*b.lastPK))
 	}
 
 	return fmt.Sprintf(`