use of org.apache.beam.runners.samza.metrics.SamzaMetricsContainer in project beam by apache.
the class UnboundedSourceSystemTest method createConsumer.
private static UnboundedSourceSystem.Consumer<String, TestCheckpointMark> createConsumer(TestUnboundedSource<String> source, int splitNum) {
SamzaPipelineOptions pipelineOptions = PipelineOptionsFactory.as(SamzaPipelineOptions.class);
// emit immediately
pipelineOptions.setWatermarkInterval(0L);
pipelineOptions.setMaxSourceParallelism(splitNum);
return new UnboundedSourceSystem.Consumer<>(source, pipelineOptions, new SamzaMetricsContainer(new MetricsRegistryMap()), "test-step");
}
use of org.apache.beam.runners.samza.metrics.SamzaMetricsContainer in project beam by apache.
the class BoundedSourceSystemTest method createConsumer.
private static BoundedSourceSystem.Consumer<String> createConsumer(BoundedSource<String> source, int splitNum) {
SamzaPipelineOptions pipelineOptions = PipelineOptionsFactory.as(SamzaPipelineOptions.class);
pipelineOptions.setMaxSourceParallelism(splitNum);
return new BoundedSourceSystem.Consumer<>(source, pipelineOptions, new SamzaMetricsContainer(new MetricsRegistryMap()), "test-step");
}
Aggregations