.collect(Collectors.toList()); Pipeline res = Pipeline.newBuilder() .setComponents(fusedComponents) .addAllRootTransformIds(rootTransformIds)
public ExecutableGraph<PTransformNode, PCollectionNode> toGraph() { return PortableGraph.forPipeline( Pipeline.newBuilder() .setComponents(components) .addAllRootTransformIds(components.getTransformsMap().keySet()) .build()); } }
RunnerApi.Pipeline.newBuilder() .setComponents(components.toComponents()) .addAllRootTransformIds(rootIds)
GreedyPipelineFuser.fuse(Pipeline.newBuilder().setComponents(components).build());
.build(); FusedPipeline fused = GreedyPipelineFuser.fuse(Pipeline.newBuilder().setComponents(components).build());
FusedPipeline fused = GreedyPipelineFuser.fuse( Pipeline.newBuilder() .addRootTransformIds("impulse") .addRootTransformIds("compositeMultiLang")
.build(); Pipeline impulse = Pipeline.newBuilder() .addRootTransformIds(impulse1Transform.getUniqueName()) .addRootTransformIds(impulse2Transform.getUniqueName())
.build()); Pipeline pipeline = Pipeline.newBuilder() .addRootTransformIds(inputTransform.getId()) .addRootTransformIds(groupByKeyOnly.getId())
@Test public void replacesOnlyMatching() { RunnerApi.Pipeline p = Pipeline.newBuilder() .addAllRootTransformIds(ImmutableList.of("first", "second")) .setComponents(
.build(); FusedPipeline fused = GreedyPipelineFuser.fuse(Pipeline.newBuilder().setComponents(components).build());
.build(); FusedPipeline fused = GreedyPipelineFuser.fuse(Pipeline.newBuilder().setComponents(components).build());
RunnerApi.Pipeline.newBuilder() .addRootTransformIds(source.getId()) .addRootTransformIds(flatten.getId())
.build(); FusedPipeline fused = GreedyPipelineFuser.fuse(Pipeline.newBuilder().setComponents(components).build());
.build(); FusedPipeline fused = GreedyPipelineFuser.fuse(Pipeline.newBuilder().setComponents(components).build());
GreedyPipelineFuser.fuse(Pipeline.newBuilder().setComponents(components).build());
@Test public void replacesMultiple() { RunnerApi.Pipeline p = Pipeline.newBuilder() .addAllRootTransformIds(ImmutableList.of("first", "second")) .setComponents(
.build(); FusedPipeline fused = GreedyPipelineFuser.fuse(Pipeline.newBuilder().setComponents(components).build());
@Test public void replaceExistingCompositeSucceeds() { Pipeline p = Pipeline.newBuilder() .addRootTransformIds("root") .setComponents(
GreedyPipelineFuser.fuse(Pipeline.newBuilder().setComponents(components).build());
.build(); FusedPipeline fused = GreedyPipelineFuser.fuse(Pipeline.newBuilder().setComponents(components).build());