hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Core] fix erase iterator while iterating over a map. (#17204)

Browse source 
(cherry picked from commit 055a90374c)
This commit is contained in:
Chen Shen 2021-07-20 11:02:55 -07:00 • committed by Jiao Dong
parent c2d70bdb12
commit f960048631
3 changed files with 24 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/ray/gcs/gcs_server/gcs_placement_group_manager.cc
View file

@ -572,6 +572,8 @@ void GcsPlacementGroupManager::OnNodeDead(const NodeID &node_id) {
void GcsPlacementGroupManager::CleanPlacementGroupIfNeededWhenJobDead(
const JobID &job_id) {
std::vector<PlacementGroupID> groups_to_remove;
for (const auto &it : registered_placement_groups_) {
auto &placement_group = it.second;
if (placement_group->GetCreatorJobId() != job_id) {
@ -579,13 +581,19 @@ void GcsPlacementGroupManager::CleanPlacementGroupIfNeededWhenJobDead(
}
placement_group->MarkCreatorJobDead();
if (placement_group->IsPlacementGroupLifetimeDone()) {
RemovePlacementGroup(placement_group->GetPlacementGroupID(), [](Status status) {});
groups_to_remove.push_back(placement_group->GetPlacementGroupID());
}
}
for (const auto &group : groups_to_remove) {
RemovePlacementGroup(group, [](Status status) {});
}
}
void GcsPlacementGroupManager::CleanPlacementGroupIfNeededWhenActorDead(
const ActorID &actor_id) {
std::vector<PlacementGroupID> groups_to_remove;
for (const auto &it : registered_placement_groups_) {
auto &placement_group = it.second;
if (placement_group->GetCreatorActorId() != actor_id) {
@ -593,9 +601,13 @@ void GcsPlacementGroupManager::CleanPlacementGroupIfNeededWhenActorDead(
}
placement_group->MarkCreatorActorDead();
if (placement_group->IsPlacementGroupLifetimeDone()) {
RemovePlacementGroup(placement_group->GetPlacementGroupID(), [](Status status) {});
groups_to_remove.push_back(placement_group->GetPlacementGroupID());
}
}
for (const auto &group : groups_to_remove) {
RemovePlacementGroup(group, [](Status status) {});
}
}
void GcsPlacementGroupManager::CollectStats() const {

5
src/ray/object_manager/plasma/create_request_queue.cc
View file

@ -190,10 +190,11 @@ void CreateRequestQueue::RemoveDisconnectedClientRequests(
for (auto it = fulfilled_requests_.begin(); it != fulfilled_requests_.end();) {
if (it->second && it->second->client == client) {
fulfilled_requests_.erase(it);
}
fulfilled_requests_.erase(it++);
} else {
it++;
}
}
}
} // namespace plasma

9
src/ray/pubsub/publisher.cc
View file

@ -300,6 +300,8 @@ bool Publisher::UnregisterSubscriberInternal(const SubscriberID &subscriber_id)
void Publisher::CheckDeadSubscribers() {
absl::MutexLock lock(&mutex_);
std::vector<SubscriberID> dead_subscribers;
for (const auto &it : subscribers_) {
const auto &subscriber = it.second;
@ -308,13 +310,16 @@ void Publisher::CheckDeadSubscribers() {
RAY_CHECK(!(disconnected && active_connection_timed_out));
if (disconnected) {
const auto &subscriber_id = it.first;
UnregisterSubscriberInternal(subscriber_id);
dead_subscribers.push_back(it.first);
} else if (active_connection_timed_out) {
// Refresh the long polling connection. The subscriber will send it again.
subscriber->PublishIfPossible(/*force*/ true);
}
}
for (const auto &subscriber_id : dead_subscribers) {
UnregisterSubscriberInternal(subscriber_id);
}
}
bool Publisher::CheckNoLeaks() const {