NestJS如何使用异步中间件?

时间:2019-06-20 20:23:01

标签: async-await middleware nestjs

我使用NestJS框架,并希望将多个中间件应用于我的应用程序中的路由。每个中间件都是一个实现NestMiddleware接口的类。这些中间件之一是异步的,在调用路由处理程序之前不会被使用。在处理路由之前,有没有办法解决这个中间件的承诺?

我的代码

异步中间件(page-loader.middleware)

import { Injectable, NestMiddleware } from '@nestjs/common';

@Injectable()

export class PageLoader implements NestMiddleware {

  async use(req: any, res: any, next: () => void) {
    try {
      req.body.html = await req.body.fetcher.fetch();
    } catch (error) {
      throw new Error(error);
    } finally {
      next();
    }
  }

}

控制器(create-article.controller)

import { Controller, Post, Body } from '@nestjs/common';
import { SaveArticleService } from './save-article.service';
import { CreateArticleDto } from './create-article.dto';

@Controller()

export class CreateArticleController {

  constructor(private readonly saveArticleService: SaveArticleService) {}

  @Post('/create')
  async create(@Body() createArticleDto: CreateArticleDto) {
    return this.saveArticleService.save(createArticleDto);
  }

}

模块(create-article.module)

import { Module, NestModule, MiddlewareConsumer, RequestMethod } from '@nestjs/common';
import { CreateArticleController } from './create-article.controller';
import { SaveArticleService } from './save-article.service';

// Another (sync) middleware
import { ExtensionExtractor } from './extension-extractor.middleware'; 

// The async middleware
import { PageLoader } from './page-loader.middleware';

@Module({
  controllers: [CreateArticleController],
  providers: [SaveArticleService],
})

export class CreateArticleModule implements NestModule {
  configure(consumer: MiddlewareConsumer) {
    consumer
    .apply(ExtensionExtractor, PageLoader)
    .forRoutes({ path: 'create', method: RequestMethod.POST});
  }
}

我没有包含控制器中使用的服务的摘要,因为它与我的问题无关。

我尝试过的

这个question并没有帮助我解决问题,因为中间件结构不同。我正在等待中间件方法来解决其诺言,而不是在等待中间件内部重用的输入。

由于NestJS API发生了巨大变化,因此这些github issue的答案无关紧要。

在此先感谢您的帮助!

1 个答案:

答案 0 :(得分:1)

您绝对可以将异步中间件与Nest结合使用;但是,使用.forRoutes({path: 'path', method: method});策略存在问题。

我建立了一个快速的中间件使用者,以演示它如何在不使用RequestMethod.GET的情况下工作。

UserModule(包括中间件)

import { MiddlewareConsumer, Module, NestModule, RequestMethod } from '@nestjs/common';
import { UserController } from './user.controller';
import { UserService } from './user.service';

function asyncTimeout(milliseconds: number): Promise<string> {
  return new Promise((resolve, reject) => {
    setTimeout(() => resolve('DONE'), milliseconds);
  });
}

@Module({
  controllers: [UserController],
  providers: [UserService],
  exports: [UserService]
})
export class UserModule implements NestModule {
  configure(consumer: MiddlewareConsumer) {
    consumer
      .apply((req, res, next) => {
        console.log('Using forRoutes(path)');
        console.log('syncronous middleware');
        next();
      },
        (async (req, res, next) => {
          console.log('Using forRoutes(path)');
          const start = Date.now();
          const done = await asyncTimeout(5000);
          console.log(done);
          console.log('Time taken:' + (Date.now() - start));
          next();
        })
      )
      .forRoutes('/')
      .apply((req, res, next) => {
        console.log('Using forRoutes({path, method})');
        console.log('syncronous middleware');
        next();
      },
        (async (req, res, next) => {
          console.log('Using forRoutes({path, method})');
          const start = Date.now();
          const done = await asyncTimeout(5000);
          console.log(done);
          console.log('Time taken:' + (Date.now() - start));
          next();
        })
      )
      .forRoutes({path: '/', method: RequestMethod.GET});
  }
}


UserController(代码段)

import { Controller, Get } from '@nestjs/common';
import { UserSerivce } from './user.service';

@Controller('user')
export class UserController {
  constructor(private readonly userService: UserService) {}

  @Get('/')
  testFunction() {
    return {greeting: 'hello'};
  }
}

输出

[2019-06-20 22:40:48.191] [INFO] | Listening at http://localhost:3333/api
Using forRoutes(path)
syncronous middleware
Using forRoutes(path)
DONE
Time taken:5002
[2019-06-20 22:40:57.346] [INFO] | [Nest] 30511 [Morgan] GET /api/user 200 5014.234 ms - 20

我在两种设置中都使用了相同的中间件功能,但是您可以看到,异步中间件在使用.forRoutes(path)时会按预期方式响应,而在使用.forRoutes({path, method})时则没有响应(请原谅我的自定义记录器)。

这应该在GitHub上作为Kamil的一个问题来解决,但可以用其他方法进行设置。如果您选择打开此代码,请随意在此处使用我的任何代码来为该问题设置示例存储库。