好的,以下是完整的代码,已按照约定规范修改了返回格式和字段命名:
const express = require('express');
const mysql = require('mysql2/promise');
const jwt = require('jsonwebtoken');
const app = express();
// 创建数据库连接池
const pool = mysql.createPool({
host: 'localhost',
user: 'root',
password: '123456',
database: 'test'
});
// 解析请求体中的 JSON 数据
app.use(express.json());
// 登录接口
app.post('/login', async (req, res) => {
const { username, password } = req.body;
try {
// 查询数据库中是否存在该用户
const [rows] = await pool.query('SELECT * FROM users WHERE username=?', [username]);
if (rows.length && rows[0].password === password) {
// 生成 JWT Token 并设置过期时间为 1 小时
const token = jwt.sign({ username }, 'secret', { expiresIn: '1h' });
// 返回成功信息以及 Token
res.status(200).json({ code: 200, message: '登录成功!', data: { token } });
} else {
res.status(401).json({ code: 401, message: '用户名或密码错误!', data: null });
}
} catch (err) {
console.error(err);
res.status(500).json({ code: 500, message: '服务器错误!', data: null });
}
});
// 获取用户信息接口(需要认证)
app.get('/user', (req, res) => {
// 获取请求头中的 Authorization 字段(JWT Token)
const authHeader = req.headers.authorization;
if (!authHeader) {
return res.status(401).json({ code: 401, message: '未授权访问!', data: null });
}
// 解析 JWT Token,获取用户名
const token = authHeader.split(' ')[1];
try {
const decoded = jwt.verify(token, 'secret');
const username = decoded.username;
// 查询数据库中该用户的信息并返回
pool.query('SELECT * FROM users WHERE username=?', [username])
.then(([rows]) => {
if (rows.length) {
res.status(200).json({ code: 200, message: '获取用户信息成功!', data: rows[0] });
} else {
res.status(404).json({ code: 404, message: '用户不存在!', data: null });
}
})
.catch((err) => {
console.error(err);
res.status(500).json({ code: 500, message: '服务器错误!', data: null });
});
} catch (err) {
console.error(err);
res.status(401).json({ code: 401, message: 'Token 不合法或已过期!', data: null });
}
});
// 启动服务器
app.listen(3000, () => console.log('Server started on port 3000'));
在上面的代码中,我们将返回信息改为符合约定规范的格式,并添加了 code
字段表示操作状态码。同时,我们也对返回消息进行了调整,以更加清晰地表达操作结果。