ZBLOG

请将这段修改了格式和字段命名规范的代码,补充到完整的代码中

好的,以下是完整的代码,已按照约定规范修改了返回格式和字段命名:

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 字段表示操作状态码。同时,我们也对返回消息进行了调整,以更加清晰地表达操作结果。

本站部分文章来源于网络,版权归原作者所有,如有侵权请联系站长删除。
转载请注明出处:https://golang.0voice.com/?id=4963

分享:
扫描分享到社交APP
上一篇
下一篇
发表列表
游客 游客
此处应有掌声~
评论列表

还没有评论,快来说点什么吧~

联系我们

在线咨询: 点击这里给我发消息

微信号:3007537140

上班时间: 10:30-22:30

关注我们
x

注册

已经有帐号?