File: //opt/alt/python37/lib/python3.7/site-packages/pyroute2/ndb/__pycache__/task_manager.cpython-37.pyc
B
��fP& � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z ddl
m Z ddl m
Z
mZmZmZ ddlmZmZmZmZ e�e�Zdd � ZG d
d � d e�ZG d d
� d
�Z dS )� N)�partial)�config� )�schema)�DBMExitException�InvalidateHandlerException�RescheduleException�ShutdownException)�cmsg�
cmsg_event� cmsg_failed� cmsg_sstartc g s. x(| D ] }|d k rx|D ]
}|V qW qW d S )N� )�argv�sequence�itemr r �J/opt/alt/python37/lib/python3.7/site-packages/pyroute2/ndb/task_manager.py�Events s
r c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )� NDBConfigc C s
|| _ d S )N)� task_manager)�selfr r r r �__init__ s zNDBConfig.__init__c C s | j �|�S )N)r Z
config_get)r �keyr r r � __getitem__" s zNDBConfig.__getitem__c C s | j �||�S )N)r Z
config_set)r r �valuer r r � __setitem__% s zNDBConfig.__setitem__c C s | j �|�S )N)r Z
config_del)r r r r r � __delitem__( s zNDBConfig.__delitem__c C s
| j �� S )N)r Z config_keys)r r r r �keys+ s zNDBConfig.keysc C s
| j �� S )N)r Z config_items)r r r r �items. s zNDBConfig.itemsc C s
| j �� S )N)r Z
config_values)r r r r �values1 s zNDBConfig.valuesN)
�__name__�
__module__� __qualname__r r r r r r r r r r r r s r c @ sN e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zddd�Z dd� Z
dS )� TaskManagerc C s6 || _ |j| _i | _|j| _d | _t�� | _| _d S )N) �ndb�log� event_mapZ _event_queue� event_queue�thread�time�ctime�gctime)r r$ r r r r 6 s zTaskManager.__init__c C s( || j krg | j |< | j | �|� d S )N)r& �append)r �event�handlerr r r �register_handler> s
z TaskManager.register_handlerc C s | j | �|� d S )N)r& �remove)r r- r. r r r �unregister_handlerC s z TaskManager.unregister_handlerc C s. t t|dd �t�r|j�t�dt|� � d S )N�payloadz unsupported event ignored: %s)�
isinstance�getattr� Exceptionr2 r% �debug�type)r �targetr- r r r �default_handlerF s zTaskManager.default_handlerc C s, |d d8 < |d dkr(| j j�� d S )N� countdownr r )r$ �
_dbm_ready�set)r �_localsr8 r- r r r �check_sources_startedK s z!TaskManager.check_sources_startedc sh �fdd�}�fdd�}G dd� dt �� � �fdd�}� �fd d
�}|}|}t���r^|}|}� ||fS ) Nc
sj y6x"� |j |j�D ]}|j�|� qW |j�t� � W n. tk
rd } z|j�|� W d d }~X Y nX d S )N)r �kwarg�response�put�
StopIterationr5 )r8 �requestr �e)�methodr r �_do_local_generatorT s z4TaskManager.wrap_method.<locals>._do_local_generatorc
sN y|j �� |j|j�� W n. tk
rH } z|j �|� W d d }~X Y nX d S )N)r@ rA r r? r5 )r8 rC rD )rE r r �_do_local_single\ s z1TaskManager.wrap_method.<locals>._do_local_singlec @ s e Zd Zdd� ZdS )z)TaskManager.wrap_method.<locals>.cmsg_reqc _ s" dd i| d< || _ || _|| _d S )Nr8 �header)r@ r r? )r r@ r r? r r r r f s z2TaskManager.wrap_method.<locals>.cmsg_req.__init__N)r r! r" r r r r r �cmsg_reqe s rI c ? s� | j tt�� �kr.xr�||�D ]
}|V q W nXt�� }� |f|�|�}| j�|f� x0|�� }t |t
�rld S t |t �r||�qV|V qVW d S )N) r( �id� threading�current_thread�queue�Queuer' rA �getr3 rB r5 )r r r? r r@ rC )rI rE r r �_do_dispatch_generatoro s
z7TaskManager.wrap_method.<locals>._do_dispatch_generatorc sf | j tt�� �kr �||�S tjdd�}� |f|�|�}| j�|f� |�� }t |t
�r^|�n|S d S )Nr )�maxsize) r( rJ rK rL rM rN r' rA rO r3 r5 )r r r? r@ rC �ret)rI rE r r �_do_dispatch_single� s
z4TaskManager.wrap_method.<locals>._do_dispatch_single)r
�inspect�isgeneratorfunction)r rE rF rG rP rS r. �proxyr )rI rE r � wrap_methodP s
zTaskManager.wrap_method� c C sx xrt |�D ]f}t||d �}t|d�r
t|jt�r6|j}|� |� �}| �|�\}}}t| |t|| �� |g| j |< q
W d S )N�publish)
�dirr4 �hasattrr3 rY �strrW �setattrr r& )r Zapi_obj�prefix�namerE r- r. rV r r r � register_api� s
zTaskManager.register_apic
s� dt � jj�i}tt�� �� _tdd� gt� fdd�gt t
� j |�gi}|� _ yXt
�� jj� jj� j � j�d��� j_
� �� jj
d� � �� jj
jd� t� �� j_W n6 tk
r� } z|� j_� jj�� d S d }~X Y nX x(� jjD ] }d |d< � jjjf |� q�W x8� jj
j �� D ]&\}}x|D ]}� � ||� �q&W �qW d }d } g }
�xl|�s�� j � � \} } t | |
�} g }
�y��x�| D �]�}|� |j � j!g�}�x`t"|�D �]R}y |d
d } || |� W �q� t#k
�rL d |d
k�r�d
|d
d < |d
d dk �r6|d
d d7 < � j�$d|f � |
�%|� n� j�&d|f � Y n� t'k
�r� y|�(|� W n* tk
�r� � j�&dt)�*� � Y nX Y n\ t+k
�r� d}P Y nB t,k
�r� d S tk
�r� � j�&d|t)�*� f � Y nX �q�W t-�-� � j. tj/k�r�t-�-� � _.�q�W W n� tk
�r� } zr� j�&d|� d| � �� y,� j�$d| � d�� � jj| j0�d� W n( t1k
�r� � j�$d| � �� Y nX W d d }~X Y nX �qRW x�t"� jjj2�D ]�} � jjj(| d d�} | d k �r�| j3d k �rΈ j�$d | � �� | �4� � jj
jd �rD� j�$d | � � jj
�5| � n � j�$d � �q�W � jj
�6� � jj
�4� x � jj7j8D ]}|�4� �qzW d S ) Nr: c S s
|j �� S )N)r2 r< )�t�xr r r �<lambda>� � z!TaskManager.run.<locals>.<lambda>c s � j j�| d�S )Nr )r$ r �mark)ra rb )r r r rc � rd r Zdb_Zconfig_r- FrH r8 Zrcounterr � r z
reschedule %szdrop %sz&could not invalidate event handler:
%sTzcould not load event:
%s
%sz exception <z > in source zrequesting source z restartZrestartzkey error for )�synczclosing source Z
db_cleanupzflush DB for the target %szleave DB for debug)9�lenr$ Z_nlrJ rK rL r( r r r
r r> r& r ZDBSchemar �sourcesr% Zchannelr` r Zbonfigr5 Z
_dbm_errorr; r< �addr r/ r' rO r � __class__r9 �tupler r6 r, �errorr r0 � traceback�
format_excr r r) r+ Z
gc_timeout�state�KeyError�cache�th�close�flush�commit�logger�handlers)
r r= r& rD �specr- rx r. �stop�sourceZ
reschedule�eventsr8 r )r r �run� s�
zTaskManager.runN)rX ) r r! r" r r/ r1 r9 r> rW r` r} r r r r r# 5 s K
r# ) rT �loggingrM rK r) rn � functoolsr Zpyroute2r rX r r| r r r r �messagesr
r r r
� getLoggerr r% r �dictr r# r r r r �<module> s