fix: call awaitTermination when close the stream writer (#1111)

googleapis · Jun 5, 2021 · 0f7db50 · 0f7db50
1 parent 4a95b95
commit 0f7db50
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/...gquerystorage/src/main/java/com/google/cloud/bigquery/storage/v1beta2/StreamWriterV2.java b/...gquerystorage/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) {
+      }
     }
   }