39 lines
888 B
JavaScript
39 lines
888 B
JavaScript
const jwt = require('jsonwebtoken');
|
|
|
|
// 生成JWT token
|
|
const generateToken = (userId, username) => {
|
|
return jwt.sign(
|
|
{ userId, username },
|
|
process.env.JWT_SECRET || 'your_jwt_secret_key_here',
|
|
{ expiresIn: '24h' }
|
|
);
|
|
};
|
|
|
|
// 验证JWT token中间件
|
|
const authenticateToken = (req, res, next) => {
|
|
const authHeader = req.headers['authorization'];
|
|
const token = authHeader && authHeader.split(' ')[1]; // Bearer TOKEN
|
|
|
|
if (!token) {
|
|
return res.status(401).json({
|
|
success: false,
|
|
message: '访问令牌缺失'
|
|
});
|
|
}
|
|
|
|
jwt.verify(token, process.env.JWT_SECRET || 'your_jwt_secret_key_here', (err, user) => {
|
|
if (err) {
|
|
return res.status(403).json({
|
|
success: false,
|
|
message: '访问令牌无效'
|
|
});
|
|
}
|
|
req.user = user;
|
|
next();
|
|
});
|
|
};
|
|
|
|
module.exports = {
|
|
generateToken,
|
|
authenticateToken
|
|
}; |