Remove raylet client from Python worker (#6018)

python/ray/includes/libcoreworker.pxd

@@ -12,6 +12,8 @@ from libcpp.vector cimport vector as c_vector
 
 from ray.includes.unique_ids cimport (
     CActorID,
+    CActorCheckpointID,
+    CClientID,
     CJobID,
     CTaskID,
     CObjectID,
@@ -31,7 +33,6 @@ from ray.includes.common cimport (
     CGcsClientOptions,
 )
 from ray.includes.task cimport CTaskSpec
-from ray.includes.libraylet cimport CRayletClient
 
 ctypedef unordered_map[c_string, c_vector[pair[int64_t, double]]] \
     ResourceMappingType
@@ -107,9 +108,6 @@ cdef extern from "ray/core_worker/core_worker.h" nogil:
             const c_vector[shared_ptr[CBuffer]] &metadatas,
             c_vector[shared_ptr[CRayObject]] *return_objects)
 
-        # TODO(edoakes): remove this once the raylet client is no longer used
-        # directly.
-        CRayletClient &GetRayletClient()
         CJobID GetCurrentJobId()
         CTaskID GetCurrentTaskId()
         const CActorID &GetActorId()
@@ -159,3 +157,13 @@ cdef extern from "ray/core_worker/core_worker.h" nogil:
                       ray_callback_function successs_callback,
                       ray_callback_function fallback_callback,
                       void* python_future)
+
+        CRayStatus PushError(const CJobID &job_id, const c_string &type,
+                             const c_string &error_message, double timestamp)
+        CRayStatus PrepareActorCheckpoint(const CActorID &actor_id,
+                                          CActorCheckpointID *checkpoint_id)
+        CRayStatus NotifyActorResumedFromCheckpoint(
+            const CActorID &actor_id, const CActorCheckpointID &checkpoint_id)
+        CRayStatus SetResource(const c_string &resource_name,
+                               const double capacity,
+                               const CClientID &client_Id)

python/ray/includes/libraylet.pxd

@@ -1,77 +0,0 @@
-from libc.stdint cimport int64_t
-from libcpp cimport bool as c_bool
-from libcpp.memory cimport unique_ptr
-from libcpp.string cimport string as c_string
-from libcpp.utility cimport pair
-from libcpp.vector cimport vector as c_vector
-
-from ray.includes.common cimport (
-    CLanguage,
-    CRayStatus,
-)
-from ray.includes.unique_ids cimport (
-    CActorCheckpointID,
-    CActorID,
-    CClientID,
-    CJobID,
-    CWorkerID,
-    CObjectID,
-    CTaskID,
-)
-from ray.includes.task cimport CTaskSpec
-
-
-cdef extern from "ray/protobuf/gcs.pb.h" nogil:
-    cdef cppclass GCSProfileEvent "ProfileTableData::ProfileEvent":
-        void set_event_type(const c_string &value)
-        void set_start_time(double value)
-        void set_end_time(double value)
-        c_string set_extra_data(const c_string &value)
-        GCSProfileEvent()
-
-    cdef cppclass GCSProfileTableData "ProfileTableData":
-        void set_component_type(const c_string &value)
-        void set_component_id(const c_string &value)
-        void set_node_ip_address(const c_string &value)
-        GCSProfileEvent *add_profile_events()
-        GCSProfileTableData()
-
-
-ctypedef pair[c_vector[CObjectID], c_vector[CObjectID]] WaitResultPair
-
-
-cdef extern from "ray/raylet/raylet_client.h" nogil:
-    cdef cppclass CRayletClient "ray::raylet::RayletClient":
-        CRayletClient(const c_string &raylet_socket,
-                      const CWorkerID &worker_id,
-                      c_bool is_worker, const CJobID &job_id,
-                      const CLanguage &language)
-        CRayStatus Disconnect()
-        CRayStatus SubmitTask(const CTaskSpec &task_spec)
-        CRayStatus FetchOrReconstruct(c_vector[CObjectID] &object_ids,
-                                      c_bool fetch_only,
-                                      c_bool is_direct_call_task,
-                                      const CTaskID &current_task_id)
-        CRayStatus NotifyUnblocked(const CTaskID &current_task_id)
-        CRayStatus Wait(const c_vector[CObjectID] &object_ids,
-                        int num_returns, int64_t timeout_milliseconds,
-                        c_bool wait_local, c_bool is_direct_call_task,
-                        const CTaskID &current_task_id,
-                        WaitResultPair *result)
-        CRayStatus PushError(const CJobID &job_id, const c_string &type,
-                             const c_string &error_message, double timestamp)
-        CRayStatus PushProfileEvents(
-            const GCSProfileTableData &profile_events)
-        CRayStatus FreeObjects(const c_vector[CObjectID] &object_ids,
-                               c_bool local_only, c_bool delete_creating_tasks)
-        CRayStatus PrepareActorCheckpoint(const CActorID &actor_id,
-                                          CActorCheckpointID &checkpoint_id)
-        CRayStatus NotifyActorResumedFromCheckpoint(
-            const CActorID &actor_id, const CActorCheckpointID &checkpoint_id)
-        CRayStatus SetResource(const c_string &resource_name,
-                               const double capacity,
-                               const CClientID &client_Id)
-        CLanguage GetLanguage() const
-        CWorkerID GetWorkerID() const
-        CJobID GetJobID() const
-        c_bool IsWorker() const