diff --git a/google-cloud-bigquerystorage/src/main/java/com/google/cloud/bigquery/storage/v1beta2/StreamWriterV2.java b/google-cloud-bigquerystorage/src/main/java/com/google/cloud/bigquery/storage/v1beta2/StreamWriterV2.java index 5d39c04d08..99b565a53f 100644 --- a/google-cloud-bigquerystorage/src/main/java/com/google/cloud/bigquery/storage/v1beta2/StreamWriterV2.java +++ b/google-cloud-bigquerystorage/src/main/java/com/google/cloud/bigquery/storage/v1beta2/StreamWriterV2.java @@ -317,6 +317,10 @@ public void close() { } if (this.ownsBigQueryWriteClient) { this.client.close(); + try { + this.client.awaitTermination(1, TimeUnit.MINUTES); + } catch (InterruptedException ignored) { + } } }