Merge pull request #576 from mattrjacobs/add-collapser-metrics

Add collapser metrics

hystrix-contrib/hystrix-codahale-metrics-publisher/src/main/java/com/netflix/hystrix/contrib/codahalemetricspublisher/HystrixCodaHaleMetricsPublisherCollapser.java

@@ -0,0 +1,201 @@
+/**
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.netflix.hystrix.contrib.codahalemetricspublisher;
+
+import com.codahale.metrics.Gauge;
+import com.codahale.metrics.MetricRegistry;
+import com.netflix.hystrix.HystrixCollapserKey;
+import com.netflix.hystrix.HystrixCollapserMetrics;
+import com.netflix.hystrix.HystrixCollapserProperties;
+import com.netflix.hystrix.strategy.metrics.HystrixMetricsPublisherCollapser;
+import com.netflix.hystrix.util.HystrixRollingNumberEvent;
+
+/**
+ * Implementation of {@link HystrixMetricsPublisherCollapser} using Coda Hale Metrics (https://github.com/codahale/metrics)
+ */
+public class HystrixCodaHaleMetricsPublisherCollapser implements HystrixMetricsPublisherCollapser {
+    private final HystrixCollapserKey key;
+    private final HystrixCollapserMetrics metrics;
+    private final HystrixCollapserProperties properties;
+    private final MetricRegistry metricRegistry;
+    private final String metricType;
+
+    public HystrixCodaHaleMetricsPublisherCollapser(HystrixCollapserKey collapserKey, HystrixCollapserMetrics metrics, HystrixCollapserProperties properties, MetricRegistry metricRegistry) {
+        this.key = collapserKey;
+        this.metrics = metrics;
+        this.properties = properties;
+        this.metricRegistry = metricRegistry;
+        this.metricType = key.name();
+    }
+
+    @Override
+    public void initialize() {
+        // allow monitor to know exactly at what point in time these stats are for so they can be plotted accurately
+        metricRegistry.register(createMetricName("currentTime"), new Gauge<Long>() {
+            @Override
+            public Long getValue() {
+                return System.currentTimeMillis();
+            }
+        });
+
+        // cumulative counts
+        createCumulativeCountForEvent("countRequestsBatched", HystrixRollingNumberEvent.COLLAPSER_REQUEST_BATCHED);
+        createCumulativeCountForEvent("countBatches", HystrixRollingNumberEvent.COLLAPSER_BATCH);
+        createCumulativeCountForEvent("countResponsesFromCache", HystrixRollingNumberEvent.RESPONSE_FROM_CACHE);
+
+        // rolling counts
+        createRollingCountForEvent("rollingRequestsBatched", HystrixRollingNumberEvent.COLLAPSER_REQUEST_BATCHED);
+        createRollingCountForEvent("rollingBatches", HystrixRollingNumberEvent.COLLAPSER_BATCH);
+        createRollingCountForEvent("rollingCountResponsesFromCache", HystrixRollingNumberEvent.RESPONSE_FROM_CACHE);
+
+        // batch size metrics
+        metricRegistry.register(createMetricName("batchSize_mean"), new Gauge<Integer>() {
+            @Override
+            public Integer getValue() {
+                return metrics.getBatchSizeMean();
+            }
+        });
+        metricRegistry.register(createMetricName("batchSize_percentile_25"), new Gauge<Integer>() {
+            @Override
+            public Integer getValue() {
+                return metrics.getBatchSizePercentile(25);
+            }
+        });
+        metricRegistry.register(createMetricName("batchSize_percentile_50"), new Gauge<Integer>() {
+            @Override
+            public Integer getValue() {
+                return metrics.getBatchSizePercentile(50);
+            }
+        });
+        metricRegistry.register(createMetricName("batchSize_percentile_75"), new Gauge<Integer>() {
+            @Override
+            public Integer getValue() {
+                return metrics.getBatchSizePercentile(75);
+            }
+        });
+        metricRegistry.register(createMetricName("batchSize_percentile_90"), new Gauge<Integer>() {
+            @Override
+            public Integer getValue() {
+                return metrics.getBatchSizePercentile(90);
+            }
+        });
+        metricRegistry.register(createMetricName("batchSize_percentile_99"), new Gauge<Integer>() {
+            @Override
+            public Integer getValue() {
+                return metrics.getBatchSizePercentile(99);
+            }
+        });
+        metricRegistry.register(createMetricName("batchSize_percentile_995"), new Gauge<Integer>() {
+            @Override
+            public Integer getValue() {
+                return metrics.getBatchSizePercentile(99.5);
+            }
+        });
+
+        // shard size metrics
+        metricRegistry.register(createMetricName("shardSize_mean"), new Gauge<Integer>() {
+            @Override
+            public Integer getValue() {
+                return metrics.getShardSizeMean();
+            }
+        });
+        metricRegistry.register(createMetricName("shardSize_percentile_25"), new Gauge<Integer>() {
+            @Override
+            public Integer getValue() {
+                return metrics.getShardSizePercentile(25);
+            }
+        });
+        metricRegistry.register(createMetricName("shardSize_percentile_50"), new Gauge<Integer>() {
+            @Override
+            public Integer getValue() {
+                return metrics.getShardSizePercentile(50);
+            }
+        });
+        metricRegistry.register(createMetricName("shardSize_percentile_75"), new Gauge<Integer>() {
+            @Override
+            public Integer getValue() {
+                return metrics.getShardSizePercentile(75);
+            }
+        });
+        metricRegistry.register(createMetricName("shardSize_percentile_90"), new Gauge<Integer>() {
+            @Override
+            public Integer getValue() {
+                return metrics.getShardSizePercentile(90);
+            }
+        });
+        metricRegistry.register(createMetricName("shardSize_percentile_99"), new Gauge<Integer>() {
+            @Override
+            public Integer getValue() {
+                return metrics.getShardSizePercentile(99);
+            }
+        });
+        metricRegistry.register(createMetricName("shardSize_percentile_995"), new Gauge<Integer>() {
+            @Override
+            public Integer getValue() {
+                return metrics.getShardSizePercentile(99.5);
+            }
+        });
+
+        // properties (so the values can be inspected and monitored)
+        metricRegistry.register(createMetricName("propertyValue_rollingStatisticalWindowInMilliseconds"), new Gauge<Number>() {
+            @Override
+            public Number getValue() {
+                return properties.metricsRollingStatisticalWindowInMilliseconds().get();
+            }
+        });
+
+        metricRegistry.register(createMetricName("propertyValue_requestCacheEnabled"), new Gauge<Boolean>() {
+            @Override
+            public Boolean getValue() {
+                return properties.requestCacheEnabled().get();
+            }
+        });
+
+        metricRegistry.register(createMetricName("propertyValue_maxRequestsInBatch"), new Gauge<Number>() {
+            @Override
+            public Number getValue() {
+                return properties.maxRequestsInBatch().get();
+            }
+        });
+
+        metricRegistry.register(createMetricName("propertyValue_timerDelayInMilliseconds"), new Gauge<Number>() {
+            @Override
+            public Number getValue() {
+                return properties.timerDelayInMilliseconds().get();
+            }
+        });
+    }
+
+    protected String createMetricName(String name) {
+        return MetricRegistry.name("", metricType, name);
+    }
+
+    protected void createCumulativeCountForEvent(String name, final HystrixRollingNumberEvent event) {
+        metricRegistry.register(createMetricName(name), new Gauge<Long>() {
+            @Override
+            public Long getValue() {
+                return metrics.getCumulativeCount(event);
+            }
+        });
+    }
+
+    protected void createRollingCountForEvent(String name, final HystrixRollingNumberEvent event) {
+        metricRegistry.register(createMetricName(name), new Gauge<Long>() {
+            @Override
+            public Long getValue() {
+                return metrics.getRollingCount(event);
+            }
+        });
+    }
+}