File: //usr/local/lib/python3.9/site-packages/agent360/plugins/__pycache__/elasticsearch.cpython-39.pyc
a
q�dg� � @ s� z0d dl mZmZ d dlmZmZ d dlmZ W n> eyn d dlmZ d dl mZ d dl
mZmZmZ Y n0 d dl Z d dl Z d dl
Z
d dlZG dd � d e j�Zed
kr�e� �� dS ) � )�urlparse� urlencode)�urlopen�Request)� HTTPError)r )r )r r r Nc @ s e Zd ZdZ dd� ZdS )�Plugin�
elasticsearchc
s4 dd� }t � }t � }t�|�dd��}t�|�}t�� |d< | �� }d� fdd � � z"� tj|� � |d
�d d �}W n t
y� Y d
S 0 d} i }
t t g} |�
� D ]v\} }
| �� �� } | D ]$}z ||
�}
W q� ty� Y q�0 q�| | v r�t|
�t ur�| �| t |
�|�|| < t |
�|
| < q�q�t�� |
d< | �|
� |S )z�
experimental monitoring plugin for elasticsearch
Add to /etc/agent360.ini:
[elasticsearch]
enabled = yes
status_page_url = http://127.0.0.1:9200/_stats
c s" dd� � t � fdd�| �� D ��S )Nc S s t | t�r| �d�S | S )N�ascii)�
isinstance�unicode�encode)�x� r �H/usr/local/lib/python3.9/site-packages/agent360/plugins/elasticsearch.py�<lambda> � z7Plugin.run.<locals>.ascii_encode_dict.<locals>.<lambda>c 3 s | ]}t � |�V qd S )N)�map)�.0�pair�� ascii_encoder r � <genexpr> r z8Plugin.run.<locals>.ascii_encode_dict.<locals>.<genexpr>)�dict�items)�datar r r �ascii_encode_dict s z%Plugin.run.<locals>.ascii_encode_dictr Zstatus_page_url�ts� �_c sf g }| � � D ]P\}}|r$|| | n|}t|tj�rN|�� |||d�� � � q |�||f� q t|�S )N)�sep)r r
� collections�MutableMapping�extend�appendr )�dZ
parent_keyr r �k�vZnew_key��flattenr r r( ' s zPlugin.run.<locals>.flatten)� object_hookZ_all�totalF)/Zget_time_in_millis� indexing_index_time_in_millisZflush_total_time_in_millisZ indexing_delete_time_in_millisr+ Z indexing_throttle_time_in_millisZ#merges_total_stopped_time_in_millisZ%merges_total_throttled_time_in_millisZmerges_total_time_in_millisZ recovery_throttle_time_in_millisZ refresh_total_time_in_millisZsearch_fetch_time_in_millisZsearch_query_time_in_millisZ search_scroll_time_in_millisZ search_suggest_time_in_millisZwarmer_total_time_in_millisZ
docs_countZ docs_deletedZ flush_totalZget_exists_totalZget_missing_totalZ get_totalZindexing_delete_totalZindexing_index_totalZindexing_noop_update_totalZ merges_totalZmerges_total_docsZ#merges_total_auto_throttle_in_bytesZquery_cache_cache_countZquery_cache_cache_sizeZquery_cache_evictionsZquery_cache_hit_countZquery_cache_miss_countZquery_cache_total_countZ
refresh_totalZrequest_cache_hit_countZrequest_cache_miss_countZsearch_fetch_totalZsearch_open_contextsZsearch_query_totalZsearch_scroll_totalZsearch_suggest_totalZsegments_countZ%segments_max_unsafe_auto_id_timestampZ warmer_totalZget_exists_time_in_millisZget_missing_time_in_millis)r r )r �urllib2r �getr �timeZget_agent_cache�json�loads�read� Exception�str�floatr �lower�strip�
ValueError�typeZabsolute_to_per_secondZset_agent_cache)�self�configr �resultsZ
next_cache�requestZ raw_responseZ
prev_cache�jZ
delta_keysr � constructors�key�value�cr r' r �run s:
" 2
z
Plugin.runN)�__name__�
__module__� __qualname__rB r r r r r s r �__main__)� urllib.parser r �urllib.requestr r � urllib.errorr � ImportError�urllibr, r. Zpluginsr/ r Z
BasePluginr rC �executer r r r �<module> s m