通过--ServiceAccount在气流kubernetes吊舱操作员

Question

我试图创建和运行一个豆荚使用气流kubernetes豆荚操作员。下面的命令被尝试并确认有效，我正试图在本地使用kubernetes pod操作符复制相同的命令

kubectl run sparkairflow -n test-namespace --image=some-docker-repo.com:hello-world --serviceaccount=airflow --restart=Never -- spark-submit --deploy-mode cluster --master k8s://kubernetes.default.cluster.local:123 \
                 --name sparkairflow \
                 --conf spark.kubernetes.namespace=test-namespace \
                 --conf spark.kubernetes.container.image=some-docker-repo.com:hello-world \
                 --conf spark.kubernetes.authenticate.driver.serviceAccountName=airflow \
...

有没有办法将serviceaccount标志传递给airflow kubernetes操作员？

谢了！

Answer 1

事实证明，airflow代码中的pod对象确实有service_account_name字段，只是它不是由KubernetesPodOperator设置的。我必须扩展KubernetesPodOperator，并通过复制所有execute方法来重写它。添加了一行，我在其中设置了pod对象的service_account_name。

不是最干净的解决方案，但它奏效了！