带阿帕奇梁的Spring

Question

 public static void main(String[] args) {
    initSpringApplicationContext();

    GcmOptions options = PipelineOptionsFactory.fromArgs(args)
        .withValidation()
        .as(GcmOptions.class);
    Pipeline pipeline = Pipeline.create(options);
    // pipeline definition
}

我想将spring应用程序上下文注入到每个ParDo函数中。

Answer 1

这里的问题是，ApplicationContext在任何辅助机器上都不可用，因为main方法只在构造作业时调用，而不是在任何辅助机器上调用。因此，InitSpringApplicationContext永远不会在任何工作者上调用。

我从未尝试过在Apache Beam中使用Spring，但我想在静态初始化器块中移动initSpringApplicationContext会得到预期的结果。

public class ApplicationContextHolder {

    private static final ApplicationContext CTX;

    static {
        CTX = initApplicationContext();
    }

    public static ApplicationContext getContext() {
        return CTX;
    }
}

请注意，这不应该被认为是在Apache Beam中使用Spring的最佳实践，因为它在Apache Beam的生命周期中没有很好地集成。例如，当在初始化应用程序上下文期间发生错误时，它将出现在使用applicationcontextholder的第一个位置。因此，我建议从静态初始化器块中提取initapplicationcontext并针对Apache Beam的生命周期显式调用它。设置阶段将是一个很好的地方。