Examples with CustomStreamlet org.apache.heron.streamlet.impl.streamlets.CustomStreamlet used on opensource projects

Example 1 with CustomStreamlet

use of org.apache.heron.streamlet.impl.streamlets.CustomStreamlet in project heron by twitter.

the class StreamletImplTest method testCustomStreamletWithGrouperFromBolt.

@Test
@SuppressWarnings("unchecked")
public void testCustomStreamletWithGrouperFromBolt() throws Exception {
    Streamlet<Double> baseStreamlet = builder.newSource(() -> Math.random());
    Streamlet<Double> streamlet = baseStreamlet.setNumPartitions(20).applyOperator(new MyBoltOperator(), new ShuffleStreamGrouping());
    assertTrue(streamlet instanceof CustomStreamlet);
    CustomStreamlet<Double, Double> mStreamlet = (CustomStreamlet<Double, Double>) streamlet;
    assertEquals(20, mStreamlet.getNumPartitions());
    SupplierStreamlet<Double> supplierStreamlet = (SupplierStreamlet<Double>) baseStreamlet;
    assertEquals(supplierStreamlet.getChildren().size(), 1);
    assertEquals(supplierStreamlet.getChildren().get(0), streamlet);
}