make armada client autocloseable (#10)

armadaproject · Dec 24, 2024 · 8046688 · 8046688
1 parent 1ac7db7
commit 8046688
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/pom.xml b/pom.xml
@@ -5,7 +5,7 @@
 
   <groupId>io.armadaproject</groupId>
   <artifactId>java-client</artifactId>
-  <version>0.0.1</version>
+  <version>0.0.2</version>
   <packaging>jar</packaging>
   <description>Java client library for the Armada project</description>
   <name>java-client</name>

diff --git a/src/main/java/io/armadaproject/ArmadaClient.java b/src/main/java/io/armadaproject/ArmadaClient.java
@@ -24,7 +24,7 @@
 import io.grpc.stub.MetadataUtils;
 import java.util.Iterator;
 
-public class ArmadaClient {
+public class ArmadaClient implements AutoCloseable {
 
   private final ManagedChannel channel;
 
@@ -84,4 +84,9 @@ public JobStatusResponse getJobStatus(JobStatusRequest jobStatusRequest) {
     return jobsBlockingStub.getJobStatus(jobStatusRequest);
   }
 
+  @Override
+  public void close() throws Exception {
+    channel.shutdown();
+  }
+
 }