Skip to content

Commit e00d99b

Browse files
sunsetxhsunsetxh
committed
rename gcs_callback_types.h to grpc_callback_types.h
Signed-off-by: yuchao-wang <wyc984872769@hotmail.com>
1 parent def1119 commit e00d99b

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

45 files changed

+349
-360
lines changed

‎src/mock/ray/gcs/store_client/in_memory_store_client.h‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ class MockInMemoryStoreClient : public InMemoryStoreClient {
3030
AsyncGet,
3131
(const std::string &table_name,
3232
const std::string &key,
33-
ToPostable<OptionalItemCallback<std::string>> callback),
33+
ToPostable<rpc::OptionalItemCallback<std::string>> callback),
3434
(override));
3535

3636
MOCK_METHOD(void,

‎src/mock/ray/gcs/store_client/redis_store_client.h‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ class MockStoreClient : public StoreClient {
2929
AsyncGet,
3030
(const std::string &table_name,
3131
const std::string &key,
32-
ToPostable<OptionalItemCallback<std::string>> callback),
32+
ToPostable<rpc::OptionalItemCallback<std::string>> callback),
3333
(override));
3434
MOCK_METHOD(void,
3535
AsyncGetAll,

‎src/mock/ray/gcs/store_client/store_client.h‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ class MockStoreClient : public StoreClient {
2929
AsyncGet,
3030
(const std::string &table_name,
3131
const std::string &key,
32-
ToPostable<OptionalItemCallback<std::string>> callback),
32+
ToPostable<rpc::OptionalItemCallback<std::string>> callback),
3333
(override));
3434
MOCK_METHOD(void,
3535
AsyncGetAll,

‎src/mock/ray/gcs_client/accessor.h‎

Lines changed: 34 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -23,27 +23,30 @@ class MockJobInfoAccessor : public JobInfoAccessor {
2323
MOCK_METHOD(void,
2424
AsyncAdd,
2525
(const std::shared_ptr<rpc::JobTableData> &data_ptr,
26-
const StatusCallback &callback),
26+
const rpc::StatusCallback &callback),
2727
(override));
2828
MOCK_METHOD(void,
2929
AsyncMarkFinished,
30-
(const JobID &job_id, const StatusCallback &callback),
30+
(const JobID &job_id, const rpc::StatusCallback &callback),
3131
(override));
3232
MOCK_METHOD(void,
3333
AsyncSubscribeAll,
34-
((const SubscribeCallback<JobID, rpc::JobTableData> &subscribe),
35-
const StatusCallback &done),
34+
((const rpc::SubscribeCallback<JobID, rpc::JobTableData> &subscribe),
35+
const rpc::StatusCallback &done),
3636
(override));
3737
MOCK_METHOD(void,
3838
AsyncGetAll,
3939
(const std::optional<std::string> &job_or_submission_id,
4040
bool skip_submission_job_info_field,
4141
bool skip_is_running_tasks_field,
42-
const MultiItemCallback<rpc::JobTableData> &callback,
42+
const rpc::MultiItemCallback<rpc::JobTableData> &callback,
4343
int64_t timeout_ms),
4444
(override));
4545
MOCK_METHOD(void, AsyncResubscribe, (), (override));
46-
MOCK_METHOD(void, AsyncGetNextJobID, (const ItemCallback<JobID> &callback), (override));
46+
MOCK_METHOD(void,
47+
AsyncGetNextJobID,
48+
(const rpc::ItemCallback<JobID> &callback),
49+
(override));
4750
};
4851

