Node.js项目中如何使用Koa2集成Swagger

栏目分类：前端开发    发布日期：2023-12-01    浏览次数：509次     收藏

这篇“Node.js项目中如何使用Koa2集成Swagger”文章的知识点大部分人都不太理解，所以小编给大家总结了以下内容，内容详细，步骤清晰，具有一定的借鉴价值，希望大家阅读完这篇文章能有所收获，下面我们一起来看看这篇“Node.js项目中如何使用Koa2集成Swagger”文章吧。

什么是Swagger

Swagger是一款RESTful API的文档生成工具，它可以帮助开发者快速、准确地编写、维护和查阅API文档。Swagger具有以下优点：

• 自动生成API文档，减少手动编写的工作量

• 提供可视化的API接口测试工具，方便调试和验证

• 支持多种语言和框架，具有良好的通用性和可扩展性

创建Koa2项目

首先，我们需要创建一个基于Koa2的Node.js项目。你可以使用以下命令创建项目：

mkdir koa2-swagger-demo
cd koa2-swagger-demo
npm init -y

然后，安装Koa2和相关依赖：

npm install koa koa-router --save

安装Swagger相关依赖

接下来，我们需要安装Swagger相关的NPM包。在本教程中，我们将使用

koa2-swagger-ui

和

swagger-jsdoc

。分别用于展示Swagger UI和生成API文档。

npm install koa2-swagger-ui swagger-jsdoc --save

配置Swagger

在项目根目录下，创建一个名为

swagger.js

的文件，用于配置Swagger。配置代码如下：

const swaggerJSDoc = require('swagger-jsdoc');
const options = {
    definition: {
        openapi: '3.0.0',
        info: {
            title: '我是标题',
            version: '1.0.0',
            description: '我是描述',
        },
        //servers的每一项，可以理解为一个服务,实际项目中，可自由修改
        servers: [
            {
                url: '/api',
                description: 'API server',
            },
        ],
    },
    apis: ['./routes/*.js'],
};

const swaggerSpec = swaggerJSDoc(options);

// 如果有Swagger规范文件转TS的需求，可在此处保留Swagger规范文件到本地，方便使用
//fs.writeFileSync('swagger.json', JSON.stringify(swaggerSpec, null, 2));

module.exports = swaggerSpec;

这里，我们定义了一个名为

options

的对象，包含了Swagger的基本信息和API接口的来源（即我们的路由文件）。然后，我们使用

swagger-jsdoc

生成API文档，并将其导出。

编写API接口

现在，我们来创建一个名为

routes

的文件夹，并在其中创建一个名为

users.js

的文件，用于定义用户相关的API接口。在users.js文件中，我们将编写以下代码：

const Router = require('koa-router');
const router = new Router();

/**
* @swagger
* tags:
*   name: Users
*   description: User management
*/

/**
* @swagger
* components:
*   schemas:
*     User:
*       type: object
*       properties:
*         id:
*           type: integer
*           description: The user ID.
*         name:
*           type: string
*           description: The user's name.
*         email:
*           type: string
*           description: The user's email.
*       required:
*         - id
*         - name
*         - email
*/

/**
* @swagger
* /users:
*   get:
*     summary: Retrieve a list of users
*     tags: [Users]
*     responses:
*       200:
*         description: A list of users.
*         content:
*           application/json:
*             schema:
*               type: array
*               items:
*                 $ref: '#/components/schemas/User'
*/
router.get('/users', async (ctx) => {
    const users = [
        { id: 1, name: 'John Doe', email: 'john.doe@example.com' },
        { id: 2, name: 'Jane Doe', email: 'jane.doe@example.com' },
    ];
    ctx.body = users;
});

module.exports = router;

注释简析：

1. tags

   ： 这部分定义了一个名为"Users"的标签。标签用于对API接口进行分类和分组。在这里，标签名为"Users"，描述为"users.js下的接口"。

   
   

   /**
    * @swagger
    * tags:
    *   name: Users
    *   description: users.js下的接口
    */
   

   
   

2. components

   和

   schemas

   ： 这部分定义了一个名为"User"的数据模型。数据模型描述了API接口中使用的数据结构。在这个例子中，"User"模型包含三个属性：

   id

   （整数类型，表示用户ID）、

   name

   （字符串类型，表示用户名）和

   email

   （字符串类型，表示用户电子邮件）。同时，

   id

   、

   name

   和

   email

   属性都被标记为必需。

   
   

   /**
    * @swagger
    * components:
    *   schemas:
    *     User:
    *       type: object
    *       properties:
    *         id:
    *           type: integer
    *           description: id.
    *         name:
    *           type: string
    *           description: name.
    *         email:
    *           type: string
    *           description: email.
    *       required:
    *         - id
    *         - name
    *         - email
    */
   

   
   

3. /users

   API接口： 这部分定义了一个获取用户列表的API接口。它描述了一个

   GET

   请求，路径为

   /users

   。这个接口使用了之前定义的"Users"标签。另外，它还定义了一个成功的响应，状态码为

   200

   ，表示返回一个用户列表。响应的内容类型为

   application/json

   ，其结构是一个包含"User"模型的数组。

   $ref: '#/components/schemas/User'

   是一个引用语法，引用了之前定义在

   components

   下的

   schemas

   中名为

   User

   的数据模型。

   
   

   /**
   * @swagger
   * /users:
   *   get:
   *     summary: 获取用户列表
   *     tags: [Users]
   *     responses:
   *       200:
   *         description: success.
   *         content:
   *           application/json:
   *             schema:
   *               type: array
   *               items:
   *                 $ref: '#/components/schemas/User'
   */
   

   
   

这段代码为API文档提供了有关用户管理接口、数据模型和响应格式的详细信息。Swagger JSDoc解析这些注释，并生成对应的OpenAPI文档。

生成API文档

接下来，我们需要在项目中启用Swagger UI。在项目根目录下，创建一个名为app.js的文件，然后编写以下代码：

const Koa = require('koa');
const Router = require('koa-router');
const swaggerUI = require('koa2-swagger-ui').koaSwagger;
const swaggerSpec = require('./swagger');
const usersRoutes = require('./routes/users');

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

router.use('/api', usersRoutes.routes(), usersRoutes.allowedMethods());

router.get(
    '/swagger',
    swaggerUI({
        routePrefix: false,
        swaggerOptions: {
            spec: swaggerSpec,
        },
    })
);

app.use(router.routes()).use(router.allowedMethods());

const PORT = process.env.PORT || 3000;
app.listen(PORT, () => {
    console.log(`Server is running at http://localhost:${PORT}`);
});

在这里，我们导入了koa2-swagger-ui和swagger-jsdoc生成的API文档。然后，我们定义了一个名为/swagger的路由，用于展示Swagger UI。最后，我们将用户相关的API接口挂载到/api路径下。

测试

node app.js

在浏览器中打开http://localhost:3000/swagger 你将看到Swagger UI及自动生成的API文档。