add datadog

1 files changed, 9 insertions, 1 deletions

diff --git a/worker.js b/worker.js
index d90f1d4..7cd795e 100644
--- a/worker.js
+++ b/worker.js
@@ -6,6 +6,7 @@ const amqp = require("amqplib"),
     Promise = require("bluebird"),
     winston = require("winston"),
     loggly = require("winston-loggly-bulk"),
+    datadog = require("winston-datadog"),
     request = require("request-promise"),
     api = require("../orm/api");
 
@@ -13,7 +14,8 @@ const MADGLORY_TOKEN = process.env.MADGLORY_TOKEN,
     QUEUE = process.env.QUEUE || "grab",
     PROCESS_QUEUE = process.env.PROCESS_QUEUE || "process",
     RABBITMQ_URI = process.env.RABBITMQ_URI || "amqp://localhost",
-    LOGGLY_TOKEN = process.env.LOGGLY_TOKEN;
+    LOGGLY_TOKEN = process.env.LOGGLY_TOKEN,
+    DATADOG_TOKEN = process.env.DATADOG_TOKEN;
 if (MADGLORY_TOKEN == undefined) throw "Need an API token";
 
 const logger = new (winston.Logger)({
@@ -34,6 +36,12 @@ if (LOGGLY_TOKEN)
         json: true
     });
 
+// datadog integration
+if (DATADOG_TOKEN)
+    logger.add(new datadog({
+        api_key: DATADOG_TOKEN
+    }), null, true);
+
 amqp.connect(RABBITMQ_URI).then(async (rabbit) => {
     process.once("SIGINT", rabbit.close.bind(rabbit));