在Linux上使用Swagger實現API限流,通常需要以下幾個步驟:
首先,確保你已經安裝了Swagger。你可以使用npm來安裝Swagger:
npm install -g swagger-jsdoc swagger-ui-express
創建一個Swagger配置文件,例如swagger.js
,并定義你的API規范:
const swaggerJsDoc = require('swagger-jsdoc');
const swaggerOptions = {
swaggerDefinition: {
openapi: '3.0.0',
info: {
title: 'API Documentation',
version: '1.0.0',
},
},
apis: ['./routes/*.js'], // 指定你的API路由文件
};
const swaggerDocs = swaggerJsDoc(swaggerOptions);
module.exports = swaggerDocs;
創建一個Express應用,并將Swagger UI集成到你的應用中:
const express = require('express');
const swaggerUi = require('swagger-ui-express');
const swaggerDocs = require('./swagger');
const app = express();
app.use('/api-docs', swaggerUi.serve, swaggerUi.setup(swaggerDocs));
// 你的API路由
const routes = require('./routes');
app.use('/api', routes);
const PORT = process.env.PORT || 3000;
app.listen(PORT, () => {
console.log(`Server is running on port ${PORT}`);
});
你可以使用express-rate-limit
中間件來實現API限流。首先,安裝express-rate-limit
:
npm install express-rate-limit
然后,在你的Express應用中配置限流中間件:
const rateLimit = require('express-rate-limit');
const apiLimiter = rateLimit({
windowMs: 15 * 60 * 1000, // 15分鐘
max: 100, // 每個IP地址在15分鐘內最多請求100次
standardHeaders: true, // 返回RateLimit-* headers
legacyHeaders: false, // 禁用RateLimit-* headers
handler: (req, res) => {
res.status(429).json({
message: 'Too many requests, please try again later.',
});
},
});
// 應用限流中間件到所有API路由
app.use('/api/', apiLimiter);
啟動你的Express應用并測試限流功能。你可以使用工具如curl
或Postman來發送請求,觀察是否在達到限制后返回429狀態碼。
curl -X GET http://localhost:3000/api/your-endpoint
通過以上步驟,你可以在Linux上使用Swagger實現API限流。首先安裝Swagger并配置Swagger文檔,然后創建Express應用并集成Swagger UI,最后使用express-rate-limit
中間件來實現API限流。這樣,你就可以有效地控制API的請求速率,保護你的后端服務免受過度請求的影響。