From f08548202db44e517c0c3e4f68c43f0064d08e6c Mon Sep 17 00:00:00 2001
From: Yuya Ebihara <ebyhry@gmail.com>
Date: Tue, 28 Nov 2023 18:32:29 +0900
Subject: [PATCH] Close BigQueryWriteClient in BigQueryPageSink

---
 .../java/io/trino/plugin/bigquery/BigQueryPageSink.java     | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/plugin/trino-bigquery/src/main/java/io/trino/plugin/bigquery/BigQueryPageSink.java b/plugin/trino-bigquery/src/main/java/io/trino/plugin/bigquery/BigQueryPageSink.java
index 7a969aad5547..bf378f654eb9 100644
--- a/plugin/trino-bigquery/src/main/java/io/trino/plugin/bigquery/BigQueryPageSink.java
+++ b/plugin/trino-bigquery/src/main/java/io/trino/plugin/bigquery/BigQueryPageSink.java
@@ -119,11 +119,15 @@ private void insertWithCommitted(JSONArray batch)
     @Override
     public CompletableFuture<Collection<Slice>> finish()
     {
+        client.close();
         Slice value = Slices.allocate(Long.BYTES);
         value.setLong(0, pageSinkId.getId());
         return completedFuture(ImmutableList.of(value));
     }
 
     @Override
-    public void abort() {}
+    public void abort()
+    {
+        client.close();
+    }
 }