mirror/cube-studio
2
0
Fork 0
mirror of https://github.com/tencentmusic/cube-studio.git synced 2024-11-21 01:16:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

添加spark serverless支持

Browse Source
This commit is contained in:
pengluan 2022-07-19 14:55:28 +08:00
parent 2a48da93e2
commit 9952166f6f
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
job-template/job/spark/launcher.py
View File

@ -169,13 +169,16 @@ def launch_sparkjob(name, **kwargs):
while True:
time.sleep(10)
sparkjob = k8s_client.get_one_crd(group=crd_info['group'], version=crd_info['version'],plural=crd_info['plural'], namespace=KFJ_NAMESPACE, name=name)
if sparkjob and (sparkjob['status'] == "Succeeded" or sparkjob['status'] == "Failed"):
if sparkjob:
status = json.loads(sparkjob['status_more']).get('applicationState', {}).get("state", '').upper()
if status=='COMPLETED' or 'FAILED' in status:
break
sparkjob = k8s_client.get_one_crd(group=crd_info['group'],version=crd_info['version'],plural=crd_info['plural'],namespace=KFJ_NAMESPACE,name=name)
print("sparkjob %s finished, status %s"%(name, sparkjob['status']))
if sparkjob['status']!='Succeeded':
print("sparkjob %s finished, status %s"%(name, sparkjob['status_more']))
status = json.loads(sparkjob['status_more']).get('applicationState', {}).get("state", '').upper()
if 'FAILED' in status:
exit(1)
@ -200,7 +203,7 @@ if __name__ == "__main__":
sparkConf = [[x.split('=')[0], x.split('=')[1]] for x in args['sparkConf'].split('\n') if '=' in x]
args['sparkConf'] = dict(zip([x[0] for x in sparkConf],[x[1] for x in sparkConf]))
args['sparkConf']['spark.driver.bindAddress']='0.0.0.0'
# args['sparkConf']['spark.driver.bindAddress']='0.0.0.0' # k8s模式下不能用
hadoopConf = [[x.split('=')[0], x.split('=')[1]] for x in args['hadoopConf'].split('\n') if '=' in x]
args['hadoopConf'] = dict(zip([x[0] for x in hadoopConf], [x[1] for x in hadoopConf]))