Python多任务多线程任务管理类

task_manage.py  

#coding=utf-8
import threading
import logging
import time

# 多任务多线程任务管理类
class task_manage():
    # name 任务名称
    # task_func 任务函数指针
    # task_args 任务函数参数，默认空
    # check_func 任务是否执行完检测函数指针，返回True表示还有任务没有执行，默认调用自身的runtimes_control函数
    # check_args 任务检测函数指针参数，默认空
    # max_thread_count 最大线程数
    # run_time 任务运行次数，设置check_func后，该参数无效
    def __init__(self,name,task_func,task_args=(),check_func=None,check_args=(),max_thread_count=1,run_time=1):
        self._is_finished = False
        self._task_func = task_func
        self._task_args = task_args
        self._max_thread_count = max_thread_count
        self._threads = []
        self._name = name
        self._check_args = check_args
        self._run_time = run_time
        self._task_index = 0
        if check_func is None:
            self._check_func = self.runtimes_control
            self._check_args = ()
        else:
            self._check_func = check_func
    # 任务运行次数控制函数
    def runtimes_control(self):
        if self._run_time > 0:
            self._run_time -= 1
            return True
        return False
    # 清除已退出线程
    def clear_exit_threads(self):
        for t in self._threads[:]:
            if not t.is_alive() :
                self._threads.remove(t)
    # 运行任务
    def run(self):
        while(len(self._threads)<self._max_thread_count and not self._is_finished):
            if self._check_func(*self._check_args):
                t = threading.Thread(target=self._task_func,args=self._task_args)
                self._threads.append(t)
                t.setDaemon(True)
                t.start()
                self._task_index += 1
                logging.debug("%s run %s" %(self._name,self._task_index))
            else:
                self._is_finished = True
                break
    # 对外接口，检测是否所有任务都执行完成
    def is_finish(self):
        self.clear_exit_threads()
        self.run()
        return self._is_finished and len(self._threads)==0


def run_task_until_all_finished(manages=[]):
    while True:
        all_finished = True
        for manage in manages:
            all_finished = manage.is_finish() and all_finished
        if all_finished : break
        time.sleep(1)
    logging.debug('all have finished!')

task_test.py           

#coding=utf-8

from base.task_manage import *

def watch_movie(move_name):
    print("I am watching movie %s now." %(move_name))

def listen_music(music_name):
    print("I am listening music %s now." %(music_name))


manages=[]
manages.append( task_manage(name="watching movie",task_func=watch_movie,task_args=(u"捉妖记",),run_time=10,max_thread_count=3))
manages.append( task_manage(name="listening music",task_func=listen_music,task_args=(u"笨小孩",),run_time=10,max_thread_count=3))

run_task_until_all_finished(manages)
print('all have finished!')
quit()

[Python]代码    

运行结果：
I am watching movie 捉妖记 now.
I am watching movie 捉妖记 now.
I am watching movie 捉妖记 now.
I am listening music 笨小孩 now.
I am listening music 笨小孩 now.
I am listening music 笨小孩 now.
I am watching movie 捉妖记 now.
I am watching movie 捉妖记 now.
I am watching movie 捉妖记 now.
I am listening music 笨小孩 now.
I am listening music 笨小孩 now.
I am listening music 笨小孩 now.
I am watching movie 捉妖记 now.
I am watching movie 捉妖记 now.
I am watching movie 捉妖记 now.
I am listening music 笨小孩 now.
I am listening music 笨小孩 now.
I am listening music 笨小孩 now.
I am watching movie 捉妖记 now.
I am listening music 笨小孩 now.
all have finished!

标签： 代码

版权申明：本站文章部分自网络，如有侵权，请联系：west999com@outlook.com
特别注意：本站所有转载文章言论不代表本站观点！
本站所提供的图片等素材，版权归原作者所有，如需使用，请与原作者联系。