feat: add enable method

packages/core/src/logger.ts

@@ -24,7 +24,7 @@ export class MidwayContextLogger<T> {
   protected contextLogger: ILogger;
   public ctx: T;
 
-  constructor(ctx, contextLogger: ILogger, contextLabelHandler?) {
+  constructor(ctx, contextLogger: ILogger) {
     this.ctx = ctx;
     this.contextLogger = contextLogger;
   }
@@ -57,6 +57,7 @@ export class MidwayContextLogger<T> {
       ...args,
       {
         label: this.formatContextLabel(),
+        ctx: this.ctx,
       },
     ]);
   }

packages/logger/src/format.ts

@@ -20,6 +20,10 @@ export const displayCommonMessage = format(
       info.ignoreFormat = false;
     }
 
+    if (!info.ctx) {
+      info.ctx = null;
+    }
+
     if (!info.LEVEL) {
       info.LEVEL = info.level.toUpperCase();
     }
@@ -43,6 +47,7 @@ export const displayCommonMessage = format(
           LEVEL: info.LEVEL,
           defaultLabel: info.defaultLabel,
           ignoreFormat: info.ignoreFormat,
+          ctx: null,
         },
         opts.defaultMeta || opts.target?.getDefaultMeta() || {}
       );

packages/logger/src/interface.ts

@@ -14,6 +14,9 @@ export interface IMidwayLogger extends ILogger {
   enableFile();
   disableError();
   enableError();
+  isEnableFile(): boolean ;
+  isEnableConsole(): boolean;
+  isEnableError(): boolean ;
   updateLevel(level: LoggerLevel): void;
   updateFileLevel(level: LoggerLevel): void;
   updateConsoleLevel(level: LoggerLevel): void;
@@ -22,6 +25,9 @@ export interface IMidwayLogger extends ILogger {
   getDefaultLabel(): string;
   getDefaultMeta(): Record<string, unknown>;
   write(...args): boolean;
+  add(transport: any): any;
+  remove(transport: any): any;
+  close(): any;
 }
 
 export type LoggerLevel = 'silly' | 'debug' | 'info' | 'warn' | 'error';

packages/logger/src/logger.ts

@@ -163,6 +163,18 @@ export class MidwayBaseLogger extends EmptyLogger implements IMidwayLogger {
     this.add(this.errTransport);
   }
 
+  isEnableFile(): boolean {
+    return !!this.fileTransport;
+  }
+
+  isEnableConsole(): boolean {
+    return !!this.consoleTransport;
+  }
+
+  isEnableError(): boolean {
+    return !!this.errTransport;
+  }
+
   updateLevel(level: LoggerLevel): void {
     this.level = level;
     this.consoleTransport.level = level;
@@ -230,6 +242,18 @@ export class MidwayBaseLogger extends EmptyLogger implements IMidwayLogger {
       return super.write.apply(this, args);
     }
   }
+
+  add(transport): any {
+    return super.add(transport);
+  }
+
+  remove(transport: any): any {
+    return super.remove(transport);
+  }
+
+  close(): any {
+    return super.close();
+  }
 }
 
 /**

packages/logger/test/index.test.ts

@@ -7,7 +7,9 @@ import {
   IMidwayLogger,
   loggers,
   format,
-  displayCommonMessage
+  displayCommonMessage,
+  EmptyTransport,
+  ILogger
 } from '../src';
 import { join } from 'path';
 import {
@@ -21,8 +23,7 @@ import {
   getCurrentDateString
 } from './util';
 import { EggLogger } from 'egg-logger';
-import { readFileSync } from "fs";
-import { ILogger } from '../dist';
+import { readFileSync, writeFileSync } from 'fs';
 
 describe('/test/index.test.ts', () => {
   it('should test create logger', async () => {
@@ -528,4 +529,50 @@ describe('/test/index.test.ts', () => {
     await removeFileOrDir(logsDir);
   });
 
+  it('should custom transport', async () => {
+    clearAllLoggers();
+    const logsDir = join(__dirname, 'logs');
+    await removeFileOrDir(logsDir);
+
+    class CustomTransport extends EmptyTransport {
+      log(info, callback) {
+        const levelLowerCase = info.level;
+        if (levelLowerCase === 'error' || levelLowerCase === 'warn') {
+          writeFileSync(join(logsDir, 'test.log'), info.message);
+        }
+        callback();
+      }
+    }
+
+    const logger = createLogger<IMidwayLogger>('logger', {
+      dir: logsDir,
+      disableError: true,
+      level: 'info',
+    });
+
+    expect(logger.isEnableConsole()).toBeTruthy();
+    expect(logger.isEnableFile()).toBeTruthy();
+    expect(logger.isEnableError()).toBeFalsy();
+
+    const customTransport = new CustomTransport({
+      level: 'warn'
+    });
+    logger.add(customTransport);
+    logger.info('hello world info');
+    logger.warn('hello world warn');
+    logger.remove(customTransport);
+    logger.warn('hello world another warn');
+    await sleep();
+
+    expect(matchContentTimes(join(logsDir, 'midway-core.log'), 'hello world info')).toEqual(1);
+    expect(matchContentTimes(join(logsDir, 'midway-core.log'), 'hello world warn')).toEqual(1);
+    expect(matchContentTimes(join(logsDir, 'midway-core.log'), 'hello world another warn')).toEqual(1);
+
+    expect(matchContentTimes(join(logsDir, 'test.log'), 'hello world info')).toEqual(0);
+    expect(matchContentTimes(join(logsDir, 'test.log'), 'hello world warn')).toEqual(1);
+    expect(matchContentTimes(join(logsDir, 'test.log'), 'hello world another warn')).toEqual(0);
+
+    await removeFileOrDir(logsDir);
+  });
+
 });