From fb4c5e01ec4d20d401e7306509505c73ae17def0 Mon Sep 17 00:00:00 2001 From: Robin Tang Date: Thu, 2 Jan 2025 15:50:38 -0800 Subject: [PATCH] Small BQ audit optimization (#1099) --- clients/bigquery/bigquery.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/clients/bigquery/bigquery.go b/clients/bigquery/bigquery.go index a7eae0de..2b437718 100644 --- 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)