Merge pull request #5 from intellifylearning/optimize

support blacklisting
camitz · Sep 17, 2016 · 5354f30 · 5354f30
2 parents 0613935 + c6e74fa
commit 5354f30
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/lib/aws-cloudwatch-statsd-backend.js b/lib/aws-cloudwatch-statsd-backend.js
@@ -47,6 +47,16 @@ CloudwatchBackend.prototype.processKey = function(key) {
   };
 };
 
+CloudwatchBackend.prototype.isBlacklisted = function(key) {
+
+  if (this.config.blacklist && this.config.blacklist.length > 0 && this.config.blacklist.indexOf(key) !== -1) {
+    // console.log("Key (counter) " + key + " is blacklisted");
+    return true;
+  }
+
+  return false;
+};
+
 CloudwatchBackend.prototype.chunk = function(arr, chunkSize) {
 
   var groups = [],
@@ -103,6 +113,10 @@ CloudwatchBackend.prototype.flush = function(timestamp, metrics) {
       continue;
     }
 
+    if (this.isBlacklisted(key)) {
+      continue;
+    }
+
     var names = this.config.processKeyForNamespace ? this.processKey(key) : {};
     namespace = this.config.namespace || names.namespace || "AwsCloudWatchStatsdBackend";
     var metricName = this.config.metricName || names.metricName || key;
@@ -127,6 +141,10 @@ CloudwatchBackend.prototype.flush = function(timestamp, metrics) {
         continue;
       }
 
+      if (this.isBlacklisted(key)) {
+        continue;
+      }
+
       var values = timers[key].sort(function(a, b) {
         return a - b;
       });
@@ -179,6 +197,10 @@ CloudwatchBackend.prototype.flush = function(timestamp, metrics) {
       continue;
     }
 
+    if (this.isBlacklisted(key)) {
+      continue;
+    }
+
     var names = this.config.processKeyForNamespace ? this.processKey(key) : {};
     namespace = this.config.namespace || names.namespace || "AwsCloudWatchStatsdBackend";
     var metricName = this.config.metricName || names.metricName || key;
@@ -202,6 +224,10 @@ CloudwatchBackend.prototype.flush = function(timestamp, metrics) {
       continue;
     }
 
+    if (this.isBlacklisted(key)) {
+      continue;
+    }
+
     var names = this.config.processKeyForNamespace ? this.processKey(key) : {};
     namespace = this.config.namespace || names.namespace || "AwsCloudWatchStatsdBackend";
     var metricName = this.config.metricName || names.metricName || key;