多线程抽象编程模型(Python版)

最近需要完成一个多线程下载的工具,对其中的多线程下载进行了一个抽象,可以对所有需要使用到多线程编程的地方统一使用这个模型来进行编写。
主要结构:
1、基于Queue标准库实现了一个类似线程池的工具,用户指定提交任务线程submitter与工作线程worker数目,所有线程分别设置为后台运行,提供等待线程运行完成的接口。
2、所有需要完成的任务抽象成task,提供单独的无参数调用方式,供worker线程调用;task以生成器的方式作为参数提供,供submitter调用。
3、所有需要进行线程交互的信息放在context类中。
主要实现代码如下:

#Submitter线程类实现,主要是`task_generator`调用
class SubmitterThread(threading.Thread):
    _DEFAULT_WAIT_TIMEOUT = 2 #seconds
    def __init__(self, queue, task_gen, timeout=2):
        super(SubmitterThread, self).__init__()
        self.queue = queue
        if not isinstance(timeout, int):
            _logger.error('Thread wait timeout value error: %s, '
                          'use default instead.' % timeout)
            self.timeout = self._DEFAULT_WAIT_TIMEOUT
        self.timeout = timeout
        self.task_generator = task_gen

    def run(self):
        while True:
            try:
                task = self.task_generator.next()
                self.queue.put(task, True, self.timeout)
            except Queue.Full:
                _logger.debug('Task queue is full. %s wait %d second%s timeout' %
                              (self.name, self.timeout, 's' if (self.timeout > 1) else ''))
                break
            except (StopIteration, ValueError) as e:
                _logger.debug('Task finished')
                break
#Worker线程实现,主要就是try块内的func调用
class WorkerThread(threading.Thread):
    _DEFAULT_WAIT_TIMEOUT = 2 #seconds
    def __init__(self, queue, timeout=2):
        super(WorkerThread, self).__init__()
        self.queue = queue
        if not isinstance(timeout, int):
            _logger.error('Thread wait timeout value error: %s, '
                          'use default instead.' % timeout)
            self.timeout = self._DEFAULT_WAIT_TIMEOUT
        self.timeout = timeout

    def run(self):
        while True:
            try:
                func = self.queue.get(True, self.timeout)
            except Queue.Empty:
                _logger.debug('Task queue is empty. %s wait %d second%s timeout' %
                              (self.name, self.timeout, 's' if (self.timeout > 1) else ''))
                break

            if not callable(func):
                time.sleep(1)
            try:
                func()
            except Exception as e:
                _logger.error('Thread %s running occurs error: %s' %
                              (self.name, e))
                print('Thread running error: %s' % e)
class Executor(object):
    """
    The really place to execute executor
    """
    thread_list = []
    submitters = 0
    workers = 0
    queue = None
    task_generator = None
    timeout = 0
    def __init__(self, task_gen, submitters=1, workers=1 , timeout=2):
        if len(self.thread_list) != 0:
            raise RuntimeError('Executor can only instance once.')
        self.queue = Queue.Queue(maxsize=submitters * 2 + workers * 2)
        self.submitters = submitters
        self.workers = workers
        self.task_generator = task_gen
        self.timeout = timeout

    def start(self):
        for i in range(self.submitters):
            submitter = SubmitterThread(self.queue, self.task_generator, self.timeout)
            self.thread_list.append(submitter)
            submitter.setName('Submitter-%d' % i)
            submitter.setDaemon(True)
            submitter.start()
        for i in range(self.workers):
            worker = WorkerThread(self.queue, self.timeout)
            self.thread_list.append(worker)
            worker.setName('Worker-%d' % i)
            worker.setDaemon(True)
            worker.start()

    def is_alive(self):
        alive = False
        for t in self.thread_list:
            if t.isAlive():
                alive = True
                break
        return alive

    def wait_to_shutdown(self):
        _logger.debug('Start to wait to shutdown')
        for t in self.thread_list:
            t.join()
            _logger.debug('Shutdown thread : %s' % t.name)

Executor类保存了线程池,提供相应接口。有了这个抽象之后,只需要实例化Executor类的对象,然后调用start方法进行多线程任务的运行。并可以用is_alive等接口再主线程内进行其他处理。
后续再使用这个抽象进行实际多线程任务的实现。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值