| New file |
| | |
|---|
| | | package com.dy.pipIrrApp.captcha; |
|---|
| | | |
|---|
| | | import com.dy.pipIrrApp.captcha.dto.CaptchaDomain; |
|---|
| | | import com.dy.pipIrrGlobal.daoBa.BaCaptchaMapper; |
|---|
| | | import com.dy.pipIrrGlobal.pojoBa.BaCaptcha; |
|---|
| | | import com.google.code.kaptcha.Producer; |
|---|
| | | import jakarta.annotation.Resource; |
|---|
| | | import lombok.extern.slf4j.Slf4j; |
|---|
| | | import org.springframework.beans.factory.annotation.Autowired; |
|---|
| | | import org.springframework.stereotype.Service; |
|---|
| | | |
|---|
| | | import javax.imageio.ImageIO; |
|---|
| | | import java.io.ByteArrayOutputStream; |
|---|
| | | import java.util.Base64; |
|---|
| | | import java.util.UUID; |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * @author ZhuBaoMin |
|---|
| | | * @date 2024-09-11 13:46 |
|---|
| | | * @LastEditTime 2024-09-11 13:46 |
|---|
| | | * @Description |
|---|
| | | */ |
|---|
| | | |
|---|
| | | @Slf4j |
|---|
| | | @Service |
|---|
| | | public class CaptchaSV { |
|---|
| | | @Autowired |
|---|
| | | private BaCaptchaMapper baCaptchaMapper; |
|---|
| | | |
|---|
| | | @Resource(name = "captchaProducer") |
|---|
| | | private Producer captchaProducer; |
|---|
| | | |
|---|
| | | @Resource(name = "captchaProducerMathOne") |
|---|
| | | private Producer captchaProducerMathOne; |
|---|
| | | |
|---|
| | | @Resource(name = "captchaProducerMathTwo") |
|---|
| | | private Producer captchaProducerMathTwo; |
|---|
| | | |
|---|
| | | private static final String TYPE_CHAR = "char"; |
|---|
| | | private static final String TYPE_MATH_ONE = "math"; |
|---|
| | | private static final String TYPE_MATH_TWO = "math2"; |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * Kaptcha生成验证码实体 |
|---|
| | | * @param type，char - 字符(缺省) | math - 一位数算式 | math2 - 两位数算式 |
|---|
| | | * @return |
|---|
| | | */ |
|---|
| | | public CaptchaDomain createGoogleCaptcha(String type) { |
|---|
| | | // 定义验证码实体 |
|---|
| | | CaptchaDomain captchaDomain = new CaptchaDomain(); |
|---|
| | | // 一位数加减乘除 |
|---|
| | | if (TYPE_MATH_ONE.equals(type)) { |
|---|
| | | // 生成文本 |
|---|
| | | String producerText = captchaProducerMathOne.createText(); |
|---|
| | | // 设置验证码字符 |
|---|
| | | captchaDomain.setText(producerText.substring(0, producerText.indexOf("@"))); |
|---|
| | | // 设置验证码答案码 |
|---|
| | | captchaDomain.setCode(producerText.substring(producerText.indexOf("@") + 1)); |
|---|
| | | // 设置验证码图片 |
|---|
| | | captchaDomain.setImage(captchaProducerMathOne.createImage(captchaDomain.getText())); |
|---|
| | | } |
|---|
| | | // 两位数加减乘除 |
|---|
| | | else if (TYPE_MATH_TWO.equals(type)) { |
|---|
| | | String producerText = captchaProducerMathTwo.createText(); |
|---|
| | | captchaDomain.setText(producerText.substring(0, producerText.indexOf("@"))); |
|---|
| | | captchaDomain.setCode(producerText.substring(producerText.indexOf("@") + 1)); |
|---|
| | | captchaDomain.setImage(captchaProducerMathTwo.createImage(captchaDomain.getText())); |
|---|
| | | } |
|---|
| | | // 缺省情况：字符 |
|---|
| | | else { |
|---|
| | | captchaDomain.setText(captchaProducer.createText()); |
|---|
| | | captchaDomain.setCode(captchaDomain.getText()); |
|---|
| | | captchaDomain.setImage(captchaProducer.createImage(captchaDomain.getText())); |
|---|
| | | } |
|---|
| | | // 生成base64 |
|---|
| | | try { |
|---|
| | | // 定义字节数组输出流 |
|---|
| | | ByteArrayOutputStream outputStream = new ByteArrayOutputStream(); |
|---|
| | | // 将图像以 jpg 的形式，写到字节数组输出流中 |
|---|
| | | ImageIO.write(captchaDomain.getImage(), "jpg", outputStream); |
|---|
| | | |
|---|
| | | // 写入base64格式 |
|---|
| | | captchaDomain.setBase64("data:image/jpg;base64," + Base64.getEncoder().encodeToString(outputStream.toByteArray())); |
|---|
| | | // 写入唯一Token |
|---|
| | | captchaDomain.setToken(UUID.randomUUID().toString()); |
|---|
| | | // 返回结果 |
|---|
| | | return captchaDomain; |
|---|
| | | } catch (Exception e) { |
|---|
| | | System.out.println(e.getMessage()); |
|---|
| | | return null; |
|---|
| | | } |
|---|
| | | } |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * 添加图片验证记录 |
|---|
| | | * @param po |
|---|
| | | * @return |
|---|
| | | */ |
|---|
| | | public Long addCaptcha(BaCaptcha po) { |
|---|
| | | baCaptchaMapper.insert(po); |
|---|
| | | return po.getId(); |
|---|
| | | } |
|---|
| | | |
|---|
| | | } |
|---|
