Merge pull request #29 from JoHuang/feat/add-close-method

Feat/add close method & fix confusing option
JaniAnttonen · Feb 13, 2020 · cedc7da · cedc7da
2 parents 024f508 + 5d23ccf
commit cedc7da
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/index.js b/index.js
@@ -22,11 +22,11 @@ class LokiTransport extends Transport {
       host: options.host,
       interval: options.interval,
       json: options.json,
-      batching: !options.batching,
+      batching: options.batching !== false,
       clearOnError: options.clearOnError,
       replaceOnError: options.replaceOnError,
       replaceTimestamp: options.replaceTimestamp,
-      gracefulShutdown: !options.gracefulShutdown
+      gracefulShutdown: options.gracefulShutdown !== false
     })
 
     this.useCustomFormat = options.format !== undefined
@@ -90,6 +90,15 @@ class LokiTransport extends Transport {
     // Trigger the optional callback
     callback()
   }
+
+  /**
+   * Send batch to loki when clean up
+   */
+  close () {
+    this.batcher.sendBatchToLoki()
+      .then(() => {}) // maybe should emit something here
+      .catch(() => {}) // maybe should emit something here
+  }
 }
 
 module.exports = LokiTransport