无法使用 SparkSubmitOperator 执行火花作业

Question

我可以使用 BashOperator 运行 Spark 作业，但我想使用 SparkSubmitOperator 使用 Spark 独立模式。

---

Here's 我的DAG for SparkSubmitOperator 和 stack-trace

args = {
    'owner': 'airflow',
    'start_date': datetime(2018, 5, 24)
}
dag = DAG('spark_job', default_args=args, schedule_interval="*/10 * * * *")

operator = SparkSubmitOperator(
    task_id='spark_submit_job',
    application='/home/ubuntu/test.py',
    total_executor_cores='1',
    executor_cores='1',
    executor_memory='2g',
    num_executors='1',
    name='airflow-spark',
    verbose=False,
    driver_memory='1g',
    conf={'master':'spark://xx.xx.xx.xx:7077'},
    dag=dag,
)

---

查看spark_submit_hook 的源代码似乎_resolve_connection() 总是设置master=yarn。如何通过 Spark 独立主 URL 更改 master 属性值？我可以设置哪些属性以在独立模式下运行Spark 作业？

Answer 1

您可以使用 Airflow Web UI 创建新连接或更改spark-default 连接。

主人可以是local、yarn、spark://HOST:PORT、mesos://HOST:PORT和k8s://https://<HOST>:<PORT>。

您还可以在 extras 中提供以下命令：

{"queue": "root.default", "deploy_mode": "cluster", "spark_home": "", "spark_binary": "spark-submit", "namespace": "default"}

“spark-submit”二进制文件应该在 PATH 中，或者 spark-home 设置在连接的附加文件中。