From e16e986190414c18e7b8ad9b5a2e4155ba7cdd26 Mon Sep 17 00:00:00 2001
From: yirutang <yiru@google.com>
Date: Fri, 4 Jun 2021 09:51:25 -0700
Subject: [PATCH] fix: call awaitTermination when close the stream writer

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