43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
# =====================================================
|
|
# =============== 全局线程池 异步任务接口 ===============
|
|
# =====================================================
|
|
|
|
from PySide2.QtCore import QThreadPool, QRunnable
|
|
|
|
from umi_log import logger
|
|
|
|
# 全局线程池
|
|
GlobalThreadPool = QThreadPool.globalInstance()
|
|
|
|
|
|
# 异步类
|
|
class Runnable(QRunnable):
|
|
def __init__(self, taskFunc, *args, **kwargs):
|
|
super().__init__()
|
|
self._taskFunc = taskFunc
|
|
self._args = args
|
|
self._kwargs = kwargs
|
|
|
|
def run(self):
|
|
try:
|
|
self._taskFunc(*self._args, **self._kwargs)
|
|
except Exception:
|
|
logger.error("异步运行发生错误。", exc_info=True, stack_info=True)
|
|
|
|
|
|
# 启动异步类
|
|
def threadPoolStart(runnable: QRunnable):
|
|
# 检查线程池是否满，并扩充
|
|
activeThreadCount = GlobalThreadPool.activeThreadCount()
|
|
if activeThreadCount >= GlobalThreadPool.maxThreadCount():
|
|
logger.debug(f"线程池已满 {activeThreadCount} ！自动扩充+1。")
|
|
GlobalThreadPool.setMaxThreadCount(activeThreadCount + 1)
|
|
GlobalThreadPool.start(runnable)
|
|
|
|
|
|
# 快捷接口：异步运行函数，返回异步类的对象
|
|
def threadRun(taskFunc, *args, **kwargs):
|
|
runnable = Runnable(taskFunc, *args, **kwargs)
|
|
threadPoolStart(runnable)
|
|
return runnable
|