File: //usr/local/lib/python3.9/site-packages/agent360/plugins/__pycache__/iostat.cpython-39.pyc
a
q�dgk � @ sj d dl Z d dlZd dlZd dlZd dlZd dlZd dlZddd�ZG dd� dej�Z e
dkrfe � � � dS )� Nc
C s~ d}i }t j�d�sdS g d�}g d�}g d�}g d�}t|d��� }|D �],}|dkrZqJ|�� } t| �t|�krx|}
nDt| �t|�kr�|}
n.t| �t|�kr�|}
nt| �t|�krJ|}
nqJtt|
| ��} | d d d
� d kr�| d d d
� dkr�n | d d
d � � � du �rqJd| d v sJd| d v �r,qJ| d u�rF| | d k�rFqJ| D ] } | d k�rJt
| | �| | < �qJ| || d < qJ|S )Nz/proc/diskstatsF)�m�mm�dev�reads�rd_mrg�
rd_sectors�
ms_reading�writes�wr_mrg�
wr_sectors�
ms_writing�cur_ios� ms_doing_io� ms_weighted)r r r r r r r r r
r r r
r r �discards�discards_merged�discarded_sectors�discarded_time)r r r r r r r r r
r r r
r r r r r r �flush�
flush_time)r r r r r r r �r� r � Znvm�������nT�loopZram) �os�path�isfile�open� readlines�split�len�dict�zip�isdigit�int)
r � file_path�resultZ columns_diskZcolumns_disk_418Zcolumns_disk_55Zcolumns_partition�lines�liner" �columns�data�key� r/ �A/usr/local/lib/python3.9/site-packages/agent360/plugins/iostat.py�diskstats_parse
sH
(
r1 c @ s e Zd ZdZ dd� ZdS )�PluginZiostatc
G s d}i }t � � |d< | �� }t� }|r.|du r�i }zHtjdd�}|�� D ].\}} i }
| jD ]} t| | �|
| < qZ|
||< qHW n* ty� } z| j }W Y d } ~ n
d } ~ 0 0 �ndi }|�� D �]T\}} i }
i ||< t � � || d< z || W n i ||< Y n0 | �� D ]R\} }
| |v �rNz| �
| |
|| �|
| < W n Y n0 |
|| | < n|
|
| < �qz(|
d |
d |
d |
d |
d
< W n d |
d
< Y n0 z|
d |
d |
d < W n d |
d < Y n0 z(d
|
d d|d |d |
d< W n d |
d< Y n0 |
||< q�| � |� |S )N)r r r r r r r
r r r r r r r r �tsFT)Zperdiskr r r r zavgrq-szr Ztps�d r i� �usage) �timeZget_agent_cacher1 �psutilZdisk_io_counters�items�_fields�getattr� Exception�messageZabsolute_to_per_secondZset_agent_cache)�selfZunusedZ
delta_keysZ
next_cacheZ
prev_cache�disks�resultsZdiskdataZdevice�valuesZ device_stats� key_value�e�valuer/ r/ r0 �runO s`
((
z
Plugin.runN)�__name__�
__module__� __qualname__rD r/ r/ r/ r0 r2 L s r2 �__main__)N) r �signal�
subprocess�sysr7 Zpluginsr6 r1 Z
BasePluginr2 rE �executer/ r/ r/ r0 �<module> s
?J