中级攻略：Koa2中的中间件的妙用

2024-02-04 06:19:17

中级攻略：Koa2中的中间件的妙用

中间件概述

中间件是 Koa2 中非常重要的一部分，它允许您在请求和响应之间拦截并处理请求。通过使用中间件，您可以实现各种各样的功能，例如身份验证、日志记录、错误处理等。

中间件的使用

在 Koa2 中，可以使用 app.use() 方法来注册中间件。例如，以下代码注册了一个简单的中间件，该中间件会在每次请求时打印请求的URL：

const Koa = require('koa');

const app = new Koa();

app.use(async (ctx, next) => {
  console.log(ctx.url);
  await next();
});

app.listen(3000);

您还可以使用多个中间件，它们将按照注册顺序依次执行。例如，以下代码注册了两个中间件：第一个中间件打印请求的URL，第二个中间件打印请求的HTTP方法：

const Koa = require('koa');

const app = new Koa();

app.use(async (ctx, next) => {
  console.log(ctx.url);
  await next();
});

app.use(async (ctx, next) => {
  console.log(ctx.method);
  await next();
});

app.listen(3000);

中间件的常见用途

路由

中间件的一个常见用途是路由请求。通过使用中间件，您可以将不同的请求路由到不同的处理程序。例如，以下代码使用中间件来将请求路由到不同的控制器：

const Koa = require('koa');
const Router = require('@koa/router');

const app = new Koa();
const router = new Router();

router.get('/users', async (ctx, next) => {
  // 处理用户请求
});

router.post('/users', async (ctx, next) => {
  // 处理创建用户请求
});

app.use(router.routes());

app.listen(3000);

身份验证

中间件的另一个常见用途是身份验证。通过使用中间件，您可以验证请求是否来自授权用户。例如，以下代码使用中间件来验证请求是否包含有效的JWT令牌：

const Koa = require('koa');
const jwt = require('jsonwebtoken');

const app = new Koa();

app.use(async (ctx, next) => {
  const token = ctx.request.headers['authorization'];

  if (!token) {
    ctx.status = 401;
    return;
  }

  try {
    const decoded = jwt.verify(token, 'secret');

    // 将用户ID添加到ctx.state对象中
    ctx.state.user = decoded.id;
  } catch (err) {
    ctx.status = 401;
    return;
  }

  await next();
});

app.listen(3000);

日志记录

中间件的另一个常见用途是日志记录。通过使用中间件，您可以将请求信息记录到日志文件中。例如，以下代码使用中间件来记录请求的URL、HTTP方法和状态码：

const Koa = require('koa');
const fs = require('fs');

const app = new Koa();

app.use(async (ctx, next) => {
  const log = `${ctx.url} ${ctx.method} ${ctx.status}\n`;

  fs.appendFile('access.log', log, (err) => {
    if (err) {
      console.error(err);
    }
  });

  await next();
});

app.listen(3000);

错误处理

中间件的另一个常见用途是错误处理。通过使用中间件，您可以捕获错误并将其发送给错误处理程序。例如，以下代码使用中间件来捕获错误并将其发送给错误处理程序：

const Koa = require('koa');

const app = new Koa();

app.use(async (ctx, next) => {
  try {
    await next();
  } catch (err) {
    ctx.status = 500;
    ctx.body = 'Internal Server Error';
  }
});

app.listen(3000);