feat:更换原来的logger实现

Author: pedro

lib/extend.ts

@@ -7,19 +7,16 @@ import {
   FileExtensionException,
   FileTooManyException
 } from './exception';
-import consola from 'consola';
 import { toLine, unsets } from './util';
 import { config } from './config';
 import { get, set, cloneDeep } from 'lodash';
 import parse from 'co-busboy';
 import sendToWormhole from 'stream-wormhole';
 import { extname } from 'path';
 
-import { Logger, FileTransport, ConsoleTransport } from 'egg-logger';
-
-// const Logger = require('egg-logger').Logger;
-// const FileTransport = require('egg-logger').FileTransport;
-// const ConsoleTransport = require('egg-logger').ConsoleTransport;
+import { Logger } from 'egg-logger';
+import { FileTransport } from './logger/file';
+import { ConsoleTransport } from './logger/console';
 
 /**
  * json序列化扩展
@@ -96,22 +93,32 @@ export const success = (app: Application) => {
  * @param app app实例
  */
 export const logging = (app: Application) => {
-  // TODO: 提供配置项
-  // const logger = new Logger();
-  // logger.set(
-  //   'file',
-  //   new FileTransport({
-  //     file: '/path/to/file',
-  //     level: 'INFO'
-  //   })
-  // );
-  // logger.set(
-  //   'console',
-  //   new ConsoleTransport({
-  //     level: 'DEBUG'
-  //   })
-  // );
-  app.context.logger = consola;
+  let options = {
+    level: 'INFO',
+    dir: 'logs',
+    sizeLimit: 1024 * 1024 * 5
+  };
+  const logConf = config.getItem('log');
+  options = { ...options, ...logConf };
+
+  const logger = new Logger({});
+
+  logger.set(
+    'file',
+    new FileTransport({
+      dir: options.dir,
+      sizeLimit: options.sizeLimit,
+      level: options.level
+    })
+  );
+  logger.set(
+    'console',
+    new ConsoleTransport({
+      level: options.level
+    })
+  );
+
+  app.context.logger = logger;
 };
 
 export interface MulOpts {

lib/middleware.ts

@@ -1,5 +1,7 @@
 import { HttpException, NotFound, MethodNotAllowed } from './exception';
 import { Context } from 'koa';
+const levels = require('egg-logger/lib/level');
+import { config } from './config';
 
 /**
  * 全局异常处理中间件
@@ -32,10 +34,25 @@ export const log = async (ctx: Context, next: () => Promise<any>) => {
     await next();
     const ms = Date.now() - start;
     ctx.set('X-Response-Time', `${ms}ms`);
-    ctx.logger.info(
-      `[${ctx.method}] -> [${ctx.url}] from: ${ctx.ip} costs: ${ms}ms`
-    );
-
+    const requestLog: boolean = config.getItem('log.requestLog');
+    const level: string = config.getItem('log.level');
+    if (requestLog) {
+      if (levels[level] >= levels['DEBUG']) {
+        const data = {
+          param: ctx.request.query,
+          body: ctx.request.body
+        };
+        ctx.logger.debug(
+          `[${ctx.method}] -> [${ctx.url}] from: ${
+            ctx.ip
+          } costs: ${ms}ms data:${JSON.stringify(data, null, 4)}`
+        );
+      } else {
+        ctx.logger.info(
+          `[${ctx.method}] -> [${ctx.url}] from: ${ctx.ip} costs: ${ms}ms`
+        );
+      }
+    }
     if (ctx.status === 404) {
       ctx.app.emit('error', new NotFound(), ctx);
     } else if (ctx.status === 405) {