File: //lib/python3.6/site-packages/agent/common/__pycache__/netinfo.cpython-36.pyc
3
�*�gZ � @ sF d dl Z d dlZd dlZd dlmZmZ e jd�ZG dd� de�Z dS )� N)�
shell_command� sys_commandZkeentunec @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�NetInfoc C s� | j � xV| jD ]L\}}y$| j|� | j|� | j||� W n tk
rR wY qX || _P qW | jsjt�tt ddd��| _
tt ddd��| _ | j � | j | j
d | j
d d kr�tjd| j � d| j
d | j
d d � d �� d S )
Nz8cat /proc/cpuinfo| grep 'physical id'| sort| uniq| wc -lF)�logz*cat /proc/cpuinfo| grep 'processor'| wc -l� r z cpu number(z ) unmatch with numa node number(z)!)�
_load_pci_dev� pci_dev_list� _load_pci_id�_load_queue_list�_load_interrupts� Exception�dev_name�AssertionError�intr Zcpu_num� processor�_load_numa_node_core_range�numa_node_core_range�loggerZwarning)�self�
virtio_devr
� r � /usr/lib/python3.6/netinfo.py�__init__ s"
zNetInfo.__init__c C s� g | _ x�tddd�jd�D ]�}tjjd|�}tjj|�s:qyVtj|�}tj d|�r�tj d|�r~| j j
tj d|�j d�|f� n| j j
||f� W q t k
r� wY qX qW d S ) Nzls /sys/class/netF)r �
z/sys/class/netzpci\d+zvirtio\dr )
r r �split�os�path�join�isdir�readlink�re�search�append�group�OSError)r Zdev�_pathZ
_link_pathr r r r ) s
zNetInfo._load_pci_devc C s8 t dd|gdd�}tjd|�jd�| _| jdks4t�d S )NZethtoolz-iF)r zbus-info: (.*)\nr � )r r r! r# �pci_idr )r r
Zpci_infor r r r ; s zNetInfo._load_pci_idc C s� t dd|� d�gdd�}tjd�tjd� }}tj||� sJtj||� rXtdj|���ttj||�d d
� d �| _ttj||�d d
� d �| _ d S )
NZlsz/sys/class/net/z/queues/F)r ztx-\d+zrx-\d+z5can not find tx/rx queue in /sys/class/net/{}/queues/c S s t tjd| �jd��S )Nz(\d+)r )r r r! r# )�xr r r �<lambda>F s z*NetInfo._load_queue_list.<locals>.<lambda>)�keyc S s t tjd| �jd��S )Nz(\d+)r )r r r! r# )r( r r r r) G s )
r r �compiler! r �format�sorted�findallZtx_queueZrx_queue)r r
Z
queue_infoZ
tx_patternZ
rx_patternr r r r
@ s zNetInfo._load_queue_listc C s2 g }x(|D ] }t jd|�jd�}|j|� q
W |S )Nz^(\d+):r )r r! r# r" )r �
interrupts�result�iZ
interrupts_idr r r Z__parse_interrupts_codeI s
z NetInfo.__parse_interrupts_codec s� dd� t ddd�jd�D �}� fdd�|D �}|j� dkrL�fd d�|D �}|j� dkrj� fd
d�|D �}|j� dkszt�| j|�| _| jj� dks�t�| jdd d � | _| jd d d � | _d S )
Nc S s g | ] }|j � �qS r )�strip)�.0r1 r r r �
<listcomp>Q s z,NetInfo._load_interrupts.<locals>.<listcomp>zcat /proc/interruptsF)r r c s( g | ] }t j� � d � � d�|�r|�qS )z-input|z-output)r r! )r3 r1 )r
r r r4 S s r c s( g | ] }t j� � d � � d�|�r|�qS )z-input|z-output)r r! )r3 r1 )r r r r4 V s c s g | ]}t j� |�r|�qS r )r r! )r3 r1 )r
r r r4 Y s � r )r r �__len__r � _NetInfo__parse_interrupts_coder/ Zinput_interrupts_listZoutput_interrupts_list)r r r
ZcontentZinterrupts_listr )r
r r r P s zNetInfo._load_interruptsc C s t dd| jgdd�}tjd|�r4tjd|�jd�}nd}tddd�}| jdkr�tjd j|��}tjd
j|��}tj||�r�tj||�jd�gd }n8tj||�r�tj||�jd�gd }nt j
d � t d ���n"| jd k�r`tjd
j|��}tjdj|��}tj||��r$tj||�jd�j d�}n:tj||��rLtj||�jd�j d�}nt j
d� t d��n�tjdj|��}tjdj|��}tj||��r�tj||�jd�j d�}nJtj||��r�tj||�jd�j d�}n"t j
d| j� �� t dj| j���dd� |D �| _
d S )NZlspciz-vvvsF)r zNUMA node: (\d)\nr r Zlscpuu NUMA 节点{} CPU：\s*(\d+)z NUMA node{} CPU\(s\):\s*(\d+)r5 z0can not find numa_node_core_range in process = 1u NUMA 节点{} CPU：\s*(\d+,\d+)z!NUMA node{} CPU\(s\):\s*(\d+,\d+)�,z0can not find numa_node_core_range in process = 2u NUMA 节点{} CPU：\s*(\d+-\d+)z!NUMA node{} CPU\(s\):\s*(\d+-\d+)�-z/can not find numa_node_core_range in process = z1can not find numa_node_core_range in process = {}c S s g | ] }t |��qS r )r )r3 r1 r r r r4 � s z6NetInfo._load_numa_node_core_range.<locals>.<listcomp>)r r' r r! r# r r r+ r, r �errorr r r )r Z numa_messageZ
numa_node_numZ cpu_messageZ
pattern_en�patternr r r r r c sD
z"NetInfo._load_numa_node_core_rangeN)
�__name__�
__module__� __qualname__r r r r
r7 r r r r r r r
s r )
Zloggingr r Zagent.common.systemr r Z getLoggerr �objectr r r r r �<module> s