Add SafeCloseable context manager wrapper

py/server/deephaven/jcompat.py

@@ -304,3 +304,24 @@ def _j_array_to_series(dtype: DType, j_array: jpy.JType, conv_null: bool) -> pd.
         s = pd.Series(data=np_array, copy=False)
 
     return s
+
+
+class SafeCloseable:
+    """A context manager wrapper of Java SafeCloseable to emulate Java try-with-resources."""
+    def __init__(self, j_safe_closeable):
+        self._j_safe_closeable = j_safe_closeable
+        self.closed = False
+
+    def __enter__(self):
+        return self
+
+    def close(self):
+        if not self.closed:
+            self.closed = True
+            self._j_safe_closeable.close()
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        self.close()
+
+    def __del__(self):
+        self.close()

py/server/tests/test_jcompat.py

@@ -5,11 +5,12 @@
 import unittest
 
 from deephaven import dtypes
-from deephaven.jcompat import j_function, j_lambda
+from deephaven.jcompat import j_function, j_lambda, SafeCloseable
 from tests.testbase import BaseTestCase
 
 import jpy
 
+_JSharedContext = jpy.get_type("io.deephaven.engine.table.SharedContext")
 
 class JCompatTestCase(BaseTestCase):
     def test_j_function(self):
@@ -29,6 +30,12 @@ def int_to_str(v: int) -> str:
         r = j_func.apply(10)
         self.assertEqual(r, "10")
 
+    def test_safe_closeable(self):
+        safe_closeable = SafeCloseable(_JSharedContext.makeSharedContext())
+        with safe_closeable:
+            pass
+        self.assertEqual(safe_closeable.closed, True)
+
 
 if __name__ == "__main__":
     unittest.main()