mirror of
https://github.com/aaPanel/aaPanel.git
synced 2026-08-20 06:35:47 +02:00
116 lines
3.9 KiB
Python
116 lines
3.9 KiB
Python
import json
|
||
import threading
|
||
import queue
|
||
import time
|
||
import traceback
|
||
|
||
from mod.base.ssh_executor import SSHExecutor
|
||
from mod.project.node.dbutil import ServerNodeDB, ExecutorDB, ExecutorLog
|
||
|
||
|
||
class Task(object):
|
||
def __init__(self, task_id: int, log_id: int):
|
||
self._edb = ExecutorDB()
|
||
self.task = self._edb.ExecutorTask.find("id = ?", (task_id,))
|
||
if not self.task:
|
||
raise RuntimeError("The specified task does not exist")
|
||
if log_id == 0:
|
||
self.task.elogs = self._edb.ExecutorLog.query("executor_task_id = ?", (self.task.id,))
|
||
else:
|
||
self.task.elogs = [self._edb.ExecutorLog.find("executor_task_id = ? AND id = ?", (self.task.id, log_id))]
|
||
if not self.task.elogs:
|
||
raise RuntimeError("Task has no execution entry")
|
||
|
||
self.end_queue = queue.Queue()
|
||
self.end_status = False
|
||
|
||
|
||
def end_func(self):
|
||
self._edb = ExecutorDB()
|
||
while not self.end_queue.empty() or not self.end_status:
|
||
if self.end_queue.empty():
|
||
time.sleep(0.1)
|
||
|
||
elog: ExecutorLog = self.end_queue.get()
|
||
self._edb.ExecutorLog.update(elog)
|
||
|
||
def start(self):
|
||
thread_list = []
|
||
s_db = ServerNodeDB()
|
||
for log in self.task.elogs:
|
||
node = s_db.get_node_by_id(log.server_id)
|
||
if not node:
|
||
log.status = 2
|
||
log.update_log("Node data loss, unable to execute\n")
|
||
self._edb.ExecutorLog.update(log)
|
||
|
||
ssh_conf = json.loads(node["ssh_conf"])
|
||
if not ssh_conf:
|
||
log.status = 2
|
||
log.update_log("Node SSH configuration data lost, unable to execute\n")
|
||
self._edb.ExecutorLog.update(log)
|
||
|
||
thread = threading.Thread(target=self.run_one, args=(ssh_conf, log))
|
||
thread.start()
|
||
thread_list.append(thread)
|
||
|
||
self._edb.close()
|
||
end_th = threading.Thread(target=self.end_func)
|
||
end_th.start()
|
||
|
||
for i in thread_list:
|
||
i.join()
|
||
self.end_status = True
|
||
end_th.join()
|
||
|
||
def run_one(self, ssh_conf: dict, elog: ExecutorLog):
|
||
ssh = SSHExecutor(
|
||
host=ssh_conf["host"],
|
||
port=ssh_conf["port"],
|
||
username=ssh_conf["username"],
|
||
password=ssh_conf["password"],
|
||
key_data=ssh_conf["pkey"],
|
||
passphrase=ssh_conf["pkey_passwd"])
|
||
elog.write_log("Start executing the task\nStart establishing SSH connection...\n")
|
||
try:
|
||
ssh.open()
|
||
def on_stdout(data):
|
||
if isinstance(data, bytes):
|
||
data = data.decode()
|
||
print(data)
|
||
elog.write_log(data)
|
||
|
||
elog.write_log("Start executing script...\n\n")
|
||
t = time.time()
|
||
res_code = ssh.execute_script_streaming(
|
||
script_content=self.task.script_content,
|
||
script_type=self.task.script_type,
|
||
timeout=60*60,
|
||
on_stdout=on_stdout,
|
||
on_stderr=on_stdout
|
||
)
|
||
take_time = round((time.time() - t)* 1000, 2)
|
||
elog.write_log("\n\nExecution completed, time-consuming [{}ms]\n".format(take_time))
|
||
if res_code == 0:
|
||
elog.status = 1
|
||
elog.write_log("Mission accomplished\n", is_end_log=True)
|
||
else:
|
||
elog.status = 3
|
||
elog.write_log("Task exception, return status code is：{}\n".format(res_code), is_end_log=True)
|
||
self.end_queue.put(elog)
|
||
except Exception as e:
|
||
traceback.print_exc()
|
||
elog.status = 2
|
||
elog.write_log("\nTask failed, error：" + str(e), is_end_log=True)
|
||
self.end_queue.put(elog)
|
||
return
|
||
|
||
|
||
# log_id 要执行的子任务，默认为 0，表示执行所有子任务
|
||
def run_executor_task(task_id: int, log_id: int = 0):
|
||
t = Task(task_id, log_id)
|
||
t.start()
|
||
|
||
|
||
|