28 lines
523 B
JavaScript
28 lines
523 B
JavaScript
import mongoose from "mongoose";
|
|
|
|
const MONGODB_URI = process.env.MONGODB_URI;
|
|
|
|
if (!MONGODB_URI) {
|
|
throw new Error("请在环境变量中设置 MONGODB_URI");
|
|
}
|
|
|
|
let cached = global.mongoose;
|
|
|
|
if (!cached) {
|
|
cached = global.mongoose = { conn: null, promise: null };
|
|
}
|
|
|
|
async function dbConnect() {
|
|
if (cached.conn) {
|
|
return cached.conn;
|
|
}
|
|
|
|
if (!cached.promise) {
|
|
cached.promise = mongoose.connect(MONGODB_URI);
|
|
}
|
|
cached.conn = await cached.promise;
|
|
return cached.conn;
|
|
}
|
|
|
|
export default dbConnect;
|