Search in sources :

Example 6 with SparkContextOptions

use of org.apache.beam.runners.spark.SparkContextOptions in project components by Talend.

the class PubSubOutputRuntimeTestIT method createSparkRunnerPipeline.

// TODO extract this to utils
private Pipeline createSparkRunnerPipeline() {
    JavaSparkContext jsc = new JavaSparkContext("local[2]", this.getClass().getName());
    PipelineOptions o = PipelineOptionsFactory.create();
    SparkContextOptions options = o.as(SparkContextOptions.class);
    options.setProvidedSparkContext(jsc);
    options.setUsesProvidedSparkContext(true);
    options.setRunner(SparkRunner.class);
    runtimeContainer = new BeamJobRuntimeContainer(options);
    return Pipeline.create(options);
}
Also used : BeamJobRuntimeContainer(org.talend.components.adapter.beam.BeamJobRuntimeContainer) SparkContextOptions(org.apache.beam.runners.spark.SparkContextOptions) PipelineOptions(org.apache.beam.sdk.options.PipelineOptions) JavaSparkContext(org.apache.spark.api.java.JavaSparkContext)

Aggregations

SparkContextOptions (org.apache.beam.runners.spark.SparkContextOptions)6 PipelineOptions (org.apache.beam.sdk.options.PipelineOptions)5 JavaSparkContext (org.apache.spark.api.java.JavaSparkContext)4 BeamJobRuntimeContainer (org.talend.components.adapter.beam.BeamJobRuntimeContainer)3 SparkConf (org.apache.spark.SparkConf)2 SparkContext (org.apache.spark.SparkContext)2