get for of PlasmaClient.connect

python/pyarrow/plasma.pyx

@@ -206,11 +206,28 @@ cdef class PlasmaClient:
         c_string store_socket_name
         c_string manager_socket_name
 
-    def __cinit__(self):
+    def __cinit__(self, store_socket_name, manager_socket_name, int release_delay):
+        """
+        Create a new PlasmaClient that is connected to a plasma store
+        and optionally a plasma manager.
+
+        Parameters
+        ----------
+        store_socket_name : str
+            Name of the socket the plasma store is listening at.
+        manager_socket_name : str
+            Name of the socket the plasma manager is listening at.
+        release_delay : int
+            The maximum number of objects that the client will keep and
+            delay releasing (for caching reasons).
+        """
         self.client.reset(new CPlasmaClient())
         self.notification_fd = -1
-        self.store_socket_name = ""
-        self.manager_socket_name = ""
+        self.store_socket_name = store_socket_name.encode()
+        self.manager_socket_name = manager_socket_name.encode()
+        with nogil:
+            check_status(self.client.get().Connect(self.store_socket_name,
+                         self.manager_socket_name, release_delay))
 
     cdef _get_object_buffers(self, object_ids, int64_t timeout_ms,
                              c_vector[CObjectBuffer]* result):
@@ -247,26 +264,6 @@ cdef class PlasmaClient:
     def manager_socket_name(self):
         return self.manager_socket_name.decode()
 
-    def connect(self, store_socket_name, manager_socket_name, int release_delay):
-        """
-        Conect the PlasmaClient to a plasma store and optionally a manager.
-
-        Parameters
-        ----------
-        store_socket_name : str
-            Name of the socket the plasma store is listening at.
-        manager_socket_name : str
-            Name of the socket the plasma manager is listening at.
-        release_delay : int
-            The maximum number of objects that the client will keep and
-            delay releasing (for caching reasons).
-        """
-        self.store_socket_name = store_socket_name.encode()
-        self.manager_socket_name = manager_socket_name.encode()
-        with nogil:
-            check_status(self.client.get().Connect(self.store_socket_name,
-                         self.manager_socket_name, release_delay))
-
     def create(self, ObjectID object_id, int64_t data_size, c_string metadata=b""):
         """
         Create a new buffer in the PlasmaStore for a particular object ID.

python/pyarrow/tests/test_plasma.py

@@ -154,11 +154,9 @@ class TestPlasmaClient(object):
         plasma_store_name, self.p = start_plasma_store(
             use_valgrind=os.getenv("PLASMA_VALGRIND") == "1")
         # Connect to Plasma.
-        self.plasma_client = plasma.PlasmaClient()
-        self.plasma_client.connect(plasma_store_name, "", 64)
+        self.plasma_client = plasma.PlasmaClient(plasma_store_name, "", 64)
         # For the eviction test
-        self.plasma_client2 = plasma.PlasmaClient()
-        self.plasma_client2.connect(plasma_store_name, "", 0)
+        self.plasma_client2 = plasma.PlasmaClient(plasma_store_name, "", 0)
 
     def teardown_method(self, test_method):
         # Check that the Plasma store is still alive.