es_jvm_pools

#!/usr/bin/env python
# -*- coding: utf-8 -*-

##############################################
# Munin plugin for Elasticsearch monitoring  #
# by Andrii Gakhov <andrii.gakhov@gmail.com> #
##############################################

import json
import urllib3

from munin import MuninPlugin


class ESJvmMemPoolSizePlugin(MuninPlugin):
    category = 'Elasticsearch'
    args = '--base 1000 --lower-limit 0'
    vlabel = 'JVM memory pools size / bytes'
    info = 'Show JVM memory pools size'

    @property
    def title(self):
        return 'Elasticsearch JVM Memory Pools size'

    @property
    def fields(self):
        fields = [
            ('young', dict(
                label='young generation',
                type='GAUGE',
            )),
            ('old', dict(
                label='old generation',
                type='GAUGE',
            ))
        ]
        return fields

    def __init__(self):
        super(ESJvmMemPoolSizePlugin, self).__init__()
        self.es_host = 'http://localhost:9200'
        self.http = urllib3.PoolManager()

    def execute(self):
        return self._get_cache_stats()

    def _get_cache_stats(self):
        url = '{}/_nodes/_local/jvm/stats'.format(self.es_host)
        response = self.http.request('GET', url)

        if response.status != 200:
            return None

        data = json.loads(response.data)
        if not data.get('nodes', {}).values():
            return None

        stats = data['nodes'].values()[0].get('jvm', {}).get('mem', {}).get('pools', {})
        return {
            'young': stats.get('Par Eden Space', {}).get('used_in_bytes'),
            'old': stats.get('CMS Old Gen', {}).get('used_in_bytes')
        }

if __name__ == '__main__':
    ESJvmMemPoolSizePlugin().run()