diff --git a/fairmq/sdk/Topology.h b/fairmq/sdk/Topology.h index 8ffe198c..46a8d530 100644 --- a/fairmq/sdk/Topology.h +++ b/fairmq/sdk/Topology.h @@ -102,6 +102,16 @@ inline TopologyStateByCollection GroupByCollectionId(const TopologyState& topolo return state; } +inline TopologyStateByTask GroupByTaskId(const TopologyState& topologyState) +{ + TopologyStateByTask state; + for (const auto& ds : topologyState) { + state[ds.taskId] = ds; + } + + return state; +} + /** * @class BasicTopology Topology.h * @tparam Executor Associated I/O executor