ValueProvider类是Apache Beam中提供的一种方便处理运行时参数(如输入文件路径)的方法。它提供了一种方式来处理具有可能从不同来源(如命令行参数、环境变量或数据流)读取的值,并且自动支持将在运行时更新这些值。
在一个Beam管道中,我们可以使用ValueProvider作为管道参数,将其传递到一个特定的DoFn或PTransform中。例如,以下代码演示了如何定义一个ValueProvider类型的管道参数,并使用它来读取输入文件路径:
public static void main(String[] args) {
PipelineOptions options = PipelineOptionsFactory.fromArgs(args).create();
Pipeline pipeline = Pipeline.create(options);
MyOptions myOptions = options.as(MyOptions.class);
PCollection lines = pipeline.apply(TextIO.read().from(myOptions.getInputFile()));
// do something with the lines
}
public interface MyOptions extends PipelineOptions {
@Description("Input file path")
@Default.String("gs:///")
ValueProvider getInputFile();
void setInputFile(ValueProvider value);
}
在上面的代码示例中,MyOptions接口定义了一个ValueProvider
例如,以下命令行可用于运行该管道并将输入文件设置为gs://my-bucket/my-input-file.txt:
mvn compile exec:java \
-Dexec.mainClass=