From 0f7db50bdd8f653f8087550cb2d1eeb47975eadf Mon Sep 17 00:00:00 2001
From: Yiru Tang <yiru@google.com>
Date: Sat, 5 Jun 2021 07:38:16 -0700
Subject: [PATCH] fix: call awaitTermination when close the stream writer
 (#1111)

---
 .../google/cloud/bigquery/storage/v1beta2/StreamWriterV2.java | 4 ++++
 1 file changed, 4 insertions(+)

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) {
+      }
     }
   }