/*
|
* Copyright (c) 2018-2999 广州市蓝海创新科技有限公司 All rights reserved.
|
*
|
* https://www.mall4j.com/
|
*
|
* 未经允许，不可做商业用途！
|
*
|
* 版权所有，侵权必究！
|
*/
|
package com.yami.shop.api.controller;
|
|
|
import cn.hutool.core.util.BooleanUtil;
|
import cn.hutool.core.util.IdUtil;
|
import cn.hutool.core.util.StrUtil;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.github.binarywang.utils.qrcode.MatrixToImageWriter;
|
import com.google.common.collect.Maps;
|
import com.google.zxing.BarcodeFormat;
|
import com.google.zxing.EncodeHintType;
|
import com.google.zxing.MultiFormatWriter;
|
import com.google.zxing.WriterException;
|
import com.google.zxing.common.BitMatrix;
|
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
|
import com.obs.services.ObsClient;
|
import com.yami.shop.bean.app.param.CheckRegisterSmsParam;
|
import com.yami.shop.bean.app.param.SendSmsParam;
|
import com.yami.shop.bean.app.param.UserPwdUpdateParam;
|
import com.yami.shop.bean.app.param.UserRegisterParam;
|
import com.yami.shop.bean.enums.SendType;
|
import com.yami.shop.bean.model.CdnTeamRelation;
|
import com.yami.shop.bean.model.MarsHandCard;
|
import com.yami.shop.bean.model.User;
|
import com.yami.shop.bean.param.im.ImConfigParam;
|
import com.yami.shop.common.bean.HuaWeiOss;
|
import com.yami.shop.common.exception.YamiShopBindException;
|
import com.yami.shop.common.response.ServerResponseEntity;
|
import com.yami.shop.common.util.PrincipalUtil;
|
import com.yami.shop.common.util.RedisUtil;
|
import com.yami.shop.security.common.enums.SysTypeEnum;
|
import com.yami.shop.security.common.manager.PasswordManager;
|
import com.yami.shop.security.common.manager.TokenStore;
|
import com.yami.shop.security.common.service.AppConnectService;
|
import com.yami.shop.security.common.service.ImTokenService;
|
import com.yami.shop.security.common.vo.TokenInfoVO;
|
import com.yami.shop.service.*;
|
import com.yami.shop.task.common.service.TaskInviteLogService;
|
import com.yami.shop.task.common.service.TaskUserRelationService;
|
import io.swagger.v3.oas.annotations.Operation;
|
import io.swagger.v3.oas.annotations.tags.Tag;
|
import lombok.AllArgsConstructor;
|
import ma.glasnost.orika.MapperFacade;
|
import org.springframework.security.crypto.password.PasswordEncoder;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.web.bind.annotation.*;
|
|
import javax.validation.Valid;
|
import java.io.*;
|
import java.nio.file.Files;
|
import java.nio.file.Path;
|
import java.nio.file.Paths;
|
import java.time.LocalDate;
|
import java.util.*;
|
import java.util.concurrent.*;
|
import java.util.regex.Matcher;
|
import java.util.regex.Pattern;
|
|
/**
|
* 用户信息
|
*
|
* @author LGH
|
*/
|
@RestController
|
@RequestMapping("/user")
|
@Tag(name = "用户注册相关接口")
|
@AllArgsConstructor
|
public class UserRegisterController {
|
|
private final UserService userService;
|
private final MapperFacade mapperFacade;
|
|
private final SmsLogService smsLogService;
|
|
private final AppConnectService appConnectService;
|
|
private final TokenStore tokenStore;
|
|
private final PasswordEncoder passwordEncoder;
|
|
private final PasswordManager passwordManager;
|
|
private final TaskInviteLogService taskInviteLogService;
|
|
private final TaskUserRelationService taskUserRelationService;
|
|
private final MarsHandCardService marsHandCardService;
|
private final SysConfigService sysConfigService;
|
|
|
private final CdnTeamRelationService cdnTeamRelationService;
|
|
private final ImTokenService imTokenService;
|
|
public static final String CHECK_REGISTER_SMS_FLAG = "checkRegisterSmsFlag";
|
|
public static final String CHECK_UPDATE_PWD_SMS_FLAG = "updatePwdSmsFlag";
|
|
|
/**
|
* 注册的头像昵称有几个值得注意的地方：
|
* 1. 如果是微信公众号 or 小程序注册，在注册之前，也就是在进入页面之前，需要调用SocialLoginController 这里面的尝试登录的接口，如果可以登录就直接登录了。
|
* 2. 关于发送验证码
|
* 1) 手机号注册,要发送验证码
|
* 3. 当注册成功的时候，已经返回token，相对来说，已经登录了
|
*/
|
@Operation(summary = "注册")
|
@PostMapping("/register")
|
@Transactional(rollbackFor = Exception.class)
|
// public ServerResponseEntity<TokenInfoVO> register(@Valid @RequestBody UserRegisterParam userRegisterParam) {
|
public ServerResponseEntity<TokenInfoVO> register(@Valid @RequestBody UserRegisterParam userRegisterParam) {
|
|
// 目前注册都是发验证码去注册的，看看有没有校验验证码成功的标识
|
// userService.validate(userRegisterParam, CHECK_REGISTER_SMS_FLAG + userRegisterParam.getCheckRegisterSmsFlag());
|
// 正在进行申请注册
|
if (userService.count(new LambdaQueryWrapper<User>().eq(User::getUserMobile, userRegisterParam.getMobile())) > 0) {
|
// 该手机号已注册，无法重新注册
|
throw new YamiShopBindException("yami.user.phone.exist");
|
}
|
if (userRegisterParam.getMobile().length() != 11) {
|
// 该手机号已注册，无法重新注册
|
throw new YamiShopBindException("手机号格式不对，应为11位！");
|
}
|
String inviteCode = userRegisterParam.getInviteCode();
|
//绑定推荐人关系
|
if (!Objects.equals(inviteCode, "") && inviteCode != null) {
|
// 根据邀请码查询推荐人信息
|
User u = userService.getOne(new LambdaQueryWrapper<User>().eq(User::getInviteCode, inviteCode));
|
if (u != null) {
|
// 注册并绑定用户
|
User user = appConnectService.registerAndBindUserNick(userRegisterParam.getMobile(), userRegisterParam.getPassword(), null, userRegisterParam.getNickName());
|
ImConfigParam imConfig = sysConfigService.getSysConfigObject("IM_CONFIG", ImConfigParam.class);
|
imTokenService.getImToken(imConfig,user,2);
|
|
CdnTeamRelation teamRelation = cdnTeamRelationService.getOne(new LambdaQueryWrapper<CdnTeamRelation>().eq(CdnTeamRelation::getUserId, u.getUserId()));
|
CdnTeamRelation cdnTeamRelation = new CdnTeamRelation();
|
cdnTeamRelation.setUserId(user.getUserId());
|
cdnTeamRelation.setLevel(0); //初始等级为0
|
cdnTeamRelation.setAgentLevelId(1);// 初始用户代理等级为会员
|
cdnTeamRelation.setPid(u.getUserId()); // 推荐人id
|
cdnTeamRelation.setJoinTime(new Date());
|
cdnTeamRelation.setCreateTime(new Date());
|
if (teamRelation != null) { //不为空时推荐人合计拼接推荐人的推荐人合计，为空时推荐人合计只有推荐人
|
if (teamRelation.getPids() != null && !teamRelation.getPids().equals("")) {
|
cdnTeamRelation.setPids(teamRelation.getPids() + (u.getUserId() + ",")); // 推荐人合计id
|
cdnTeamRelation.setSort(teamRelation.getSort() + 1);
|
} else {
|
cdnTeamRelation.setPid(u.getUserId()); // 推荐人id
|
cdnTeamRelation.setPids("," + u.getUserId() + ","); // 推荐人合计i
|
cdnTeamRelation.setSort(teamRelation.getSort() + 1);
|
}
|
} else {
|
cdnTeamRelation.setPid(u.getUserId()); // 推荐人id
|
cdnTeamRelation.setPids("," + u.getUserId() + ","); // 推荐人合计i
|
cdnTeamRelation.setSort(1);
|
}
|
cdnTeamRelationService.save(cdnTeamRelation);
|
} else {
|
return ServerResponseEntity.showFailMsg("该邀请码无效!");
|
}
|
} else {
|
return ServerResponseEntity.showFailMsg("请输入邀请码!");
|
}
|
return ServerResponseEntity.success();
|
}
|
|
@GetMapping("/getUserInviteCode")
|
@Operation(summary = "获取用户邀请码", description = "获取用户邀请码")
|
public ServerResponseEntity<Boolean> getUserInviteCode(@RequestParam String mobile) {
|
User user = userService.getOne(new LambdaQueryWrapper<User>().eq(User::getUserMobile, mobile));
|
if (user != null) {
|
return ServerResponseEntity.success(true);
|
}
|
return ServerResponseEntity.success(false);
|
}
|
|
|
@PutMapping("/sendRegisterSms")
|
@Operation(summary = "发送注册验证码", description = "发送注册验证码")
|
public ServerResponseEntity<Void> register(@Valid @RequestBody SendSmsParam sendSmsParam) {
|
if (userService.count(new LambdaQueryWrapper<User>().eq(User::getUserMobile, sendSmsParam.getMobile())) > 0) {
|
// 该手机号已注册，无法重新注册
|
throw new YamiShopBindException("yami.user.phone.exist");
|
}
|
// 每个手机号每分钟只能发十个注册的验证码，免得接口被利用
|
smsLogService.sendSms(SendType.REGISTER, sendSmsParam.getMobile(), sendSmsParam.getMobile(), Maps.newHashMap());
|
return ServerResponseEntity.success();
|
}
|
|
@PutMapping("/checkRegisterSms")
|
@Operation(summary = "校验验证码", description = "校验验证码返回校验成功的标识")
|
public ServerResponseEntity<String> register(@Valid @RequestBody CheckRegisterSmsParam checkRegisterSmsParam) {
|
// 每个ip每分钟只能发十个注册的验证码，免得接口被利用
|
if (!smsLogService.checkValidCode(checkRegisterSmsParam.getMobile(), checkRegisterSmsParam.getValidCode(), SendType.REGISTER)) {
|
// 验证码有误或已过期
|
throw new YamiShopBindException("yami.user.code.error");
|
}
|
String checkRegisterSmsFlag = IdUtil.simpleUUID();
|
RedisUtil.set(CHECK_REGISTER_SMS_FLAG + checkRegisterSmsFlag, checkRegisterSmsParam.getMobile(), 600);
|
return ServerResponseEntity.success(checkRegisterSmsFlag);
|
}
|
|
@PutMapping("/sendBindSms")
|
@Operation(summary = "发送绑定验证码", description = "发送绑定验证码")
|
public ServerResponseEntity<Void> bindSms(@Valid @RequestBody SendSmsParam sendSmsParam) {
|
// 每个手机号每分钟只能发十个注册的验证码，免得接口被利用
|
smsLogService.sendSms(SendType.VALID, sendSmsParam.getMobile(), sendSmsParam.getMobile(), Maps.newHashMap());
|
return ServerResponseEntity.success();
|
}
|
|
|
@PutMapping("/checkUpdatePwdSms")
|
@Operation(summary = "修改密码校验验证码", description = "校验验证码返回校验成功的标识")
|
public ServerResponseEntity<String> checkUpdatePwdSms(@RequestBody CheckRegisterSmsParam checkRegisterSmsParam) {
|
boolean isCheckPass = false;
|
if (Objects.nonNull(checkRegisterSmsParam) && Objects.nonNull(checkRegisterSmsParam.getMobile())) {
|
Matcher m = Pattern.compile(PrincipalUtil.MOBILE_REGEXP).matcher(checkRegisterSmsParam.getMobile());
|
isCheckPass = m.matches();
|
}
|
if (!isCheckPass) {
|
throw new YamiShopBindException("yami.user.err.phone");
|
}
|
|
User user = userService.getOne(new LambdaQueryWrapper<User>().eq(User::getUserMobile, checkRegisterSmsParam.getMobile()));
|
if (user == null) {
|
// 此用户不存在，请先注册
|
throw new YamiShopBindException("yami.user.account.no.exist");
|
}
|
String defaultCode = sysConfigService.getValue("UPDATE_DEFAULT_CODE");
|
// 验证码登录
|
if(!defaultCode.equals(checkRegisterSmsParam.getValidCode())) {
|
if (!smsLogService.checkValidCode(user.getUserMobile(), checkRegisterSmsParam.getValidCode(), SendType.UPDATE_PASSWORD)) {
|
// 验证码有误或已过期
|
throw new YamiShopBindException("yami.user.code.error");
|
}
|
}
|
String checkRegisterSmsFlag = IdUtil.simpleUUID();
|
RedisUtil.set(CHECK_UPDATE_PWD_SMS_FLAG + checkRegisterSmsFlag, checkRegisterSmsParam.getMobile(), 600);
|
return ServerResponseEntity.success(checkRegisterSmsFlag);
|
}
|
|
@PutMapping("/updatePwd")
|
@Operation(summary = "修改密码", description = "修改密码")
|
public ServerResponseEntity<Void> updatePwd(@Valid @RequestBody UserPwdUpdateParam userPwdUpdateParam) {
|
User user = userService.getOne(new LambdaQueryWrapper<User>().eq(User::getUserMobile, userPwdUpdateParam.getMobile()));
|
if (user == null) {
|
// 无法获取用户信息
|
throw new YamiShopBindException("yami.user.no.exist");
|
}
|
UserRegisterParam registerParam = mapperFacade.map(userPwdUpdateParam, UserRegisterParam.class);
|
// 看看有没有校验验证码成功的标识
|
userService.validate(registerParam, CHECK_UPDATE_PWD_SMS_FLAG + userPwdUpdateParam.getCheckRegisterSmsFlag());
|
String decryptPassword = passwordManager.decryptPassword(userPwdUpdateParam.getPassword());
|
if (StrUtil.isBlank(decryptPassword)) {
|
// 新密码不能为空
|
throw new YamiShopBindException("yami.user.password.no.exist");
|
}
|
if (StrUtil.equals(passwordEncoder.encode(decryptPassword), user.getLoginPassword())) {
|
// 新密码不能与原密码相同!
|
throw new YamiShopBindException("yami.user.password.check");
|
}
|
user.setModifyTime(new Date());
|
user.setLoginPassword(passwordEncoder.encode(decryptPassword));
|
userService.updateById(user);
|
tokenStore.deleteAllToken(SysTypeEnum.ORDINARY.value().toString(), user.getUserId());
|
return ServerResponseEntity.success();
|
}
|
|
|
@PostMapping("/test")
|
public ServerResponseEntity updateCardsWithThreadPool() {
|
// 创建线程池
|
ExecutorService executorService = new ThreadPoolExecutor(
|
4, // 核心线程数
|
8, // 最大线程数
|
60L, TimeUnit.SECONDS, // 空闲线程存活时间
|
new ArrayBlockingQueue<>(100), // 阻塞队列容量
|
new ThreadPoolExecutor.CallerRunsPolicy() // 拒绝策略
|
);
|
|
// 分页查询数据
|
Page<MarsHandCard> page = new Page<>(1, 500); // 每页500条记录
|
List<MarsHandCard> marsHandCardList;
|
|
do {
|
// 获取当前页的数据
|
IPage<MarsHandCard> currentPage = marsHandCardService.page(page, Wrappers.lambdaQuery(MarsHandCard.class));
|
marsHandCardList = currentPage.getRecords();
|
|
// 并发处理数据
|
List<Future<?>> futures = new ArrayList<>();
|
for (MarsHandCard marsHandCard : marsHandCardList) {
|
Future<?> future = executorService.submit(() -> {
|
String QRCodeName = marsHandCard.getCardNumber() + ".png";
|
LocalDate now = LocalDate.now();
|
String ossImagePath = "handCard/" + now.getYear() + "/" + now.getMonthValue() + "/" + QRCodeName;
|
|
try {
|
uploadQRCode(marsHandCard.getCardNumber(), 300, 300);
|
String s = "https://hxsl.obs.cn-north-4.myhuaweicloud.com/" + ossImagePath;
|
marsHandCardService.update(Wrappers.lambdaUpdate(MarsHandCard.class)
|
.eq(MarsHandCard::getId, marsHandCard.getId())
|
.set(MarsHandCard::getQrCode, s));
|
} catch (Exception e) {
|
throw new RuntimeException(e);
|
}
|
});
|
futures.add(future);
|
}
|
|
// 等待所有任务完成
|
CompletableFuture.allOf(futures.stream().map(CompletableFuture::completedFuture).toArray(CompletableFuture[]::new)).join();
|
|
// 准备下一页数据
|
page.setCurrent(page.getCurrent() + 1);
|
} while (!marsHandCardList.isEmpty());
|
|
// 关闭线程池
|
executorService.shutdown();
|
|
return ServerResponseEntity.success();
|
}
|
|
public static String uploadQRCode(String content, int width, int height) throws Exception {
|
String format = "png";
|
// 定义二维码的参数
|
HashMap<EncodeHintType, Object> hints = new HashMap<>();
|
// 定义字符集编码格式
|
hints.put(EncodeHintType.CHARACTER_SET, "utf-8");
|
// 纠错的等级 L > M > Q > H 纠错的能力越高可存储的越少，一般使用M
|
hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.M);
|
// 设置图片边距
|
hints.put(EncodeHintType.MARGIN, 2);
|
|
try {
|
// 最终生成 参数列表 （1.内容 2.格式 3.宽度 4.高度 5.二维码参数）
|
BitMatrix bitMatrix = new MultiFormatWriter().encode(content, BarcodeFormat.QR_CODE, width, height, hints);
|
LocalDate now = LocalDate.now();
|
String fullLocalImagePath = "C://" + now.getYear() + "/" + now.getMonthValue();
|
Path path = Paths.get(fullLocalImagePath);
|
File directory = path.toFile();
|
// 确保目录存在
|
if (!directory.exists()) {
|
directory.mkdirs();
|
}
|
// 生成文件名
|
String fileName = content + "." + format;
|
Path filePath = path.resolve(fileName);
|
// 使用FileOutputStream或者Files.newOutputStream()将二维码写入文件
|
try (OutputStream os = Files.newOutputStream(filePath)) {
|
MatrixToImageWriter.writeToStream(bitMatrix, format, os);
|
}
|
return "";
|
} catch (IOException | WriterException e) {
|
throw new RuntimeException("Error generating or saving the QR code.", e);
|
}
|
}
|
|
public static String uploadQRCodeToOSS(String content, int width, int height, String ossImagePath, HuaWeiOss aliOss) throws Exception {
|
String format = "png";
|
// 定义二维码的参数
|
HashMap<EncodeHintType, Object> hints = new HashMap<>();
|
// 定义字符集编码格式
|
hints.put(EncodeHintType.CHARACTER_SET, "utf-8");
|
// 纠错的等级 L > M > Q > H 纠错的能力越高可存储的越少，一般使用M
|
hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.M);
|
// 设置图片边距
|
hints.put(EncodeHintType.MARGIN, 2);
|
|
try {
|
// 最终生成 参数列表 （1.内容 2.格式 3.宽度 4.高度 5.二维码参数）
|
BitMatrix bitMatrix = new MultiFormatWriter().encode(content, BarcodeFormat.QR_CODE, width, height, hints);
|
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
|
MatrixToImageWriter.writeToStream(bitMatrix, format, outputStream);
|
byte[] imageBytes = outputStream.toByteArray();
|
|
if (Objects.nonNull(aliOss) && BooleanUtil.isTrue(aliOss.getIsOpen())){
|
// 创建ObsClient实例
|
ObsClient obsClient = new ObsClient(aliOss.getAccessKeyId(), aliOss.getSecretAccessKey(), aliOss.getEndpoint());
|
InputStream input = null;
|
try {
|
input = new ByteArrayInputStream(imageBytes);
|
obsClient.putObject(aliOss.getBucketName(), ossImagePath, input);
|
} catch (Exception e) {
|
System.out.println("华为Oss上传文件错误");
|
} finally {
|
obsClient.close();
|
if (Objects.nonNull(input)) {
|
input.close();
|
}
|
}
|
}
|
// 返回华为云OBS图片地址
|
String bucketName = aliOss.getBucketName();//shjw2024
|
String endpoint = aliOss.getEndpoint(); //https://oss-cn-shanghai.aliyuncs.com
|
String pre = endpoint.substring(0, 8);
|
String fix = endpoint.substring(8);
|
String path = pre + bucketName + "." + fix;
|
String img = path + "/" + ossImagePath;
|
return img;
|
} catch (IOException | WriterException e) {
|
throw new RuntimeException(e);
|
}
|
}
|
}
|
