用户登录功能
本篇文章助力你实现简单的用户登录功能,并且提供验证码验证服务。
提醒一下各位同僚,工具类的出现是为了简化开发,提高我们的开发效率哈,但是还是提醒一下珍惜开发新功能的机会,无论功能简单与否,都是你锻炼成长的大好机会!
发车!
验证码类
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import java.io.Serializable;
/**
* 验证码类
*/
@Data
public class VerifyCode implements Serializable {
@ApiModelProperty("Base64值")
private String base64Val;
@ApiModelProperty("验证码值")
private String value;
}
验证码工具类
import org.apache.commons.lang3.RandomUtils;
import org.bouncycastle.util.encoders.Base64;
import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.Random;
/**
* 生成验证码工具类
*/
public final class VerifyCodeUtil {
//验证码类,用于最后返回此对象,包含验证码图片base64和真值
private static VerifyCode validate = null;
//随机类,用于生成随机参数
private static Random random = new Random();
//随机生成字符串的取值范围
private static String randString = "0123456789abcdefghijkmnpqrtyABCDEFGHIJLMNQRTY";
//图片宽度
private static int width = 97;
//图片高度
private static int height = 38;
//字符的数量
private static int StringNum = 4;
//干扰线数量
private static int lineSize = 40;
/**
* 获取随机字符,并返回字符的String格式
*
* @param index (指定位置)
* @return
*/
private static String getRandomChar(int index) {
return String.valueOf(randString.charAt(index));
}
/**
* 获取随机指定区间的随机数
*
* @param min (指定最小数)
* @param max (指定最大数)
* @return
*/
private static int getRandomNum(int min, int max) {
return RandomUtils.nextInt(min, max);
}
/**
* 获得字体 (名称、样式、磅值)
*
* @return
*/
private static Font getFont() {
return new Font("Fixedsys", Font.CENTER_BASELINE, 25);
}
/**
* 获得颜色
*
* @return
*/
private static Color getRandColor(int frontColor, int backColor) {
if (frontColor > 255) frontColor = 255;
if (backColor > 255) backColor = 255;
int red = frontColor + random.nextInt(backColor - frontColor - 16);
int green = frontColor + random.nextInt(backColor - frontColor - 14);
int blue = frontColor + random.nextInt(backColor - frontColor - 18);
return new Color(red, green, blue);
}
/**
* 绘制字符串,返回绘制的字符串
*
* @param g
* @param randomString
* @param i
* @return
*/
private static String drawString(Graphics g, String randomString, int i) {
Graphics2D g2d = (Graphics2D) g;
//设置字体
g2d.setFont(getFont());
//设置颜色
g2d.setColor(new Color(random.nextFloat(), random.nextFloat(), random.nextFloat()));
String randChar = String.valueOf(getRandomChar(random.nextInt(randString.length())));
//组装
randomString += randChar;
int rot = getRandomNum(5, 10);
g2d.translate(random.nextInt(3), random.nextInt(3));
g2d.rotate(rot * Math.PI / 180);
g2d.drawString(randChar, 13 * i, 20);
g2d.rotate(-rot * Math.PI / 180);
return randomString;
}
/**
* 绘制干扰线
*
* @param g
*/
private static void drawLine(Graphics g) {
//起点(x,y) 偏移量x1、y1
int x = random.nextInt(width);
int y = random.nextInt(height);
int xl = random.nextInt(13);
int yl = random.nextInt(15);
g.setColor(new Color(random.nextFloat(), random.nextFloat(), random.nextFloat()));
g.drawLine(x, y, x + xl, y + yl);
}
/**
* @return String 返回base64
* @MethodName: getRandomCode
* @Description: 生成Base64图片验证码
*/
public static VerifyCode GetRandomCode() {
validate = validate == null ? new VerifyCode() : validate;
// BufferedImage类是具有缓冲区的Image类,Image类是用于描述图像信息的类
BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_BGR);
// 获得BufferedImage对象的Graphics对象
Graphics g = image.getGraphics();
//填充矩形
g.fillRect(0, 0, width, height);
//设置字体
g.setFont(new Font("Times New Roman", Font.ROMAN_BASELINE, 18));
//设置颜色
g.setColor(getRandColor(110, 133));
//绘制干扰线
for (int i = 0; i <= lineSize; i++) {
drawLine(g);
}
//绘制字符
String randomString = "";
for (int i = 1; i <= StringNum; i++) {
randomString = drawString(g, randomString, i);
validate.setValue(randomString);
}
//释放绘图资源
g.dispose();
ByteArrayOutputStream bs = null;
try {
bs = new ByteArrayOutputStream();
ImageIO.write(image, "png", bs);
String imgsrc = Base64.toBase64String(bs.toByteArray());
validate.setBase64Val(imgsrc);
} catch (Exception e) {
e.printStackTrace();
} finally {
try {
bs.close();
} catch (IOException e) {
e.printStackTrace();
} finally {
bs = null;
}
}
return validate;
}
}
用户登录业务层
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import java.util.Objects;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
@Slf4j
@Service
@Transactional
public class SysUserInfoServiceImpl implements SysUserInfoService {
@Autowired
private RedisTemplate redisTemplate;
@Autowired
private SysUserInfoDao sysUserInfoDao;
@Override
public String createVerifyCode(String tranId) {
// 获取验证码
VerifyCode validateCode = VerifyCodeUtil.GetRandomCode();
log.info("生成验证码, tranId={}, validateCode={}", tranId, validateCode);
// 验证码值保存Redis, 有效期5分钟
redisTemplate.opsForValue().set(RedisConst.VERIFY_CODE_PREFIX + tranId, validateCode.getValue(), 5, TimeUnit.MINUTES);
// 返回Base64值
return validateCode.getBase64Val();
}
@Override
public LoginUserInfoVo login(SysUserLoginVo vo) {
log.info("用户登录, 入参vo={}", vo);
// 校验验证码
String codeRedisKey = RedisConst.VERIFY_CODE_PREFIX + vo.getTranId();
Object codeFromRedis = redisTemplate.opsForValue().get(codeRedisKey);
if (Objects.isNull(codeFromRedis)) {
log.info("验证码已失效");
throw new BusinessException("验证码错误");
} else {
if (!Objects.equals(vo.getVerifyCode().toLowerCase(), codeFromRedis.toString().toLowerCase())) {
log.info("验证码输入错误");
throw new BusinessException("验证码错误");
}
}
// 校验用户密码
LoginUserInfoVo loginUserInfo = sysUserInfoDao.selectLoginUserInfo(vo.getUserCode());
if (Objects.isNull(loginUserInfo)) {
log.info("userCode={}登录用户不存在", vo.getUserCode());
throw new BusinessException("用户名或密码错误");
} else {
String pwdFromDB = loginUserInfo.getPassword();
if (!Objects.equals(vo.getPassword(), pwdFromDB)) {
log.info("userCode={}登录用户密码输入错误", vo.getUserCode());
throw new BusinessException("用户名或密码错误");
}
loginUserInfo.setPassword("");
loginUserInfo.setToken(UUID.randomUUID().toString());
}
// 删除Redis验证码
redisTemplate.delete(codeRedisKey);
// 返回数据
return loginUserInfo;
}
@Override
public int updatePassword(UpdatePasswordVo vo) {
// 校验用户
LoginUserInfoVo loginUserInfo = sysUserInfoDao.selectLoginUserInfo(vo.getUserCode());
if (Objects.isNull(loginUserInfo)) {
log.info("userCode={}登录用户不存在", vo.getUserCode());
throw new BusinessException("用户名或旧密码错误");
} else {
if (!Objects.equals(vo.getOldPassword(), loginUserInfo.getPassword())) {
log.info("userCode={}登录用户旧密码错误", vo.getUserCode());
throw new BusinessException("用户名或旧密码错误");
}
}
// 修改密码
int rows = sysUserInfoDao.updatePassword(vo);
if (rows > 0) {
log.info("修改密码成功");
} else {
log.error("修改密码失败, userCode={}", vo.getUserCode());
}
return rows;
}
}
至此,用户登录功能的代码介绍完毕,还是希望大家能认真学习下实现的逻辑,踏下心来并不复杂哦~