Examples with SinkTransformation org.apache.flink.streaming.api.transformations.SinkTransformation used on opensource projects

Example 1 with SinkTransformation

use of org.apache.flink.streaming.api.transformations.SinkTransformation in project flink by apache.

the class DataStreamSink method forSink.

@Internal
public static <T> DataStreamSink<T> forSink(DataStream<T> inputStream, Sink<T> sink) {
    final StreamExecutionEnvironment executionEnvironment = inputStream.getExecutionEnvironment();
    SinkTransformation<T, T> transformation = new SinkTransformation<>(inputStream, sink, inputStream.getType(), "Sink", executionEnvironment.getParallelism());
    executionEnvironment.addOperator(transformation);
    return new DataStreamSink<>(transformation);
}

Example 2 with SinkTransformation

use of org.apache.flink.streaming.api.transformations.SinkTransformation in project flink by apache.

the class DataStreamSinkTest method testGettingTransformationWithNewSinkAPI.

@Test
public void testGettingTransformationWithNewSinkAPI() {
    StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
    final Transformation<?> transformation = env.fromElements(1, 2).sinkTo(TestSink.newBuilder().build()).getTransformation();
    assertTrue(transformation instanceof SinkTransformation);
}

Example 3 with SinkTransformation

use of org.apache.flink.streaming.api.transformations.SinkTransformation in project flink by apache.

the class PythonOperatorChainingOptimizer method replaceInput.

private static void replaceInput(Transformation<?> transformation, Transformation<?> oldInput, Transformation<?> newInput) {
    try {
        if (transformation instanceof OneInputTransformation || transformation instanceof FeedbackTransformation || transformation instanceof SideOutputTransformation || transformation instanceof ReduceTransformation || transformation instanceof SinkTransformation || transformation instanceof LegacySinkTransformation || transformation instanceof TimestampsAndWatermarksTransformation || transformation instanceof PartitionTransformation) {
            final Field inputField = transformation.getClass().getDeclaredField("input");
            inputField.setAccessible(true);
            inputField.set(transformation, newInput);
        } else if (transformation instanceof TwoInputTransformation) {
            final Field inputField;
            if (((TwoInputTransformation<?, ?, ?>) transformation).getInput1() == oldInput) {
                inputField = transformation.getClass().getDeclaredField("input1");
            } else {
                inputField = transformation.getClass().getDeclaredField("input2");
            }
            inputField.setAccessible(true);
            inputField.set(transformation, newInput);
        } else if (transformation instanceof UnionTransformation || transformation instanceof AbstractMultipleInputTransformation) {
            final Field inputsField = transformation.getClass().getDeclaredField("inputs");
            inputsField.setAccessible(true);
            List<Transformation<?>> newInputs = Lists.newArrayList();
            newInputs.addAll(transformation.getInputs());
            newInputs.remove(oldInput);
            newInputs.add(newInput);
            inputsField.set(transformation, newInputs);
        } else if (transformation instanceof AbstractBroadcastStateTransformation) {
            final Field inputField;
            if (((AbstractBroadcastStateTransformation<?, ?, ?>) transformation).getRegularInput() == oldInput) {
                inputField = transformation.getClass().getDeclaredField("regularInput");
            } else {
                inputField = transformation.getClass().getDeclaredField("broadcastInput");
            }
            inputField.setAccessible(true);
            inputField.set(transformation, newInput);
        } else {
            throw new RuntimeException("Unsupported transformation: " + transformation);
        }
    } catch (NoSuchFieldException | IllegalAccessException e) {
        // This should never happen
        throw new RuntimeException(e);
    }
}