如何用Next.js中间件实现Mongoose自动连接MongoDB？是否更优？

一、实现自动连接的高阶函数中间件 #

你可以通过高阶函数封装连接逻辑，让所有需要操作数据库的API路由自动执行连接，无需手动调用connect()。具体步骤如下：

1. 扩展连接工具文件 #

在你的src/utils/connect.js中添加一个高阶函数，用于包裹API路由的handler：

// src/utils/connect.js
import mongoose from 'mongoose'

const MONGO_URI = process.env.MONGO_URI

if (!MONGO_URI) {
    throw new Error(
        'Please define the MONGO_URI environment variable inside .env.local'
    )
}

let cached = global.mongoose
if (!cached) {
    cached = global.mongoose = null
}

export default async function connect() {
    if (cached) {
        return cached
    }
    cached = await mongoose.connect(MONGO_URI, {
        bufferCommands: false,
        useNewUrlParser: true,
        useUnifiedTopology: true
    })
    return cached
}

// 新增：自动连接的高阶函数中间件
export function withMongooseConnect(handler) {
    return async (req, res) => {
        try {
            // 先执行数据库连接
            await connect()
            // 再调用原API handler
            return handler(req, res)
        } catch (error) {
            console.error('MongoDB连接失败:', error)
            return res.status(500).json({ error: '数据库连接失败' })
        }
    }
}

2. 修改API路由使用中间件 #

在需要操作数据库的API路由中，用withMongooseConnect包裹你的handler即可，无需手动调用connect()：

// src/pages/api/test.js
import { withMongooseConnect } from '../../utils/connect'
import Test from '../../models/Test';

async function handler(req, res) {
    // 直接操作数据库，无需手动连接
    const data = await Test.find({});
    res.status(200).json({ name: 'John Doe', data })
}

// 用中间件包裹handler
export default withMongooseConnect(handler);

二、方案合理性评估 #

这种方案是非常合适的，理由如下：

优点 #

1. 消除重复代码：避免在每个数据库相关的API路由中重复写await connect()，符合DRY原则
2. 缓存机制兼容：你的connect()已经实现了连接缓存，多次调用只会复用已有的连接，不会导致连接数爆炸，性能不受影响
3. 集中管理逻辑：连接配置、错误处理都集中在connect.js中，后续修改连接参数或添加全局连接逻辑（比如日志）只需修改这一个文件
4. 灵活性高：可以选择只给需要数据库操作的路由包裹中间件，不需要的路由完全不受影响

注意事项 #

• 确保所有需要操作数据库的API路由都使用该中间件包裹，否则会因未连接数据库而报错
• 错误处理：中间件中的try-catch可以统一捕获连接失败的情况，避免单个路由重复写错误处理逻辑
• 如果你使用Next.js 13+的App Router，也可以将这个逻辑封装成server component或route handler的包装函数，原理和Pages Router的高阶函数一致

内容的提问来源于stack exchange，提问作者Rawand