U
    “öh´	  ã                   @   s‚   d dl mZ d dlZd dlmZmZ zd dlmZ W n e	k
rL   dZY nX G dd„ deƒZ
G dd„ de
ƒZG d	d
„ d
e
ƒZdS )é    )ÚabstractmethodN)ÚBaseExecutorÚrun_job)ÚBrokenProcessPoolc                       s2   e Zd Ze‡ fdd„ƒZdd„ Zddd„Z‡  ZS )	ÚBasePoolExecutorc                    s   t t| ƒ ¡  || _d S ©N)Úsuperr   Ú__init__Ú_pool)ÚselfÚpool©Ú	__class__© ú>/tmp/pip-unpacked-wheel-ehb4gh6l/apscheduler/executors/pool.pyr	      s    zBasePoolExecutor.__init__c              
      sˆ   ‡ ‡fdd„}zˆj  tˆ ˆ j|ˆjj¡}W nL tk
rx   ˆj d¡ ˆj  ˆj j	¡ˆ_ ˆj  tˆ ˆ j|ˆjj¡}Y nX | 
|¡ d S )Nc                    sZ   t | dƒr|  ¡ n|  ¡ t|  ¡ dd ƒf\}}|rDˆ ˆ j||¡ nˆ ˆ j|  ¡ ¡ d S )NÚexception_infoÚ__traceback__)Úhasattrr   Ú	exceptionÚgetattrZ_run_job_errorÚidZ_run_job_successÚresult)ÚfÚexcÚtb©Újobr   r   r   Úcallback   s    ÿz1BasePoolExecutor._do_submit_job.<locals>.callbackz<Process pool is broken; replacing pool with a fresh instance)r
   Zsubmitr   Z_jobstore_aliasZ_loggerÚnamer   Úwarningr   Z_max_workersZadd_done_callback)r   r   Z	run_timesr   r   r   r   r   Ú_do_submit_job   s     zBasePoolExecutor._do_submit_jobTc                 C   s   | j  |¡ d S r   )r
   Úshutdown)r   Úwaitr   r   r   r!   $   s    zBasePoolExecutor.shutdown)T)Ú__name__Ú
__module__Ú__qualname__r   r	   r    r!   Ú__classcell__r   r   r   r   r      s   r   c                       s"   e Zd ZdZd‡ fdd„	Z‡  ZS )ÚThreadPoolExecutora  
    An executor that runs jobs in a concurrent.futures thread pool.

    Plugin alias: ``threadpool``

    :param max_workers: the maximum number of spawned threads.
    :param pool_kwargs: dict of keyword arguments to pass to the underlying
        ThreadPoolExecutor constructor
    é
   Nc                    s0   |pi }t jjt|ƒf|Ž}tt| ƒ |¡ d S r   )Ú
concurrentÚfuturesr'   Úintr   r	   ©r   Úmax_workersÚpool_kwargsr   r   r   r   r	   3   s    zThreadPoolExecutor.__init__)r(   N©r#   r$   r%   Ú__doc__r	   r&   r   r   r   r   r'   (   s   
r'   c                       s"   e Zd ZdZd‡ fdd„	Z‡  ZS )ÚProcessPoolExecutora#  
    An executor that runs jobs in a concurrent.futures process pool.

    Plugin alias: ``processpool``

    :param max_workers: the maximum number of spawned processes.
    :param pool_kwargs: dict of keyword arguments to pass to the underlying
        ProcessPoolExecutor constructor
    r(   Nc                    s0   |pi }t jjt|ƒf|Ž}tt| ƒ |¡ d S r   )r)   r*   r1   r+   r   r	   r,   r   r   r   r	   D   s    zProcessPoolExecutor.__init__)r(   Nr/   r   r   r   r   r1   9   s   
r1   )Úabcr   Úconcurrent.futuresr)   Zapscheduler.executors.baser   r   Zconcurrent.futures.processr   ÚImportErrorr   r'   r1   r   r   r   r   Ú<module>   s   
