| 31 | } |
| 32 | |
| 33 | std::vector<Node> Postmaster::assignNodeRank(const std::vector<Node>& nodes) { |
| 34 | int server_r = 0; |
| 35 | int worker_r = 0; |
| 36 | int sch_r = 0; |
| 37 | int other_r = 0; |
| 38 | |
| 39 | auto ret = nodes; |
| 40 | for (auto& o : ret) { |
| 41 | int k = 0; |
| 42 | auto r = o.role(); |
| 43 | if (r == Node::SERVER) { |
| 44 | k = server_r++; |
| 45 | } else if (r == Node::WORKER) { |
| 46 | k = worker_r++; |
| 47 | } else if (r == Node::SCHEDULER) { |
| 48 | CHECK_EQ(++sch_r, 1); |
| 49 | } else { |
| 50 | k = other_r++; |
| 51 | } |
| 52 | o.set_rank(k); |
| 53 | } |
| 54 | return ret; |
| 55 | } |
| 56 | |
| 57 | // void Postmaster::createApp( |
| 58 | // const std::vector<Node>& nodes, const std::vector<AppConfig>& apps) { |