25 lines
825 B
JavaScript
25 lines
825 B
JavaScript
|
import mongoose from 'mongoose'
|
||
|
|
import User from './model/user.js'
|
||
|
|
|
||
|
|
const connectDB = async () => {
|
||
|
|
try {
|
||
|
|
const conn = await mongoose.connect(process.env.MONGO_URI)
|
||
|
|
console.log(`MongoDB Connected: ${conn.connection.host}`)
|
||
|
|
// 检查是否存在管理员用户，如果不存在则创建一个默认管理员
|
||
|
|
const adminUser = await User.findOne({ username: process.env.ADMIN_USERNAME })
|
||
|
|
if (!adminUser) {
|
||
|
|
await User.create({
|
||
|
|
username: process.env.ADMIN_USERNAME,
|
||
|
|
nickname: '管理员',
|
||
|
|
password: process.env.ADMIN_PASSWORD,
|
||
|
|
role: 'admin',
|
||
|
|
})
|
||
|
|
}
|
||
|
|
} catch (error) {
|
||
|
|
console.error('Error connecting to MongoDB:', error)
|
||
|
|
process.exit(1)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
export default connectDB
|