Small BQ audit optimization (#1099)

artie-labs · Jan 2, 2025 · fb4c5e0 · fb4c5e0
1 parent 91fa452
commit fb4c5e0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/clients/bigquery/bigquery.go b/clients/bigquery/bigquery.go
@@ -111,7 +111,7 @@ func (s *Store) PrepareTemporaryTable(ctx context.Context, tableData *optimizati
 
 func (s *Store) auditStagingTable(ctx context.Context, bqTempTableID dialect.TableIdentifier, tableData *optimization.TableData) error {
 	var stagingTableRowsCount uint64
-	expectedRowCount := uint64(len(tableData.Rows()))
+	expectedRowCount := uint64(tableData.NumberOfRows())
 	// The streaming metadata does not appear right away, we'll wait up to 5s for it to appear.
 	for i := 0; i < 10; i++ {
 		time.Sleep(500 * time.Millisecond)