log component name

shocknet · May 7, 2024 · da9d9e6 · da9d9e6
1 parent 4f40c60
commit da9d9e6
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 4 deletions.
diff --git a/src/services/helpers/logger.ts b/src/services/helpers/logger.ts
@@ -78,6 +78,12 @@ export const getLogger = (params: LoggerParams): PubLogger => {
             }
             toLog.push(params.appName)
         }
+        if (params.component) {
+            if (disabledComponents.includes(params.component)) {
+                return
+            }
+            toLog.push(params.component)
+        }
         if (params.userId) {
             toLog.push(params.userId)
         }
@@ -87,7 +93,13 @@ export const getLogger = (params: LoggerParams): PubLogger => {
         writers.forEach(w => w(final))
     }
 }
-const disabledApps: string[] = []
-export const disableLoggers = (appNamesToDisable: string[]) => {
+let disabledApps: string[] = []
+let disabledComponents: string[] = []
+export const resetDisabledLoggers = () => {
+    disabledApps = []
+    disabledComponents = []
+}
+export const disableLoggers = (appNamesToDisable: string[], componentsToDisable: string[]) => {
     disabledApps.push(...appNamesToDisable)
+    disabledComponents.push(...componentsToDisable)
 }
diff --git a/src/tests/spamExternalPayments.spec.ts b/src/tests/spamExternalPayments.spec.ts
@@ -4,7 +4,7 @@ import { Describe, expect, expectThrowsAsync, runSanityCheck, safelySetUserBalan
 export const ignore = false
 
 export default async (T: TestBase) => {
-    disableLoggers(["EventsLogManager", "htlcTracker", "watchdog"])
+    disableLoggers([], ["EventsLogManager", "htlcTracker", "watchdog"])
     await safelySetUserBalance(T, T.user1, 2000)
     await testSpamExternalPayment(T)
     await runSanityCheck(T)

diff --git a/src/tests/spamMixedPayments.spec.ts b/src/tests/spamMixedPayments.spec.ts
@@ -5,7 +5,7 @@ import * as Types from '../../proto/autogenerated/ts/types.js'
 export const ignore = false
 
 export default async (T: TestBase) => {
-    disableLoggers(["EventsLogManager", "htlcTracker", "watchdog"])
+    disableLoggers([], ["EventsLogManager", "htlcTracker", "watchdog"])
     await safelySetUserBalance(T, T.user1, 2000)
     await testSpamExternalPayment(T)
     await runSanityCheck(T)

diff --git a/src/tests/testBase.ts b/src/tests/testBase.ts
@@ -9,6 +9,7 @@ import chaiString from 'chai-string'
 import { defaultInvoiceExpiry } from '../services/storage/paymentStorage.js'
 import SanityChecker from '../services/main/sanityChecker.js'
 import LND from '../services/lnd/lnd.js'
+import { resetDisabledLoggers } from '../services/helpers/logger.js'
 chai.use(chaiString)
 export const expect = chai.expect
 export type Describe = (message: string, failure?: boolean) => void
@@ -69,6 +70,7 @@ export const teardown = async (T: TestBase) => {
     T.externalAccessToMainLnd.Stop()
     T.externalAccessToOtherLnd.Stop()
     T.externalAccessToThirdLnd.Stop()
+    resetDisabledLoggers()
     console.log("teardown")
 }