casremote.py: Add close() method to close gRPC channel

Also support use as context manager.

1 files changed, 12 insertions, 0 deletions

diff --git a/src/buildstream/_cas/casremote.py b/src/buildstream/_cas/casremote.py
index 9cc1a5488..05b57d3d4 100644
--- a/src/buildstream/_cas/casremote.py
+++ b/src/buildstream/_cas/casremote.py
@@ -183,6 +183,18 @@ class CASRemote():
 
             self._initialized = True
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        self.close()
+        return False
+
+    def close(self):
+        if self.channel:
+            self.channel.close()
+            self.channel = None
+
     # check_remote
     #
     # Used when checking whether remote_specs work in the buildstream main