Examples with ChannelStateWriter org.apache.flink.runtime.checkpoint.channel.ChannelStateWriter used on opensource projects

Example 1 with ChannelStateWriter

use of org.apache.flink.runtime.checkpoint.channel.ChannelStateWriter in project flink by apache.

the class StreamTask method injectChannelStateWriterIntoChannels.

private void injectChannelStateWriterIntoChannels() {
    final Environment env = getEnvironment();
    final ChannelStateWriter channelStateWriter = subtaskCheckpointCoordinator.getChannelStateWriter();
    for (final InputGate gate : env.getAllInputGates()) {
        gate.setChannelStateWriter(channelStateWriter);
    }
    for (ResultPartitionWriter writer : env.getAllWriters()) {
        if (writer instanceof ChannelStateHolder) {
            ((ChannelStateHolder) writer).setChannelStateWriter(channelStateWriter);
        }
    }
}