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

yirutang · web-flow · commit 0f7db50bdd8f · 2021-06-05T10:38:16.000-04:00
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
@@ -317,6 +317,10 @@ public void close() {
     }
     if (this.ownsBigQueryWriteClient) {
       this.client.close();
+      try {
+        this.client.awaitTermination(1, TimeUnit.MINUTES);
+      } catch (InterruptedException ignored) {
+      }
     }
   }
 

Original file line number	Diff line number	Diff line change
`@@ -317,6 +317,10 @@ public void close() {`
`317`	`317`	`}`
`318`	`318`	`if (this.ownsBigQueryWriteClient) {`
`319`	`319`	`this.client.close();`
	`320`	`+ try {`
	`321`	`+ this.client.awaitTermination(1, TimeUnit.MINUTES);`
	`322`	`+ } catch (InterruptedException ignored) {`
	`323`	`+ }`
`320`	`324`	`}`
`321`	`325`	`}`
`322`	`326`