- 替换 dotenv 引入方式,使用 destructuring 导入 config 函数- 调用 config()函数以加载环境变量
-移除冗余的 import('dotenv').config() 语句
51 lines
1.2 KiB
JavaScript
51 lines
1.2 KiB
JavaScript
import mongoose from 'mongoose'
|
|
import logger from 'morgan'
|
|
|
|
import { config } from 'dotenv'
|
|
|
|
config()
|
|
|
|
// 使用环境变量存储敏感信息
|
|
const account = process.env.MONGO_ACCOUNT
|
|
const password = process.env.MONGO_PASSWORD
|
|
const host = process.env.MONGO_HOST || '122.152.201.90'
|
|
const port = process.env.MONGO_PORT || '27017'
|
|
|
|
let isConnected = false
|
|
|
|
export async function connectMongoDB() {
|
|
if (process.env.NODE_ENV === 'development') {
|
|
logger({ account, password, host, port })
|
|
}
|
|
|
|
// 如果已经连接,不再重复连接
|
|
if (isConnected) {
|
|
logger('MongoDB 已经连接')
|
|
return
|
|
}
|
|
|
|
try {
|
|
const mongoDBUrl = `mongodb://${account}:${password}@${host}:${port}`
|
|
|
|
await mongoose.connect(mongoDBUrl)
|
|
|
|
mongoose.connection.on('open', () => {
|
|
logger('MongoDB 连接成功')
|
|
isConnected = true
|
|
})
|
|
|
|
mongoose.connection.on('error', (err) => {
|
|
logger('MongoDB 连接错误:', err.message)
|
|
isConnected = false
|
|
})
|
|
|
|
mongoose.connection.on('close', () => {
|
|
logger('MongoDB 连接关闭')
|
|
isConnected = false
|
|
})
|
|
} catch (err) {
|
|
logger('MongoDB 连接初始化失败:', err.message)
|
|
throw err
|
|
}
|
|
}
|