use of org.apache.flink.util.IterableUtils in project flink by apache.
the class DefaultExecutionGraph method releasePartitionGroups.
private void releasePartitionGroups(final List<ConsumedPartitionGroup> releasablePartitionGroups) {
if (releasablePartitionGroups.size() > 0) {
// Remove the cache of ShuffleDescriptors when ConsumedPartitionGroups are released
for (ConsumedPartitionGroup releasablePartitionGroup : releasablePartitionGroups) {
IntermediateResult totalResult = checkNotNull(intermediateResults.get(releasablePartitionGroup.getIntermediateDataSetID()));
totalResult.clearCachedInformationForPartitionGroup(releasablePartitionGroup);
}
final List<ResultPartitionID> releasablePartitionIds = releasablePartitionGroups.stream().flatMap(IterableUtils::toStream).map(this::createResultPartitionId).collect(Collectors.toList());
partitionTracker.stopTrackingAndReleasePartitions(releasablePartitionIds);
}
}
Aggregations