fix: bigquery job statistics for rate limit (#4272)

achettyiitr · web-flow · commit 9842f64571f8 · 2024-01-03T16:44:03.000+05:30
diff --git a/warehouse/integrations/bigquery/bigquery.go b/warehouse/integrations/bigquery/bigquery.go
@@ -466,6 +466,11 @@ func (bq *BigQuery) jobStatistics(
 	)
 	bqJob, err := bqJobGetCall.Context(ctx).Location(job.Location()).Fields("statistics").Do()
 	if err != nil {
+		// In case of rate limit error, return empty statistics
+		var e *googleapi.Error
+		if errors.As(err, &e) && e.Code == 429 {
+			return &bqService.JobStatistics{}, nil
+		}
 		return nil, fmt.Errorf("getting job: %w", err)
 	}
 	return bqJob.Statistics, nil

Original file line number	Diff line number	Diff line change
`@@ -466,6 +466,11 @@ func (bq *BigQuery) jobStatistics(`
`466`	`466`	`)`
`467`	`467`	`bqJob, err := bqJobGetCall.Context(ctx).Location(job.Location()).Fields("statistics").Do()`
`468`	`468`	`if err != nil {`
	`469`	`+ // In case of rate limit error, return empty statistics`
	`470`	`+ var e *googleapi.Error`
	`471`	`+ if errors.As(err, &e) && e.Code == 429 {`
	`472`	`+ return &bqService.JobStatistics{}, nil`
	`473`	`+ }`
`469`	`474`	`return nil, fmt.Errorf("getting job: %w", err)`
`470`	`475`	`}`
`471`	`476`	`return bqJob.Statistics, nil`