com.automattic.elasticsearch.statsd.StatsdReporterIndices Maven / Gradle / Ivy

Go to download
package com.automattic.elasticsearch.statsd;

import org.elasticsearch.action.admin.indices.stats.CommonStats;
import org.elasticsearch.action.admin.indices.stats.IndexShardStats;
import org.elasticsearch.action.admin.indices.stats.IndexStats;
import org.elasticsearch.action.admin.indices.stats.IndicesStatsResponse;

public class StatsdReporterIndices extends StatsdReporterIndexStats {

    private final IndicesStatsResponse indicesStatsResponse;
    private final Boolean reportIndices;
    private final Boolean reportShards;

    public StatsdReporterIndices(IndicesStatsResponse indicesStatsResponse, Boolean reportIndices, Boolean reportShards) {
        this.indicesStatsResponse = indicesStatsResponse;
        this.reportIndices = reportIndices;
        this.reportShards = reportShards;
    }

    public void run() {
        try {
            // First report totals
            this.sendCommonStats(
                    this.buildMetricName("indices"),
                    this.indicesStatsResponse.getTotal()
            );

            if (this.reportIndices) {
                for (IndexStats indexStats : this.indicesStatsResponse.getIndices().values()) {
                    String indexPrefix = "index." + indexStats.getIndex();

                    this.sendCommonStats(
                            this.buildMetricName(indexPrefix + ".total"),
                            indexStats.getTotal()
                    );

                    if (this.reportShards) {
                        for (IndexShardStats indexShardStats : indexStats.getIndexShards().values()) {
                            this.sendCommonStats(
                                    this.buildMetricName(indexPrefix + "." + indexShardStats.getShardId().id()),
                                    indexShardStats.getTotal()
                            );
                        }
                    }
                }
            }
        } catch (Exception e) {
            this.logException(e);
        }
    }

    private void sendCommonStats(String prefix, CommonStats stats) {
        this.sendDocsStats(prefix + ".docs", stats.getDocs());
        this.sendStoreStats(prefix + ".store", stats.getStore());
        this.sendIndexingStats(prefix + ".indexing", stats.getIndexing());
        this.sendGetStats(prefix + ".get", stats.getGet());
        this.sendSearchStats(prefix + ".search", stats.getSearch());
        this.sendMergeStats(prefix + ".merges", stats.getMerge());
        this.sendRefreshStats(prefix + ".refresh", stats.getRefresh());
        this.sendFlushStats(prefix + ".flush", stats.getFlush());
        this.sendWarmerStats(prefix + ".warmer", stats.getWarmer());
        this.sendFielddataCacheStats(prefix + ".fielddata", stats.getFieldData());
        this.sendPercolateStats(prefix + ".percolate", stats.getPercolate());
        this.sendCompletionStats(prefix + ".completion", stats.getCompletion());
        this.sendSegmentsStats(prefix + ".segments", stats.getSegments());
        this.sendQueryCacheStats(prefix + ".query_cache", stats.getQueryCache());
        this.sendRequestCacheStats(prefix + ".request_cache", stats.getRequestCache());
        //TODO: getTranslog
        //TODO: getSuggest
    }
}