tursom/memgraph
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add ExecuteOnWorkers

Browse Source 
Reviewers: florijan

Reviewed By: florijan

Differential Revision: https://phabricator.memgraph.io/D1130
This commit is contained in:
Dominik Gleich 2018-01-23 10:48:12 +01:00
parent 3bee31d8f3
commit 5a698444bf
1 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src/distributed/rpc_worker_clients.hpp
View File

@ -1,5 +1,9 @@
#pragma once
#include <functional>
#include <future>
#include <type_traits>
#include "communication/messaging/distributed.hpp"
#include "communication/rpc/rpc.hpp"
#include "distributed/coordination.hpp"
@ -36,6 +40,28 @@ class RpcWorkerClients {
auto GetWorkerIds() { return coordination_.GetWorkerIds(); }
/**
* Promises to execute function on workers rpc clients.
* @Tparam TResult - deduced automatically from method
* @param skip_worker_id - worker which to skip (set to -1 to avoid skipping)
* @param execute - Method which takes an rpc client and returns a result for
* it
* @return list of futures filled with function 'execute' results when applied
* to rpc clients
*/
template <typename TResult>
auto ExecuteOnWorkers(
int skip_worker_id,
std::function<TResult(communication::rpc::Client &)> execute) {
std::vector<std::future<TResult>> futures;
for (auto &client : clients_) {
if (client.first == skip_worker_id) continue;
futures.emplace_back(
std::async(std::launch::async, execute(client.second)));
}
return futures;
}
private:
communication::messaging::System &system_;
// TODO make Coordination const, it's member GetEndpoint must be const too.