elasticsearch的监控脚本

监控elasticsearch cluster 集群

通过主动模式将数据发送给zabbix server

 import json
 import struct
 import socket
 import re
 from elasticsearch import Elasticsearch

 def get_dict_value(res):
     for k, v in res.items():
         value = v
     return value

 def get_metric_trapper(hosts):
     sending_data = {'request': 'sender data', 'data': []}
     es = Elasticsearch(hosts=hosts)
     res = es.nodes.stats(node_id='_local')['nodes']
     v = get_dict_value(res)
     host_name = v['name']
     for key_info in monitor_key_info:
         key_one = key_info['key_one']
         key_two = key_info['key_two']
         key_three = key_info['key_three']
         sending_data['data'].append({
             'host': host_name,
             'value': v[key_one][key_two][key_three],
             'key': 'es.{}.{}.{}'.format(key_one, key_two, key_three)
         })
     return sending_data

 def send_data(sender_data):
     code =
     if len(sender_data['data']) == :
         code =
         return code
     mydata = json.dumps(sender_data)
     data_length = len(mydata)
     data_header = str(struct.pack('q', data_length))
     data_to_send = 'ZBXD\1' + str(data_header) + str(mydata)
     socket.setdefaulttimeout()
     try:
         sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
         sock.connect(('zabbix-IP', ))
         sock.send(data_to_send)
     except Exception as e:
         print(e)
         code =
         return code
     response_header = sock.recv()
     if not response_header == 'ZBXD\1':
         code =
         return code
     response_data_header = sock.recv()
     response_data_header = response_data_header[:]
     response_len = struct.unpack('i', response_data_header)[]
     response_raw = sock.recv(response_len)
     sock.close()
     response = json.loads(response_raw)
     match = re.match('^.*failed.+?(\d+).*$', response['info'].lower() if 'info' in response else '')
     if match is None:
         code =
         return code
     else:
         fails = int(match.group())
         if fails > :
             code =
             return code
     return code

 if __name__ == '__main__':
     hosts = 'ip:9200'
     monitor_key_info = [
         {'key_one': 'jvm', 'key_two': 'mem', 'key_three': 'heap_used_percent'},
         {'key_one': 'jvm', 'key_two': 'mem', 'key_three': 'heap_used_in_bytes'},
         {'key_one': 'jvm', 'key_two': 'mem', 'key_three': 'heap_max_in_bytes'},
         {'key_one': 'indices', 'key_two': 'store', 'key_three': 'size_in_bytes'},
         {'key_one': 'indices', 'key_two': 'segments', 'key_three': 'count'},
         {'key_one': 'indices', 'key_two': 'search', 'key_three': 'query_time_in_millis'},
         {'key_one': 'indices', 'key_two': 'indexing', 'key_three': 'index_time_in_millis'},
         {'key_one': 'indices', 'key_two': 'docs', 'key_three': 'count'},
         {'key_one': 'indices', 'key_two': 'docs', 'key_three': 'deleted'},
         {'key_one': 'os', 'key_two': 'cpu', 'key_three': 'percent'},
     ]
     v = get_metric_trapper(hosts=hosts)
     relust = send_data(v)
     print(relust)