[core worker] Python core worker object interface (#5272)

2026-07-27 11:26:41 +08:00 · 2019-09-12 23:07:46 -07:00
parent 1b880191b0
commit 07c4c6367a
49 changed files with 1157 additions and 552 deletions
@@ -0,0 +1,23 @@
+from libcpp cimport bool as c_bool
+from libcpp.string cimport string as c_string
+
+from ray.includes.common cimport CGcsClientOptions
+
+
+cdef class GcsClientOptions:
+    """Cython wrapper class of C++ `ray::gcs::GcsClientOptions`."""
+    cdef:
+        unique_ptr[CGcsClientOptions] gcs_client_options
+
+    def __init__(self, redis_ip, int redis_port,
+                 redis_password, c_bool is_test_client=False):
+        if not redis_password:
+            redis_password = ""
+        self.gcs_client_options.reset(
+            new CGcsClientOptions(redis_ip.encode("ascii"),
+                                  redis_port,
+                                  redis_password.encode("ascii"),
+                                  is_test_client))
+
+    cdef CGcsClientOptions* native(self):
+        return <CGcsClientOptions*>(self.gcs_client_options.get())