How to use

ImmutableList

in

org.apache.beam.vendor.guava.v20_0.com.google.common.collect

 @Override
 public Iterable<Class<? extends PipelineRunner<?>>> getPipelineRunners() {
  return ImmutableList.of(FlinkRunner.class, TestFlinkRunner.class);
 }
}

@Override
public Iterable<ValueT> read() {
 try {
  Iterable<ValueT> result =
    flinkStateBackend
      .getPartitionedState(
        namespace.stringKey(), StringSerializer.INSTANCE, flinkStateDescriptor)
      .values();
  return result != null ? ImmutableList.copyOf(result) : Collections.emptyList();
 } catch (Exception e) {
  throw new RuntimeException("Error get map state values.", e);
 }
}

@Test
public void shouldUseTransformOverrides() {
 boolean[] testParameters = {true, false};
 for (boolean streaming : testParameters) {
  FlinkPipelineOptions options = PipelineOptionsFactory.as(FlinkPipelineOptions.class);
  options.setStreaming(streaming);
  options.setRunner(FlinkRunner.class);
  FlinkPipelineExecutionEnvironment flinkEnv = new FlinkPipelineExecutionEnvironment(options);
  Pipeline p = Mockito.spy(Pipeline.create(options));
  flinkEnv.translate(p);
  ArgumentCaptor<ImmutableList> captor = ArgumentCaptor.forClass(ImmutableList.class);
  Mockito.verify(p).replaceAll(captor.capture());
  ImmutableList<PTransformOverride> overridesList = captor.getValue();
  assertThat(overridesList.isEmpty(), is(false));
  assertThat(
    overridesList.size(), is(FlinkTransformOverrides.getDefaultOverrides(options).size()));
 }
}

static List<PTransformOverride> getDefaultOverrides(FlinkPipelineOptions options) {
 ImmutableList.Builder<PTransformOverride> builder = ImmutableList.builder();
 builder

 @Override
 public Iterable<Class<? extends PipelineOptions>> getPipelineOptions() {
  return ImmutableList.of(FlinkPipelineOptions.class);
 }
}

@Override
public Iterable<KeyT> read() {
 try {
  Iterable<KeyT> result =
    flinkStateBackend
      .getPartitionedState(
        namespace.stringKey(), StringSerializer.INSTANCE, flinkStateDescriptor)
      .keys();
  return result != null ? ImmutableList.copyOf(result) : Collections.emptyList();
 } catch (Exception e) {
  throw new RuntimeException("Error get map state keys.", e);
 }
}

@Override
public List<? extends Coder<?>> getCoderArguments() {
 return ImmutableList.of(keyCoder, elemCoder, windowCoder);
}

@Override
public Iterable<T> read() {
 try {
  Iterable<T> result =
    flinkStateBackend
      .getPartitionedState(
        namespace.stringKey(), StringSerializer.INSTANCE, flinkStateDescriptor)
      .keys();
  return result != null ? ImmutableList.copyOf(result) : Collections.emptyList();
 } catch (Exception e) {
  throw new RuntimeException("Error read from state.", e);
 }
}

@Override
public AccumT addInput(
  K key,
  AccumT accumulator,
  AccumT value,
  PipelineOptions options,
  SideInputReader sideInputReader,
  Collection<? extends BoundedWindow> windows) {
 return combineFnRunner.mergeAccumulators(
   ImmutableList.of(accumulator, value), options, sideInputReader, windows);
}

  };
 return result != null ? ImmutableList.copyOf(result) : Collections.emptyList();
} catch (Exception e) {
 throw new RuntimeException("Error reading state.", e);

@Override
public List<BoundedSource<KV<K, V>>> split(long desiredBundleSizeBytes, PipelineOptions options)
  throws Exception {
 // desiredBundleSizeBytes is not being considered as splitting based on this
 // value is not supported by inputFormat getSplits() method.
 if (inputSplit != null) {
  LOGGER.info("Not splitting source {} because source is already split.", this);
  return ImmutableList.of((BoundedSource<KV<K, V>>) this);
 }
 computeSplitsIfNecessary();
 LOGGER.info(
   "Generated {} splits. Size of first split is {} ",
   inputSplits.size(),
   inputSplits.get(0).getSplit().getLength());
 return inputSplits
   .stream()
   .map(
     serializableInputSplit -> {
      return new HadoopInputFormatBoundedSource<>(
        conf,
        keyCoder,
        valueCoder,
        keyTranslationFunction,
        valueTranslationFunction,
        serializableInputSplit);
     })
   .collect(Collectors.toList());
}

Collections.emptyMap(),
mainOutput,
ImmutableList.of(additionalOutput),
outputManagerFactory,
jobInfo,
FlinkExecutableStageContext.factory(options),
createOutputMap(mainOutput, ImmutableList.of(additionalOutput)),
WindowingStrategy.globalDefault(),
null,

WindowingStrategy.globalDefault(),
ImmutableList.of(view1, view2), /* side inputs */
PipelineOptionsFactory.as(FlinkPipelineOptions.class),
null,

WindowingStrategy.of(FixedWindows.of(Duration.millis(100))),
ImmutableList.of(view1, view2), /* side inputs */
PipelineOptionsFactory.as(FlinkPipelineOptions.class),
keyCoder,

WindowingStrategy.of(FixedWindows.of(Duration.millis(100))),
ImmutableList.of(view1, view2), /* side inputs */
PipelineOptionsFactory.as(FlinkPipelineOptions.class),
null,

WindowingStrategy.of(FixedWindows.of(Duration.millis(100))),
ImmutableList.of(view1, view2), /* side inputs */
PipelineOptionsFactory.as(FlinkPipelineOptions.class),
keyCoder,

    WindowingStrategy.of(FixedWindows.of(Duration.millis(100))),
    ImmutableList.of(view1, view2), /* side inputs */
    PipelineOptionsFactory.as(FlinkPipelineOptions.class),
    keyCoder,
    new RawUnionValue(
      1,
      valuesInWindow(ImmutableList.of("hello", "ciao"), new Instant(0), firstWindow))));
testHarness.processElement2(
  new StreamRecord<>(
    new RawUnionValue(
      2, valuesInWindow(ImmutableList.of("foo", "bar"), new Instant(0), secondWindow))));
      1,
      valuesInWindow(
        ImmutableList.of("hello", "ciao"), new Instant(1000), firstWindow))));
testHarness.processElement2(
  new StreamRecord<>(
    new RawUnionValue(
      2,
      valuesInWindow(ImmutableList.of("foo", "bar"), new Instant(1000), secondWindow))));

Collections.emptyMap(),
mainOutput,
ImmutableList.of(additionalOutput1, additionalOutput2),
new DoFnOperator.MultiOutputOutputManagerFactory(
  mainOutput, tagsToOutputTags, tagsToCoders, tagsToIds),

PAssert.that(result).containsInAnyOrder(KV.of("foo", ImmutableList.of(4L, 3L, 3L)));

1,
valuesInWindow(
  ImmutableList.of(KV.of((Void) null, "hello"), KV.of((Void) null, "ciao")),
  new Instant(0),
  firstWindow))));
2,
valuesInWindow(
  ImmutableList.of(KV.of((Void) null, "foo"), KV.of((Void) null, "bar")),
  new Instant(0),
  secondWindow))));