File: //usr/local/lib/python3.6/site-packages/agent360/plugins/__pycache__/elasticsearch.cpython-36.pyc
3
3 Eg� � @ s� y0d dl mZmZ d dlmZmZ d dlmZ W n@ ek
rp d dlmZ d dl mZ d dl
mZmZmZ Y nX d dl Z d dl Z d dl
Z
d dlZG dd � d e j�Zed
kr�e� j� dS ) � )�urlparse� urlencode)�urlopen�Request)� HTTPError)r )r )r r r Nc @ s e Zd ZdZ dd� ZdS )�Plugin�
elasticsearchc 7 sH dd� }t � }t � }tj|jdd��}tj|�}tj� |d< | j� }d<� fdd � � y"� tj|j � |d
�d d �}W n t
k
r� d
S X d=} i }
t t g} x�|j
� D ]�\} }
| j� j� } x2| D ]*}y ||
�}
W n tk
� r� Y nX � q�W | | k� r�t|
�t k � r�| j| t |
�|�|| < t |
�|
| < n � q�W tj� |
d< | j|
� |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�| j� D ��S )Nc S s t | t�r| jd�S | S )N�ascii)�
isinstance�unicode�encode)�x� r �G/root/tmp/pip-build-f6rmvwrq/agent360/agent360/plugins/elasticsearch.py�<lambda> s z7Plugin.run.<locals>.ascii_encode_dict.<locals>.<lambda>c 3 s | ]}t � |�V qd S )N)�map)�.0�pair)� ascii_encoder r � <genexpr> s 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 sj g }x\| j � D ]P\}}|r&|| | n|}t|tj�rP|j� |||d�j � � q|j||f� qW 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�get_time_in_millis� indexing_index_time_in_millis�flush_total_time_in_millis� indexing_delete_time_in_millis� indexing_throttle_time_in_millis�#merges_total_stopped_time_in_millis�%merges_total_throttled_time_in_millis�merges_total_time_in_millis� recovery_throttle_time_in_millis� refresh_total_time_in_millis�search_fetch_time_in_millis�search_query_time_in_millis� search_scroll_time_in_millis� search_suggest_time_in_millis�warmer_total_time_in_millis�
docs_count� docs_deleted� flush_total�get_exists_total�get_missing_total� get_total�indexing_delete_total�indexing_index_total�indexing_noop_update_total� merges_total�merges_total_docs�#merges_total_auto_throttle_in_bytes�query_cache_cache_count�query_cache_cache_size�query_cache_evictions�query_cache_hit_count�query_cache_miss_count�query_cache_total_count�
refresh_total�request_cache_hit_count�request_cache_miss_count�search_fetch_total�search_open_contexts�search_query_total�search_scroll_total�search_suggest_total�segments_count�%segments_max_unsafe_auto_id_timestamp� warmer_total�get_exists_time_in_millis�get_missing_time_in_millis)r r )/r( r) r* r+ r) r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA rB rC rD rE rF rG rH rI rJ rK rL rM rN rO rP rQ rR rS rT rU )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 Z constructors�key�value�cr )r% r �run s�
"
z
Plugin.runN)�__name__�
__module__� __qualname__rk r r r r r s r �__main__)� urllib.parser r �urllib.requestr r � urllib.errorr � ImportError�urllibrV rX ZpluginsrY r Z
BasePluginr rl �executer r r r �<module> s m