Add enabled property to Log plugin

thundra-io · Oct 1, 2019 · 13048d4 · 13048d4
1 parent 86b7f2c
commit 13048d4
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -65,8 +65,9 @@ module.exports = (options: any) => {
             const logPlugin = new Log(config.logConfig);
             Logger.getLogManager().addListener(logPlugin);
         }
-
-        config.plugins.push(Log.getInstance());
+        const logInstance = Log.getInstance();
+        logInstance.enable();
+        config.plugins.push(logInstance);
     }
 
     if (!(Utils.getConfiguration(envVariableKeys.THUNDRA_DISABLE_XRAY) === 'true') && config.xrayConfig.enabled) {

diff --git a/src/plugins/Log.ts b/src/plugins/Log.ts
@@ -14,6 +14,7 @@ class Log {
 
     options: LogConfig;
     reporter: any;
+    enabled: boolean;
     pluginContext: PluginContext;
     apiKey: any;
     logData: LogData;
@@ -33,7 +34,7 @@ class Log {
         if (options) {
             this.tracer = options.tracer;
         }
-
+        this.enabled = false;
         Log.instance = this;
         this.config = options;
     }
@@ -96,6 +97,9 @@ class Log {
     }
 
     reportLog(logInfo: any): void {
+        if (!this.enabled) {
+            return;
+        }
         const logData = new LogData();
         const activeSpan = this.tracer ? this.tracer.getActiveSpan() : undefined;
         const spanId = activeSpan ? activeSpan.spanContext.spanId : '';
@@ -143,6 +147,14 @@ class Log {
         });
     }
 
+    enable(): void {
+        this.enabled = true;
+    }
+
+    disable(): void {
+        this.enabled = false;
+    }
+
     destroy(): void {
         if (Utils.getConfiguration(envVariableKeys.THUNDRA_LAMBDA_LOG_CONSOLE_DISABLE) !== 'true') {
             this.unShimConsole();