67 lines
2.3 KiB
JavaScript
67 lines
2.3 KiB
JavaScript
|
// db.js —— 只做两件事：导出连接池；启动时建库/建表/插 admin
|
|||
|
|
require('dotenv').config() //自动寻找.env文件，并把里面的键值对全部导入进来
|
|||
|
|
const mysql = require('mysql2/promise')
|
|||
|
|
const bcrypt = require('bcryptjs') //用来给密码做哈希加密
|
|||
|
|
|
|||
|
|
// ============ 1. 连接池 ============
|
|||
|
|
// 业务代码里用 pool.query() / pool.execute()，会自动从池里取/还连接
|
|||
|
|
const pool = mysql.createPool({
|
|||
|
|
host: process.env.DB_HOST,
|
|||
|
|
port: Number(process.env.DB_PORT) || 3306,
|
|||
|
|
user: process.env.DB_USER,
|
|||
|
|
password: process.env.DB_PASSWORD,
|
|||
|
|
database: process.env.DB_NAME,
|
|||
|
|
waitForConnections: true, // 池里没连接时排队等
|
|||
|
|
connectionLimit: 10, // 最多 10 个连接
|
|||
|
|
decimalNumbers: true,
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
// ============ 2. 启动时初始化 ============
|
|||
|
|
// 思路：先用一个"无 database"的连接建库，再回到 pool 建表、插 admin
|
|||
|
|
async function initDB() {
|
|||
|
|
const { DB_HOST, DB_PORT, DB_USER, DB_PASSWORD, DB_NAME } = process.env
|
|||
|
|
|
|||
|
|
// 2.1 建库（如果不存在）
|
|||
|
|
const conn = await mysql.createConnection({
|
|||
|
|
host: DB_HOST,
|
|||
|
|
port: Number(DB_PORT) || 3306,
|
|||
|
|
user: DB_USER,
|
|||
|
|
password: DB_PASSWORD,
|
|||
|
|
})
|
|||
|
|
await conn.query(
|
|||
|
|
`CREATE DATABASE IF NOT EXISTS \`${DB_NAME}\` DEFAULT CHARACTER SET utf8mb4`
|
|||
|
|
)
|
|||
|
|
await conn.end()
|
|||
|
|
|
|||
|
|
// 2.2 建表
|
|||
|
|
await pool.query(`
|
|||
|
|
CREATE TABLE IF NOT EXISTS users (
|
|||
|
|
id INT PRIMARY KEY AUTO_INCREMENT,
|
|||
|
|
username VARCHAR(50) UNIQUE NOT NULL,
|
|||
|
|
password VARCHAR(255) NOT NULL,
|
|||
|
|
nickname VARCHAR(50) DEFAULT '',
|
|||
|
|
avatar VARCHAR(255) DEFAULT '',
|
|||
|
|
role VARCHAR(20) DEFAULT 'user',
|
|||
|
|
created_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
|||
|
|
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4
|
|||
|
|
`)
|
|||
|
|
|
|||
|
|
// 2.3 插 admin（仅当不存在时）
|
|||
|
|
const [rows] = await pool.query(
|
|||
|
|
'SELECT id FROM users WHERE username = ?',
|
|||
|
|
['admin']
|
|||
|
|
)
|
|||
|
|
if (rows.length === 0) {
|
|||
|
|
const hash = await bcrypt.hash('123456', 10)
|
|||
|
|
await pool.query(
|
|||
|
|
'INSERT INTO users (username, password, nickname, role) VALUES (?, ?, ?, ?)',
|
|||
|
|
['admin', hash, '超级管理员', 'admin']
|
|||
|
|
)
|
|||
|
|
console.log('[init] 已创建默认账号 admin / 123456')
|
|||
|
|
} else {
|
|||
|
|
console.log('[init] admin 已存在，跳过')
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
module.exports = { pool, initDB }
|