4952
} // namespace gcs
@@ -56,35 +59,35 @@ class MockNodeInfoAccessor : public NodeInfoAccessor {
5659
public:
5760
MOCK_METHOD(void,
5861
RegisterSelf,
59-
(rpc::GcsNodeInfo && local_node_info, const StatusCallback &callback),
62+
(rpc::GcsNodeInfo && local_node_info, const rpc::StatusCallback &callback),
6063
(override));
6164
MOCK_METHOD(void,
6265
AsyncRegister,
63-
(const rpc::GcsNodeInfo &node_info, const StatusCallback &callback),
66+
(const rpc::GcsNodeInfo &node_info, const rpc::StatusCallback &callback),
6467
(override));
6568
MOCK_METHOD(void,
6669
AsyncCheckAlive,
6770
(const std::vector<NodeID> &node_ids,
6871
int64_t timeout_ms,
69-
const MultiItemCallback<bool> &callback),
72+
const rpc::MultiItemCallback<bool> &callback),
7073
(override));
7174
MOCK_METHOD(void,
7275
AsyncGetAll,
73-
(const MultiItemCallback<rpc::GcsNodeInfo> &callback,
76+
(const rpc::MultiItemCallback<rpc::GcsNodeInfo> &callback,
7477
int64_t timeout_ms,
7578
const std::vector<NodeID> &node_ids),
7679
(override));
7780
MOCK_METHOD(void,
7881
AsyncGetAllNodeAddressAndLiveness,
79-
(const MultiItemCallback<rpc::GcsNodeAddressAndLiveness> &callback,
82+
(const rpc::MultiItemCallback<rpc::GcsNodeAddressAndLiveness> &callback,
8083
int64_t timeout_ms,
8184
const std::vector<NodeID> &node_ids),
8285
(override));
8386
MOCK_METHOD(
8487
void,
8588
AsyncSubscribeToNodeAddressAndLivenessChange,
8689
(std::function<void(NodeID, const rpc::GcsNodeAddressAndLiveness &)> subscribe,
87-
StatusCallback done),
90+
rpc::StatusCallback done),
8891
(override));
8992
MOCK_METHOD(std::optional<rpc::GcsNodeAddressAndLiveness>,
9093
GetNodeAddressAndLiveness,
@@ -115,11 +118,11 @@ class MockNodeResourceInfoAccessor : public NodeResourceInfoAccessor {
115118
public:
116119
MOCK_METHOD(void,
117120
AsyncGetAllAvailableResources,
118-
(const MultiItemCallback<rpc::AvailableResources> &callback),
121+
(const rpc::MultiItemCallback<rpc::AvailableResources> &callback),
119122
(override));
120123
MOCK_METHOD(void,
121124
AsyncGetAllResourceUsage,
122-
(const ItemCallback<rpc::ResourceUsageBatchData> &callback),
125+
(const rpc::ItemCallback<rpc::ResourceUsageBatchData> &callback),
123126
(override));
124127
};
125128

@@ -144,7 +147,8 @@ class MockTaskInfoAccessor : public TaskInfoAccessor {
144147
public:
145148
MOCK_METHOD(void,
146149
AsyncAddTaskEventData,
147-
(std::unique_ptr<rpc::TaskEventData> data_ptr, StatusCallback callback),
150+
(std::unique_ptr<rpc::TaskEventData> data_ptr,
151+
rpc::StatusCallback callback),
148152
(override));
149153
};
150154

@@ -158,27 +162,27 @@ class MockWorkerInfoAccessor : public WorkerInfoAccessor {
158162
public:
159163
MOCK_METHOD(void,
160164
AsyncSubscribeToWorkerFailures,
161-
(const ItemCallback<rpc::WorkerDeltaData> &subscribe,
162-
const StatusCallback &done),
165+
(const rpc::ItemCallback<rpc::WorkerDeltaData> &subscribe,
166+
const rpc::StatusCallback &done),
163167
(override));
164168
MOCK_METHOD(void,
165169
AsyncReportWorkerFailure,
166170
(const std::shared_ptr<rpc::WorkerTableData> &data_ptr,
167-
const StatusCallback &callback),
171+
const rpc::StatusCallback &callback),
168172
(override));
169173
MOCK_METHOD(void,
170174
AsyncGet,
171175
(const WorkerID &worker_id,
172-
const OptionalItemCallback<rpc::WorkerTableData> &callback),
176+
const rpc::OptionalItemCallback<rpc::WorkerTableData> &callback),
173177
(override));
174178
MOCK_METHOD(void,
175179
AsyncGetAll,
176-
(const MultiItemCallback<rpc::WorkerTableData> &callback),
180+
(const rpc::MultiItemCallback<rpc::WorkerTableData> &callback),
177181
(override));
178182
MOCK_METHOD(void,
179183
AsyncAdd,
180184
(const std::shared_ptr<rpc::WorkerTableData> &data_ptr,
181-
const StatusCallback &callback),
185+
const rpc::StatusCallback &callback),
182186
(override));
183187
MOCK_METHOD(void, AsyncResubscribe, (), (override));
184188
};
@@ -198,18 +202,18 @@ class MockPlacementGroupInfoAccessor : public PlacementGroupInfoAccessor {
198202
MOCK_METHOD(void,
199203
AsyncGet,
200204
(const PlacementGroupID &placement_group_id,
201-
const OptionalItemCallback<rpc::PlacementGroupTableData> &callback),
205+
const rpc::OptionalItemCallback<rpc::PlacementGroupTableData> &callback),
202206
(override));
203207
MOCK_METHOD(void,
204208
AsyncGetByName,
205209
(const std::string &placement_group_name,
206210
const std::string &ray_namespace,
207-
const OptionalItemCallback<rpc::PlacementGroupTableData> &callback,
211+
const rpc::OptionalItemCallback<rpc::PlacementGroupTableData> &callback,
208212
int64_t timeout_ms),
209213
(override));
210214
MOCK_METHOD(void,
211215
AsyncGetAll,
212-
(const MultiItemCallback<rpc::PlacementGroupTableData> &callback),
216+
(const rpc::MultiItemCallback<rpc::PlacementGroupTableData> &callback),
213217
(override));
214218
MOCK_METHOD(Status,
215219
SyncRemovePlacementGroup,
@@ -234,14 +238,14 @@ class MockInternalKVAccessor : public InternalKVAccessor {
234238
(const std::string &ns,
235239
const std::string &prefix,
236240
const int64_t timeout_ms,
237-
const OptionalItemCallback<std::vector<std::string>> &callback),
241+
const rpc::OptionalItemCallback<std::vector<std::string>> &callback),
238242
(override));
239243
MOCK_METHOD(void,
240244
AsyncInternalKVGet,
241245
(const std::string &ns,
242246
const std::string &key,
243247
const int64_t timeout_ms,
244-
const OptionalItemCallback<std::string> &callback),
248+
const rpc::OptionalItemCallback<std::string> &callback),
245249
(override));
246250
MOCK_METHOD(void,
247251
AsyncInternalKVPut,
@@ -250,26 +254,26 @@ class MockInternalKVAccessor : public InternalKVAccessor {
250254
const std::string &value,
251255
bool overwrite,
252256
const int64_t timeout_ms,
253-
const OptionalItemCallback<bool> &callback),
257+
const rpc::OptionalItemCallback<bool> &callback),
254258
(override));
255259
MOCK_METHOD(void,
256260
AsyncInternalKVExists,
257261
(const std::string &ns,
258262
const std::string &key,
259263
const int64_t timeout_ms,
260-
const OptionalItemCallback<bool> &callback),
264+
const rpc::OptionalItemCallback<bool> &callback),
261265
(override));
262266
MOCK_METHOD(void,
263267
AsyncInternalKVDel,
264268
(const std::string &ns,
265269
const std::string &key,
266270
bool del_by_prefix,
267271
const int64_t timeout_ms,
268-
const OptionalItemCallback<int> &callback),
272+
const rpc::OptionalItemCallback<int> &callback),
269273
(override));
270274
MOCK_METHOD(void,
271275
AsyncGetInternalConfig,
272-
(const OptionalItemCallback<std::string> &callback),
276+
(const rpc::OptionalItemCallback<std::string> &callback),
273277
(override));
274278
};
275279

