From c461bd0c3c721b68b954fe23d50c6cbaefdb12fe Mon Sep 17 00:00:00 2001
From: “mitao” <“2763622819@qq.com”>
Date: 星期三, 11 三月 2026 20:38:18 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'
---
xp-robotaxi-operation-boot/src/main/java/com/xiaopeng/operation/controller/XpInvitationCodeController.java | 192 ++++++++++++++++++++++++++++++++++++-----------
1 files changed, 147 insertions(+), 45 deletions(-)
diff --git a/xp-robotaxi-operation-boot/src/main/java/com/xiaopeng/operation/controller/XpInvitationCodeController.java b/xp-robotaxi-operation-boot/src/main/java/com/xiaopeng/operation/controller/XpInvitationCodeController.java
index 0f0c320..9a53ade 100644
--- a/xp-robotaxi-operation-boot/src/main/java/com/xiaopeng/operation/controller/XpInvitationCodeController.java
+++ b/xp-robotaxi-operation-boot/src/main/java/com/xiaopeng/operation/controller/XpInvitationCodeController.java
@@ -5,9 +5,11 @@
import cn.afterturn.easypoi.excel.ExcelImportUtil;
import cn.afterturn.easypoi.excel.entity.ExportParams;
import cn.afterturn.easypoi.excel.entity.ImportParams;
-import cn.hutool.json.JSONObject;
+import cn.afterturn.easypoi.excel.entity.TemplateExportParams;
+import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.util.PhoneUtil;
+import cn.hutool.core.io.resource.ClassPathResource;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
-import com.xiaopeng.api.model.account.SysUser;
import com.xiaopeng.api.model.operation.XpInvitationCode;
import com.xiaopeng.api.model.operation.XpWhitelist;
import com.xiaopeng.common.core.domain.R;
@@ -15,20 +17,18 @@
import com.xiaopeng.common.core.utils.DateUtils;
import com.xiaopeng.common.core.utils.WebUtils;
import com.xiaopeng.common.core.web.page.PageInfo;
-import com.xiaopeng.common.log.annotation.Log;
-import com.xiaopeng.common.log.enums.BusinessType;
+import com.xiaopeng.operation.constant.I18nKeyConstant;
+import com.xiaopeng.operation.dto.InvitationAddDto;
import com.xiaopeng.operation.exportExcle.XpInvitationCodeExportExcel;
-import com.xiaopeng.operation.exportExcle.XpWhitelistExportExcel;
import com.xiaopeng.operation.importExcel.InvitationCodeImportExcel;
import com.xiaopeng.operation.query.InvitationPageListQuery;
-import com.xiaopeng.operation.query.WhitelistPageListQuery;
import com.xiaopeng.operation.service.IXpInvitationCodeService;
import com.xiaopeng.operation.service.IXpWhitelistService;
import io.swagger.annotations.*;
import lombok.extern.slf4j.Slf4j;
-import org.apache.poi.openxml4j.exceptions.NotOfficeXmlFileException;
import org.apache.poi.ss.usermodel.Workbook;
-import org.springframework.transaction.annotation.Propagation;
+import org.springframework.context.MessageSource;
+import org.springframework.context.i18n.LocaleContextHolder;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;
import org.springframework.util.StringUtils;
@@ -38,7 +38,6 @@
import javax.annotation.Resource;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
-import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.net.URLEncoder;
@@ -46,6 +45,7 @@
import java.time.format.DateTimeFormatter;
import java.util.*;
import java.util.stream.Collectors;
+import java.util.regex.Pattern;
/**
* <p>
@@ -64,6 +64,11 @@
private IXpInvitationCodeService xpInvitationCodeService;
@Resource
private IXpWhitelistService xpWhitelistService;
+ @Resource
+ private MessageSource messageSource;
+
+ private static final Pattern CHINESE_PATTERN = Pattern.compile("\\p{IsHan}");
+ private static final long INVITATION_CODE_LIMIT = 500L;
@ApiOperation(value = "邀请码管理分页列表",response = XpInvitationCode.class)
@PostMapping("/api/v1/invitation/pageList")
@@ -74,15 +79,24 @@
/**
* 生成邀请码
*/
- @PutMapping("/api/v1/invitation/add")
+ @PostMapping("/api/v1/invitation/add")
@ResponseBody
@ApiOperation(value = "生成邀请码")
- @ApiImplicitParams({
- @ApiImplicitParam(value = "手机号 多个逗号隔开", name = "phones", dataType = "String", required = true)
- })
@Transactional
- public R add(@RequestParam("phones") String phones){
- return xpInvitationCodeService.add(phones);
+ public R add(@RequestBody InvitationAddDto dto){
+ long existing = xpInvitationCodeService.count();
+ long incoming = Arrays.stream(Optional.ofNullable(dto.getPhones()).orElse("").split(","))
+ .map(String::trim)
+ .filter(StringUtils::hasLength)
+ .count();
+ if (existing >= INVITATION_CODE_LIMIT || existing + incoming > INVITATION_CODE_LIMIT) {
+ String msg = messageSource.getMessage(I18nKeyConstant.INVITATION_LIMIT_EXCEEDED,
+ new Object[]{INVITATION_CODE_LIMIT, existing, incoming},
+ I18nKeyConstant.INVITATION_LIMIT_EXCEEDED,
+ LocaleContextHolder.getLocale());
+ return R.fail(msg);
+ }
+ return xpInvitationCodeService.add(dto.getPhones());
}
/**
* 解绑
@@ -101,7 +115,7 @@
xpInvitationCodeService.update(xpInvitationCodeUpdateWrapper);
return R.ok();
}
- @ApiOperation(value = "点位导入模板下载")
+ @ApiOperation(value = "邀请码导入模板下载")
@GetMapping("/api/v1/invitation/import-template")
public void importTemplate() {
List<InvitationCodeImportExcel> invitationCodeImportExcels = new ArrayList<>();
@@ -110,14 +124,17 @@
Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), InvitationCodeImportExcel.class, invitationCodeImportExcels);
HttpServletResponse response = WebUtils.response();
ServletOutputStream outputStream = null;
+ //1.获取excel模板
+ ClassPathResource classPathResource = new ClassPathResource("template/邀请码导入模板.xlsx");
+ TemplateExportParams params = new TemplateExportParams(classPathResource.getPath());
+ response.setContentType("application/vnd.ms-excel");
+ response.setCharacterEncoding("utf-8");
try {
String fileName = URLEncoder.encode("邀请码导入模板.xlsx", "utf-8");
- response.setHeader("Content-Disposition", "attachment;filename=" + fileName);
- response.setHeader("content-Type", "application/vnd.ms-excel");
- response.setHeader("Pragma", "no-cache");
- response.setHeader("Cache-Control", "no-cache");
+ response.setHeader("Content-dispodition", "attachment;filename=" + fileName);
outputStream = response.getOutputStream();
workbook.write(outputStream);
+
} catch (IOException e) {
e.printStackTrace();
System.out.println("邀请码导入模板下载失败!");
@@ -131,7 +148,7 @@
}
@ApiOperation(value = "邀请码导入")
@PostMapping("/base/v1/invitation/importExcel")
- public R<String> importExcel(@RequestPart("file") MultipartFile file) {
+ public R<String> importExcel(@RequestParam("file") MultipartFile file) {
ImportParams params = new ImportParams();
params.setHeadRows(1); //表头行数
InputStream inputStream = null;
@@ -146,31 +163,97 @@
throw new ServiceException("邀请码导入失败!");
} finally {
try {
- inputStream.close();
+ if (inputStream != null) {
+ inputStream.close();
+ }
} catch (IOException e) {
throw new ServiceException(e.getMessage());
}
}
if (CollectionUtils.isEmpty(invitationCodeImportExcels)) {
- throw new ServiceException("邀请码数据为空!");
+ return R.fail(messageSource.getMessage(I18nKeyConstant.INVITATION_IMPORT_DATA_EMPTY, null, I18nKeyConstant.INVITATION_IMPORT_DATA_EMPTY, LocaleContextHolder.getLocale()));
}
- // 校验邀请码是否有重复的
- long count = invitationCodeImportExcels.stream().map(InvitationCodeImportExcel::getInvitationCode)
- .distinct().count();
- if (count!=invitationCodeImportExcels.size()){
- return R.fail("邀请码存在重复!");
+
+ long existing = xpInvitationCodeService.count();
+ long incoming = invitationCodeImportExcels.size();
+ if (existing >= INVITATION_CODE_LIMIT || existing + incoming > INVITATION_CODE_LIMIT) {
+ String msg = messageSource.getMessage(I18nKeyConstant.INVITATION_LIMIT_EXCEEDED,
+ new Object[]{INVITATION_CODE_LIMIT, existing, incoming},
+ I18nKeyConstant.INVITATION_LIMIT_EXCEEDED,
+ LocaleContextHolder.getLocale());
+ return R.fail(msg);
}
- Map<String, Long> codeMap = xpInvitationCodeService.list().stream()
- .collect(Collectors.toMap(XpInvitationCode::getInvitationCode, XpInvitationCode::getId));
- JSONObject result = new JSONObject();
+
+ // 批量校验：账号必须为手机号；邀请码不能为空且不能包含中文；同时校验文件内重复/数据库已存在
+ List<String> errors = new ArrayList<>();
+ Set<String> importCodes = new HashSet<>();
+ for (int i = 0; i < invitationCodeImportExcels.size(); i++) {
+ InvitationCodeImportExcel row = invitationCodeImportExcels.get(i);
+ int rowNum = i + 2;
+
+ String invitationCode = trimToNull(row.getInvitationCode());
+ String bindAccount = trimToNull(row.getBindAccount());
+ row.setInvitationCode(invitationCode);
+ row.setBindAccount(bindAccount);
+
+ if (!StringUtils.hasLength(invitationCode)) {
+ errors.add(messageSource.getMessage(I18nKeyConstant.INVITATION_IMPORT_CODE_EMPTY, new Object[]{rowNum}, I18nKeyConstant.INVITATION_IMPORT_CODE_EMPTY, LocaleContextHolder.getLocale()));
+ } else {
+ if (containsChinese(invitationCode)) {
+ errors.add(messageSource.getMessage(I18nKeyConstant.INVITATION_IMPORT_CODE_CONTAINS_CHINESE, new Object[]{rowNum, bindAccount}, I18nKeyConstant.INVITATION_IMPORT_CODE_CONTAINS_CHINESE, LocaleContextHolder.getLocale()));
+ }
+ importCodes.add(invitationCode);
+ }
+
+ if (!StringUtils.hasLength(bindAccount)) {
+ errors.add(messageSource.getMessage(I18nKeyConstant.INVITATION_IMPORT_ACCOUNT_EMPTY, new Object[]{rowNum}, I18nKeyConstant.INVITATION_IMPORT_ACCOUNT_EMPTY, LocaleContextHolder.getLocale()));
+ } else if (!PhoneUtil.isPhone(bindAccount)) {
+ errors.add(messageSource.getMessage(I18nKeyConstant.INVITATION_IMPORT_ACCOUNT_INVALID, new Object[]{rowNum, bindAccount}, I18nKeyConstant.INVITATION_IMPORT_ACCOUNT_INVALID, LocaleContextHolder.getLocale()));
+ }
+
+ if (StringUtils.hasLength(row.getBindTime())) {
+ row.setBindTime(row.getBindTime().trim());
+ }
+ }
+
+ Map<String, Long> codeCountMap = invitationCodeImportExcels.stream()
+ .map(InvitationCodeImportExcel::getInvitationCode)
+ .filter(StringUtils::hasLength)
+ .collect(Collectors.groupingBy(code -> code, Collectors.counting()));
+ List<String> duplicatedCodes = codeCountMap.entrySet().stream()
+ .filter(entry -> entry.getValue() != null && entry.getValue() > 1)
+ .map(Map.Entry::getKey)
+ .sorted()
+ .collect(Collectors.toList());
+ if (!duplicatedCodes.isEmpty()) {
+ errors.add(messageSource.getMessage(I18nKeyConstant.INVITATION_IMPORT_CODE_DUPLICATE, new Object[]{String.join(",", duplicatedCodes)}, I18nKeyConstant.INVITATION_IMPORT_CODE_DUPLICATE, LocaleContextHolder.getLocale()));
+ }
+
+ if (!importCodes.isEmpty()) {
+ List<XpInvitationCode> existsList = xpInvitationCodeService.lambdaQuery()
+ .in(XpInvitationCode::getInvitationCode, importCodes)
+ .list();
+ if (!CollectionUtils.isEmpty(existsList)) {
+ List<String> existsCodes = existsList.stream()
+ .map(XpInvitationCode::getInvitationCode)
+ .filter(StringUtils::hasLength)
+ .distinct()
+ .sorted()
+ .collect(Collectors.toList());
+ if (!existsCodes.isEmpty()) {
+ errors.add(messageSource.getMessage(I18nKeyConstant.INVITATION_IMPORT_CODE_EXISTS, new Object[]{String.join(",", existsCodes)}, I18nKeyConstant.INVITATION_IMPORT_CODE_EXISTS, LocaleContextHolder.getLocale()));
+ }
+ }
+ }
+
+ if (!errors.isEmpty()) {
+ return R.fail(messageSource.getMessage(I18nKeyConstant.INVITATION_IMPORT_VALIDATE_FAIL, new Object[]{String.join("\n", errors)},I18nKeyConstant.INVITATION_IMPORT_VALIDATE_FAIL, LocaleContextHolder.getLocale()));
+ }
+
List<XpInvitationCode> addList = new ArrayList<>();
List<XpWhitelist> addWhitelistList = new ArrayList<>();
for (InvitationCodeImportExcel locationExcel : invitationCodeImportExcels) {
- Long id = codeMap.get(locationExcel.getInvitationCode());
- if (id!=null){
- result.append("邀请码：[", locationExcel.getInvitationCode()+"]已存在");
- }
locationExcel.setBindTime(locationExcel.getBindTime());
XpInvitationCode xpInvitationCode = new XpInvitationCode();
xpInvitationCode.setInvitationCode(locationExcel.getInvitationCode());
@@ -182,11 +265,7 @@
}
xpInvitationCode.setBindTime(DateUtils.stringToLocalDateTime(locationExcel.getBindTime()));
- if (StringUtils.hasLength(locationExcel.getBindAccount())){
- xpInvitationCode.setStatus(2);
- }else{
- xpInvitationCode.setStatus(1);
- }
+ xpInvitationCode.setStatus(2);
XpWhitelist xpWhitelist = new XpWhitelist();
xpWhitelist.setAccount(locationExcel.getBindAccount());
xpWhitelist.setStatus(true);
@@ -195,12 +274,24 @@
addWhitelistList.add(xpWhitelist);
addList.add(xpInvitationCode);
}
- if(!result.isEmpty()){
- return R.ok(result.toString());
- }
xpInvitationCodeService.saveBatch(addList);
xpWhitelistService.saveBatch(addWhitelistList);
return R.ok();
+ }
+
+ private static String trimToNull(String value) {
+ if (!StringUtils.hasLength(value)) {
+ return null;
+ }
+ String trimmed = value.trim();
+ return StringUtils.hasLength(trimmed) ? trimmed : null;
+ }
+
+ private static boolean containsChinese(String value) {
+ if (!StringUtils.hasLength(value)) {
+ return false;
+ }
+ return CHINESE_PATTERN.matcher(value).find();
}
@ApiOperation(value = "邀请码导出")
@PostMapping(value = "/api/v1/invitation/exportExcel")
@@ -209,7 +300,7 @@
List<XpInvitationCodeExportExcel> xpWhitelistExportExcels = new ArrayList<>();
for (XpInvitationCode xpInvitationCode : list) {
XpInvitationCodeExportExcel xpWhitelistExportExcel = new XpInvitationCodeExportExcel();
- xpWhitelistExportExcel.setId(xpInvitationCode.getId());
+// xpWhitelistExportExcel.setId(xpInvitationCode.getId());
xpWhitelistExportExcel.setBind_account(xpInvitationCode.getBindAccount());
xpWhitelistExportExcel.setInvitationCode(xpInvitationCode.getInvitationCode());
LocalDateTime createTime = xpInvitationCode.getCreateTime();
@@ -253,5 +344,16 @@
xpInvitationCodeService.removeBatchByIds(longs);
return R.ok();
}
-
+
+ @ApiOperation(value = "校验生成邀请码的手机号是否合法")
+ @PostMapping("/api/v1/invitation/validateCode")
+ public R<Boolean> validateCode(@RequestParam String phone){
+ if (!PhoneUtil.isMobile(phone)){
+ return R.ok(false);
+ }
+ List<XpInvitationCode> existsList = xpInvitationCodeService.lambdaQuery()
+ .in(XpInvitationCode::getBindAccount, phone)
+ .list();
+ return R.ok(CollUtil.isEmpty(existsList));
+ }
}
--
Gitblit v1.7.1