U
    nËj{
  ã                   @   sl   d dl Zd dlZd dlmZ d dlmZ d dlmZm	Z	 G dd„ deƒZ
G dd„ de
ƒZG d	d
„ d
e
ƒZdS )é    N)Úabstractmethod)ÚBrokenProcessPool)ÚBaseExecutorÚrun_jobc                       s2   e Zd Ze‡ fdd„ƒZdd„ Zddd„Z‡  ZS )	ÚBasePoolExecutorc                    s   t ƒ  ¡  || _d S ©N)ÚsuperÚ__init__Ú_pool)ÚselfÚpool©Ú	__class__© ú>/tmp/pip-unpacked-wheel-giblih11/apscheduler/executors/pool.pyr	   
   s    
zBasePoolExecutor.__init__c                    s6   ‡ ‡fdd„}ˆj  tˆ ˆ j|ˆjj¡}| |¡ 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>.callback)r
   Zsubmitr   Z_jobstore_aliasÚ_loggerÚnameZadd_done_callback)r   r   Ú	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                    s,   |pi }t jjt|ƒf|Ž}tƒ  |¡ d S r   )Ú
concurrentÚfuturesr(   Úintr   r	   ©r   Úmax_workersÚpool_kwargsr   r   r   r   r	   /   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‡ 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                    sD   |pi | _ | j  dt d¡¡ tjjt|ƒf| j Ž}tƒ  	|¡ d S )NZ
mp_contextÚspawn)
r/   Ú
setdefaultÚmultiprocessingZget_contextr*   r+   r1   r,   r   r	   r-   r   r   r   r	   @   s    
ÿÿzProcessPoolExecutor.__init__c                    s^   zt ƒ  ||¡ W nF tk
rX   | j d¡ | jj| jjf| jŽ| _t ƒ  ||¡ Y nX d S )Nz<Process pool is broken; replacing pool with a fresh instance)	r   r!   r   r   Úwarningr
   r   Z_max_workersr/   )r   r   r    r   r   r   r!   H   s    ÿÿÿz"ProcessPoolExecutor._do_submit_job)r)   N)r$   r%   r&   r0   r	   r!   r'   r   r   r   r   r1   5   s   
r1   )Úconcurrent.futuresr*   r4   Úabcr   Zconcurrent.futures.processr   Zapscheduler.executors.baser   r   r   r(   r1   r   r   r   r   Ú<module>   s   