task_kill.py
3.1 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
# coding=utf-8
#author: 4N
#createtime: 2020/9/4
#email: nheweijun@sina.com
from ..models import db,Task,Table,InsertingLayerName,Process
from app.util.component.ApiTemplate import ApiTemplate
from app.util.component.StructurePrint import StructurePrint
import os
import signal
import platform
import datetime
import uuid
from app.util.component.UserCheck import UserCheck
class Api(ApiTemplate):
api_name = "停止任务"
def para_check(self):
pass
def process(self):
res = {}
try:
task_guid = self.para.get("task_guid")
task = Task.query.filter_by(guid=task_guid).one_or_none()
if not task:
raise Exception("任务不存在！")
#验证权限
UserCheck.verify(task.creator)
pid = task.task_pid
try:
if platform.system().lower().__contains__("windows"):
exec_result = os.popen('taskkill.exe /pid:' + str(pid))
else:
#分布式下，很难吧，还需要记录机器，分布式是不可行的
os.kill(pid,signal.SIGILL)
except Exception as e:
StructurePrint.print("Kill task 失败")
raise e
#处理kill任务后的事情
self.fix_task(task)
res["msg"] = "Kill成功！"
res["result"] = True
except Exception as e:
db.session.rollback()
raise e
return res
def fix_task(self,task):
if task.task_type==1:
ilns = InsertingLayerName.query.filter_by(task_guid=task.guid).all()
for iln in ilns:
db.session.delete(iln)
if task.task_type==2:
table = Table.query.filter_by(guid=task.table_guid).one_or_none()
if len(table.relate_table_vacuates.all())>0:
Table.query.filter_by(guid=task.table_guid).update({"is_vacuate":1})
else:
Table.query.filter_by(guid=task.table_guid).update({"is_vacuate": 0})
if task.task_type==3:
pass
if task.task_type==4:
pass
if task.task_type==5:
pass
Task.query.filter_by(guid=task.guid).update({"state":-1})
message = "{} 任务被强行中止！".format(datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S'))
task_process = Process(guid=uuid.uuid1().__str__(), message=message, time=datetime.datetime.now(),
task_guid=task.guid)
db.session.add(task_process)
db.session.commit()
return None
api_doc = {
"tags": ["任务接口"],
"parameters": [
{"name": "task_guid",
"in": "formData",
"type": "string"},
],
"responses": {
200: {
"schema": {
"properties": {
}
}
}
}
}