DataStreamSink<Long> broadcastSink = broadcast.print(); StreamPartitioner<?> broadcastPartitioner = env.getStreamGraph().getStreamEdges(src.getId(), broadcastSink.getTransformation().getId()).get(0).getPartitioner(); assertTrue(broadcastPartitioner instanceof BroadcastPartitioner); DataStreamSink<Long> shuffleSink = shuffle.print(); StreamPartitioner<?> shufflePartitioner = env.getStreamGraph().getStreamEdges(src.getId(), shuffleSink.getTransformation().getId()).get(0).getPartitioner(); assertTrue(shufflePartitioner instanceof ShufflePartitioner); DataStreamSink<Long> forwardSink = forward.print(); StreamPartitioner<?> forwardPartitioner = env.getStreamGraph().getStreamEdges(src.getId(), forwardSink.getTransformation().getId()).get(0).getPartitioner(); assertTrue(forwardPartitioner instanceof ForwardPartitioner); DataStreamSink<Long> rebalanceSink = rebalance.print(); StreamPartitioner<?> rebalancePartitioner = env.getStreamGraph().getStreamEdges(src.getId(), rebalanceSink.getTransformation().getId()).get(0).getPartitioner(); assertTrue(rebalancePartitioner instanceof RebalancePartitioner); DataStreamSink<Long> globalSink = global.print(); StreamPartitioner<?> globalPartitioner = env.getStreamGraph().getStreamEdges(src.getId(), globalSink.getTransformation().getId()).get(0).getPartitioner(); assertTrue(globalPartitioner instanceof GlobalPartitioner);
env.getStreamGraph().getStreamEdges(map.getId(), unionFilter.getId()); } catch (RuntimeException e) { fail(e.getMessage()); env.getStreamGraph().getStreamEdges(flatMap.getId(), unionFilter.getId()); } catch (RuntimeException e) { fail(e.getMessage()); DataStreamSink<Integer> sink = select.print(); StreamEdge splitEdge = env.getStreamGraph().getStreamEdges(unionFilter.getId(), sink.getTransformation().getId()).get(0); assertEquals("a", splitEdge.getSelectedNames().get(0)); StreamEdge newSplitEdge = env.getStreamGraph().getStreamEdges(unionFilter.getId(), sinkWithIdentifier.getTransformation().getId()).get(0); assertEquals("a", newSplitEdge.getSelectedNames().get(0)); env.getStreamGraph().getStreamEdges(map.getId(), coMap.getId()); } catch (RuntimeException e) { fail(e.getMessage()); env.getStreamGraph().getStreamEdges(flatMap.getId(), coMap.getId()); } catch (RuntimeException e) { fail(e.getMessage());
assertTrue(isPartitioned(env.getStreamGraph().getStreamEdges(src1.getId(), id1))); assertTrue(isPartitioned(env.getStreamGraph().getStreamEdges(src1.getId(), id2))); assertTrue(isPartitioned(env.getStreamGraph().getStreamEdges(src1.getId(), id3))); assertTrue(isPartitioned(env.getStreamGraph().getStreamEdges(src1.getId(), id4))); assertTrue(isPartitioned(env.getStreamGraph().getStreamEdges(src1.getId(), pid1))); assertTrue(isPartitioned(env.getStreamGraph().getStreamEdges(src1.getId(), pid2))); assertTrue(isPartitioned(env.getStreamGraph().getStreamEdges(src1.getId(), pid3))); assertTrue(isPartitioned(env.getStreamGraph().getStreamEdges(src1.getId(), pid4))); assertTrue(isCustomPartitioned(env.getStreamGraph().getStreamEdges(src1.getId(), cid1))); assertTrue(isCustomPartitioned(env.getStreamGraph().getStreamEdges(src1.getId(), cid2))); assertTrue(isCustomPartitioned(env.getStreamGraph().getStreamEdges(src1.getId(), cid3))); assertTrue(isPartitioned(env.getStreamGraph().getStreamEdges(src1.getId(), downStreamId1))); assertTrue(isPartitioned(env.getStreamGraph().getStreamEdges(src2.getId(), downStreamId1))); assertTrue(isPartitioned(env.getStreamGraph().getStreamEdges(src1.getId(), downStreamId2))); assertTrue(isPartitioned(env.getStreamGraph().getStreamEdges(src2.getId(), downStreamId2))); assertTrue(isPartitioned(env.getStreamGraph().getStreamEdges(src1.getId(), downStreamId3))); assertTrue(isPartitioned(env.getStreamGraph().getStreamEdges(src2.getId(), downStreamId3))); assertTrue(isPartitioned(env.getStreamGraph().getStreamEdges(src1.getId(), downStreamId4))); assertTrue(isPartitioned(env.getStreamGraph().getStreamEdges(src2.getId(), downStreamId4))); assertTrue(isPartitioned(env.getStreamGraph().getStreamEdges(src1.getId(), downStreamId5))); assertTrue(isPartitioned(env.getStreamGraph().getStreamEdges(src2.getId(), downStreamId5)));
StreamEdge edge = graph.getStreamEdges( transformationIdMap.get(edgeProperty.getSource()), transformationIdMap.get(edgeProperty.getTarget())).get(edgeProperty.getIndex());