75 lines
3.4 KiB
JavaScript
75 lines
3.4 KiB
JavaScript
const express = require("express");
|
|
const router = express.Router();
|
|
const db = require("../db/mysql");
|
|
|
|
const jwt = require("../utils/jwt");
|
|
|
|
router.post("/login", (req, res) => {
|
|
const { username, password } = req.body;
|
|
|
|
const sql = `
|
|
SELECT * FROM chat_mysql.users
|
|
WHERE username = ? AND password = ?;
|
|
`;
|
|
db.query(sql, [username, password], (error, results) => {
|
|
if (error) {
|
|
console.error("数据库查询失败：", error);
|
|
|
|
return res.status(500).json({
|
|
code: 500,
|
|
message: "登录失败",
|
|
});
|
|
}
|
|
if (results.length > 0) {
|
|
// 生成 JWT token
|
|
const token = jwt.createToken({
|
|
userId: results[0].user_id,
|
|
username: results[0].username,
|
|
});
|
|
res.json({
|
|
code: 200,
|
|
message: "登录成功",
|
|
data: {
|
|
token: token,
|
|
username: results[0].username,
|
|
},
|
|
});
|
|
} else {
|
|
res.json({
|
|
code: 401,
|
|
message: "用户名或密码错误",
|
|
data: null,
|
|
});
|
|
}
|
|
});
|
|
});
|
|
|
|
router.post("/register", (req, res) => {
|
|
const { username, password } = req.body;
|
|
|
|
const sql = `
|
|
INSERT INTO users (username, password)
|
|
VALUES (?, ?)
|
|
`;
|
|
|
|
db.query(sql, [username, password], (error, result) => {
|
|
if (error) {
|
|
console.error(error);
|
|
|
|
return res.status(500).json({
|
|
code: 500,
|
|
message: "注册失败",
|
|
});
|
|
}
|
|
|
|
res.json({
|
|
code: 200,
|
|
message: "注册成功",
|
|
data: {
|
|
id: result.insertId,
|
|
},
|
|
});
|
|
});
|
|
});
|
|
|
|
module.exports = router; |