[Placement group] Input validation (#11152)

* Add a basic input validation.

* Addressed code review.
This commit is contained in:
SangBin Cho
2020-10-14 13:56:41 -07:00
committed by GitHub
parent a10e36ca04
commit 666fcde8ca
2 changed files with 21 additions and 7 deletions
+10 -2
View File
@@ -77,8 +77,16 @@ class PlacementGroupSpecBuilder {
auto mutable_bundle_id = message_bundle->mutable_bundle_id();
mutable_bundle_id->set_bundle_index(i);
mutable_bundle_id->set_placement_group_id(placement_group_id.Binary());
message_bundle->mutable_unit_resources()->insert(resources.begin(),
resources.end());
auto mutable_unit_resources = message_bundle->mutable_unit_resources();
for (auto it = resources.begin(); it != resources.end();) {
auto current = it++;
// Remove a resource with value 0 because they are not allowed.
if (current->second == 0) {
resources.erase(current);
} else {
mutable_unit_resources->insert({current->first, current->second});
}
}
}
return *this;
}