‎src/mock/ray/gcs_client/accessors/actor_info_accessor.h‎

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -32,15 +32,15 @@ class FakeActorInfoAccessor : public gcs::ActorInfoAccessorInterface {
3232

3333
// Stub implementations for interface methods not used by this test
3434
void AsyncGet(const ActorID &,
35-
const gcs::OptionalItemCallback<rpc::ActorTableData> &) override {}
35+
const rpc::OptionalItemCallback<rpc::ActorTableData> &) override {}
3636
void AsyncGetAllByFilter(const std::optional<ActorID> &,
3737
const std::optional<JobID> &,
3838
const std::optional<std::string> &,
39-
const gcs::MultiItemCallback<rpc::ActorTableData> &,
39+
const rpc::MultiItemCallback<rpc::ActorTableData> &,
4040
int64_t = -1) override {}
4141
void AsyncGetByName(const std::string &,
4242
const std::string &,
43-
const gcs::OptionalItemCallback<rpc::ActorTableData> &,
43+
const rpc::OptionalItemCallback<rpc::ActorTableData> &,
4444
int64_t = -1) override {}
4545
Status SyncGetByName(const std::string &,
4646
const std::string &,
@@ -56,20 +56,20 @@ class FakeActorInfoAccessor : public gcs::ActorInfoAccessorInterface {
5656
}
5757
void AsyncReportActorOutOfScope(const ActorID &,
5858
uint64_t,
59-
const gcs::StatusCallback &,
59+
const rpc::StatusCallback &,
6060
int64_t = -1) override {}
6161
void AsyncRegisterActor(const TaskSpecification &task_spec,
62-
const gcs::StatusCallback &callback,
62+
const rpc::StatusCallback &callback,
6363
int64_t = -1) override {
6464
async_register_actor_callback_ = callback;
6565
}
6666
void AsyncRestartActorForLineageReconstruction(const ActorID &,
6767
uint64_t,
68-
const gcs::StatusCallback &,
68+
const rpc::StatusCallback &,
6969
int64_t = -1) override {}
7070
Status SyncRegisterActor(const TaskSpecification &) override { return Status::OK(); }
7171
void AsyncKillActor(
72-
const ActorID &, bool, bool, const gcs::StatusCallback &, int64_t = -1) override {}
72+
const ActorID &, bool, bool, const rpc::StatusCallback &, int64_t = -1) override {}
7373
void AsyncCreateActor(
7474
const TaskSpecification &task_spec,
7575
const rpc::ClientCallback<rpc::CreateActorReply> &callback) override {
@@ -78,8 +78,8 @@ class FakeActorInfoAccessor : public gcs::ActorInfoAccessorInterface {
7878

7979
void AsyncSubscribe(
8080
const ActorID &actor_id,
81-
const gcs::SubscribeCallback<ActorID, rpc::ActorTableData> &subscribe,
82-
const gcs::StatusCallback &done) override {
81+
const rpc::SubscribeCallback<ActorID, rpc::ActorTableData> &subscribe,
82+
const rpc::StatusCallback &done) override {
8383
auto callback_entry = std::make_pair(actor_id, subscribe);
8484
callback_map_.emplace(actor_id, subscribe);
8585
subscribe_finished_callback_map_[actor_id] = done;
@@ -124,14 +124,14 @@ class FakeActorInfoAccessor : public gcs::ActorInfoAccessorInterface {
124124
return true;
125125
}
126126

127-
absl::flat_hash_map<ActorID, gcs::SubscribeCallback<ActorID, rpc::ActorTableData>>
127+
absl::flat_hash_map<ActorID, rpc::SubscribeCallback<ActorID, rpc::ActorTableData>>
128128
callback_map_;
129-
absl::flat_hash_map<ActorID, gcs::StatusCallback> subscribe_finished_callback_map_;
129+
absl::flat_hash_map<ActorID, rpc::StatusCallback> subscribe_finished_callback_map_;
130130
absl::flat_hash_map<ActorID, uint32_t> actor_subscribed_times_;
131131

132132
// Callbacks for AsyncCreateActor and AsyncRegisterActor
133133
rpc::ClientCallback<rpc::CreateActorReply> async_create_actor_callback_;
134-
gcs::StatusCallback async_register_actor_callback_;
134+
rpc::StatusCallback async_register_actor_callback_;
135135
};
136136

137137
} // namespace gcs

‎src/ray/common/BUILD.bazel‎

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -390,14 +390,6 @@ ray_cc_library(
390390
],
391391
)
392392

393-
ray_cc_library(
394-
name = "gcs_callback_types",
395-
hdrs = ["gcs_callback_types.h"],
396-
deps = [
397-
"//src/ray/common:status",
398-
],
399-
)
400-
401393
ray_cc_library(
402394
name = "metrics",
403395
hdrs = ["metrics.h"],

‎src/ray/common/gcs_callback_types.h‎

Lines changed: 0 additions & 55 deletions
This file was deleted.

0 commit comments

Comments
 (0)