zhijiangW commented on a change in pull request #8687:
[FLINK-12612][coordination] Track stored partition on the TaskExecutor
URL: https://github.com/apache/flink/pull/8687#discussion_r293200602
##########
File path:
flink-runtime/src/test/java/org/apache/flink/runtime/taskexecutor/TaskExecutorPartitionLifecycleTest.java
##########
@@ -276,4 +409,69 @@ private TestingTaskExecutor
createTestingTaskExecutor(TaskManagerServices taskMa
null),
new TestingFatalErrorHandler());
}
+
+ private static class TestJobAwareShuffleEnvironment implements
JobAwareShuffleEnvironment<ResultPartitionWriter, InputGate> {
+
+ private final Function<JobID, Boolean>
hasPartitionsOccupyingLocalResourcesFunction;
+ private Consumer<JobID> listener = null;
+
+ private TestJobAwareShuffleEnvironment(Function<JobID, Boolean>
hasPartitionsOccupyingLocalResourcesFunction) {
+ this.hasPartitionsOccupyingLocalResourcesFunction =
hasPartitionsOccupyingLocalResourcesFunction;
+ }
+
+ @Override
+ public int start() throws IOException {
+ return 0;
+ }
+
+ @Override
+ public Collection<ResultPartitionWriter>
createResultPartitionWriters(JobID jobId, String taskName, ExecutionAttemptID
executionAttemptID, Collection<ResultPartitionDeploymentDescriptor>
resultPartitionDeploymentDescriptors, MetricGroup outputGroup, MetricGroup
buffersGroup) {
+ return null;
+ }
+
+ @Override
+ public void releaseFinishedPartitions(JobID jobId,
Collection<ResultPartitionID> partitionIds) {
+
+ }
+
+ @Override
+ public void
releaseAllFinishedPartitionsForJobAndMarkJobInactive(JobID jobId) {
+
+ }
+
+ @Override
+ public Collection<ResultPartitionID>
getPartitionsOccupyingLocalResources() {
+ return null;
+ }
+
+ @Override
+ public boolean hasPartitionsOccupyingLocalResources(JobID
jobId) {
+ return
hasPartitionsOccupyingLocalResourcesFunction.apply(jobId);
+ }
+
+ @Override
+ public void markJobActive(JobID jobId) {
+
+ }
+
+ @Override
+ public void
setPartitionFailedOrFinishedListener(Consumer<JobID> listener) {
+ this.listener = listener;
+ }
+
+ @Override
+ public Collection<InputGate> createInputGates(String taskName,
ExecutionAttemptID executionAttemptID, PartitionProducerStateProvider
partitionProducerStateProvider, Collection<InputGateDeploymentDescriptor>
inputGateDeploymentDescriptors, MetricGroup parentGroup, MetricGroup
inputGroup, MetricGroup buffersGroup) {
+ return null;
+ }
+
+ @Override
+ public boolean updatePartitionInfo(ExecutionAttemptID
consumerID, PartitionInfo partitionInfo) throws IOException,
InterruptedException {
+ return false;
+ }
+
+ @Override
+ public void close() throws Exception {
+
Review comment:
ditto: remove empty line
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
