重新封装代码

2026-09-09 22:33:09 +08:00
parent 51c41e8fb7
commit 52ceba6e6d
100 changed files with 9828 additions and 89 deletions
@@ -0,0 +1,75 @@
+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;