如何批量杀死 Apache DolphinScheduler 运行中的工作流?

在调研了 DolphinScheduler 之后,在项目上实际使用了一段时间,遇到了任务过多僵死的问题,解决思路分享如下。

关键词: 大数据、数据调度、工作流、批量停止

整体说明

在调研了 DolphinScheduler 之后,在项目上实际使用了一段时间,遇到了任务过多僵死的问题,解决思路分享如下。

17555000230402

问题背景

  • 任务依赖很多:从下图可以看出,有些任务的后续依赖很多,一旦失败,后续任务都得等待。这样就很容易导致后续任务被堵死。

17555000391941

  • 循环任务很多,有很多任务本身是循环调用的其他任务的任务,会不停生成任务,循环任务本身也是任务,这就有可能导致任务堵死。

17555000570087

问题现象

就是很多任务,都在运行中的状态,但是就是不执行,因为他们占用了各自任务组的上限,导致卡死。当任务积累到一定程度,手工去杀死,太慢了,而且真的很累。

问题分析

我们现在就是要杀死这些任务,但是有一些要注意的点:

  • 任务是全量任务,杀死不丢失数据: 只有全量任务,今天才能随意的去杀死,如果是增量的,就需要手工单独处理

  • 下游任务今日不生成数据,不影响展示结果:下游的任务,或者说给客户展示的数据,这种高风险数据,不会因为今天的任务被杀死,而影响展示效果,数据一天不更新,不影响最终结果

解决方案

  • 调用API批量杀死工作流:DolphinScheduler 是自带了很多的 API 接口的,可以批量新建或者停止工作流

  • Python脚本:所以写了个脚本来完成这个事

脚本名称:dolpschedule-kill.py

# -*- coding: utf-8 -*-import requests# 项目环境,没法安装 python3 ,所以这个是 python2.7 的代码# 配置信息BASE_URL = "http://XXX.XXX.XXX.XXX:12345/dolphinscheduler"# 项目编码# 查询后台数据库,或者页面F12 - 项目管理 - 项目列表 - 项目编码PROJECT_CODE = "12194663850176"# 1. 获取 Token# 登录页面 - 安全中心 - 令牌管理 - 创建令牌token = "6bff15e17667d95fdffceda08a19cc6c"# 2. 获取正在运行任务列表def get_running_tasks(token, pageNo=1, pageSize=10):    headers = {"token": token}    task_list_url = "{0}/projects/{1}/process-instances?pageNo={2}&pageSize={3}&stateType=RUNNING_EXECUTION".format(        BASE_URL, PROJECT_CODE, pageNo, pageSize)    resp = requests.get(task_list_url, headers=headers)    return [item['id'] for item in resp.json()['data']['totalList']]# 3. 批量停止任务def batch_stop_tasks(token, task_ids):    headers = {"token": token}    for task_id in task_ids:        stop_url = "{0}/projects/{1}/executors/execute?processInstanceId={2}&executeType=STOP".format(            BASE_URL, PROJECT_CODE, task_id)        resp = requests.post(stop_url, headers=headers)        print("Task {0} stopped: {1}".format(task_id, resp.status_code))# 主流程if __name__ == "__main__":    # 每次执行,杀死 100 个任务,可调整    running_tasks_ids = get_running_tasks(token, pageNo=1, pageSize=100)    print("Found {0} running tasks".format(len(running_tasks_ids)))    batch_stop_tasks(token, running_tasks_ids)
  • 脚本执行

python dolpschedule-kill.py
  • 执行效果

每个任务返回 200 ,说明执行任务成功

17555004719068

最终解决

最终,僵死的工作流全部被我杀死了,不过有时候会有一些僵死的任务实例(不是工作流)存在,这个没法用接口杀死。只能在后台修改,可以看我之前的文章 《DolphinScheduler 6 个高频 SQL 操作技巧》 解决无法被杀死的僵死任务实例。

原文链接:https://blog.csdn.net/pengpenhhh/article/details/149134569