当前位置：首页 > article >正文

express，MySQL 实现登录接口，如果用户未注册直接注册

article 2024/10/2 6:18:49

要实现一个登录接口，并在用户未注册时直接注册他们，您需要在您的 Express 应用中处理两个逻辑路径：一个是验证用户是否已存在（登录），另一个是在用户不存在时创建新用户（注册）。以下是一个简化的实现示例，它结合了登录和注册功能：

const express = require('express');
const mysql = require('mysql2/promise');
const bodyParser = require('body-parser');
const bcrypt = require('bcrypt'); // 用于密码哈希
const saltRounds = 10; // 密码哈希的盐轮数

const app = express();
const port = 3000;

// 配置 MySQL 连接池
const pool = mysql.createPool({
    host: 'localhost',
    user: 'root', // 根据实际情况修改
    password: 'your_password', // 根据实际情况修改
    database: 'login_db'
});

// 使用 body-parser 中间件解析 JSON 请求体
app.use(bodyParser.json());

// 登录/注册接口
app.post('/auth', async (req, res) => {
    const { username, password } = req.body;

    if (!username || !password) {
        return res.status(400).json({ message: 'Username and password are required' });
    }

    try {
        const connection = await pool.getConnection();

        // 检查用户是否存在
        const [rows] = await connection.execute('SELECT * FROM users WHERE username = ?', [username]);

        if (rows.length > 0) {
            // 用户已存在，验证密码
            const hashedPassword = rows[0].password;
            const isMatch = await bcrypt.compare(password, hashedPassword);

            if (isMatch) {
                return res.status(200).json({ message: 'Login successful', user: { username: rows[0].username } });
            } else {
                return res.status(401).json({ message: 'Invalid username or password' });
            }
        } else {
            // 用户不存在，注册新用户
            const hashedPassword = await bcrypt.hash(password, saltRounds);

            await connection.execute('INSERT INTO users (username, password) VALUES (?, ?)', [username, hashedPassword]);

            return res.status(201).json({ message: 'User registered and logged in successfully', user: { username } });
        }

        connection.release();
    } catch (err) {
        console.error(err);
        return res.status(500).json({ message: 'Internal server error' });
    }
});

app.listen(port, () => {
    console.log(`Server is running on http://localhost:${port}`);
});