CLean up.

artie-labs · Jun 18, 2024 · 7c75641 · 7c75641
1 parent ecd459e
commit 7c75641
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/clients/bigquery/bigquery.go b/clients/bigquery/bigquery.go
@@ -141,7 +141,7 @@ func (s *Store) putTableViaStorageWriteAPI(ctx context.Context, bqTableID TableI
 		managedwriter.WithDestinationTable(
 			managedwriter.TableParentFromParts(bqTableID.ProjectID(), bqTableID.Dataset(), bqTableID.Table()),
 		),
-		managedwriter.WithType(managedwriter.DefaultStream),
+		managedwriter.WithType(managedwriter.PendingStream),
 		managedwriter.WithSchemaDescriptor(schemaDescriptor),
 		managedwriter.EnableWriteRetries(true),
 	)
@@ -177,6 +177,10 @@ func (s *Store) putTableViaStorageWriteAPI(ctx context.Context, bqTableID TableI
 		}
 	}
 
+	if _, err = managedStream.Finalize(ctx); err != nil {
+		return fmt.Errorf("error during Finalize: %w", err)
+	}
+
 	req := &storagepb.BatchCommitWriteStreamsRequest{
 		Parent:       managedwriter.TableParentFromStreamName(managedStream.StreamName()),
 		WriteStreams: []string{managedStream.StreamName()},