java/Robotaxi.git

			@@ -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,20 @@
			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.apache.shardingsphere.transaction.annotation.ShardingSphereTransactionType;
			import org.apache.shardingsphere.transaction.core.TransactionType;
			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 +40,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 +47,7 @@
			import java.time.format.DateTimeFormatter;
			import java.util.*;
			import java.util.stream.Collectors;
			import java.util.regex.Pattern;

			/**
			* <p>
			@@ -64,6 +66,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 +81,23 @@
			/**
			* 生成邀请码
			*/
			@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 +116,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 +125,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 +149,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 +164,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.isMobile(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 +266,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 +275,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 +301,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 +345,14 @@
			xpInvitationCodeService.removeBatchByIds(longs);
			return R.ok();
			}


			@ApiOperation(value = "校验生成邀请码的手机号是否合法")
			@GetMapping("/api/v1/invitation/validateCode")
			public R<List<String>> validateCode(@RequestParam String phone){
			List<String> phoneList = Arrays.asList(phone.split(","));
			List<String> existPhones = phoneList.stream().filter(item -> xpInvitationCodeService.lambdaQuery()
			.in(XpInvitationCode::getBindAccount, item)
			.count() > 0).collect(Collectors.toList());
			return R.ok(existPhones);
			}
			}