Merge branch '1.0' of http://120.76.84.145:10101/gitblit/r/java/Robotaxi in...

liujie

2026-06-09 c96fbe6b0aeb413a17c429133ee6d81d138144b9

 xp-robotaxi-operation-boot/src/main/java/com/xiaopeng/operation/controller/XpInvitationCodeController.java

@@ -6,32 +6,43 @@
import cn.afterturn.easypoi.excel.entity.ExportParams;
import cn.afterturn.easypoi.excel.entity.ImportParams;
import cn.afterturn.easypoi.excel.entity.TemplateExportParams;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.DesensitizedUtil;
import cn.hutool.core.util.PhoneUtil;
import cn.hutool.core.io.resource.ClassPathResource;
import cn.hutool.json.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.xiaopeng.api.model.account.SysUser;
import com.xiaopeng.api.dto.account.UcUserInfo;
import com.xiaopeng.api.feignClient.account.SysGroupMenuClient;
import com.xiaopeng.api.feignClient.account.UserResourcesClient;
import com.xiaopeng.api.model.operation.XpInvitationCode;
import com.xiaopeng.api.model.operation.XpWhitelist;
import com.xiaopeng.api.vo.account.LoginUser;
import com.xiaopeng.common.core.domain.R;
import com.xiaopeng.common.core.exception.ServiceException;
import com.xiaopeng.common.core.utils.DateUtils;
import com.xiaopeng.common.core.utils.SensitiveDataCryptoUtil;
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.common.log.annotation.AdminOperationLog;
import com.xiaopeng.common.log.enums.AdminBusinessType;
import com.xiaopeng.common.log.enums.AdminOperationType;
import com.xiaopeng.common.security.service.TokenService;
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.beans.factory.annotation.Autowired;
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;
@@ -41,7 +52,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;
@@ -49,6 +59,10 @@
import java.time.format.DateTimeFormatter;
import java.util.*;
import java.util.stream.Collectors;
import java.util.regex.Pattern;
import com.xiaopeng.common.log.annotation.AdminOperationLog;
import com.xiaopeng.common.log.enums.AdminBusinessType;
import com.xiaopeng.common.log.enums.AdminOperationType;

/**
 * <p>
@@ -67,23 +81,57 @@
    private IXpInvitationCodeService xpInvitationCodeService;
    @Resource
    private IXpWhitelistService xpWhitelistService;
    @Resource
    private MessageSource messageSource;
    @Resource
    private SysGroupMenuClient groupMenuClient;
    @Resource
    private TokenService tokenService;
    @Autowired
    private UserResourcesClient userResourcesClient;

    @ApiOperation(value = "邀请码管理分页列表",response = XpInvitationCode.class)
    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")
    public R<PageInfo<XpInvitationCode>> pageList(@RequestBody InvitationPageListQuery query) {
        LoginUser loginUser = tokenService.getLoginUser();
        PageInfo<XpInvitationCode> pageList = xpInvitationCodeService.pageList(query);
        if(!loginUser.getPhoneDesensitizationExempt()){
            pageList.getRecords().forEach(s -> s.setBindAccount(DesensitizedUtil.mobilePhone(s.getBindAccount())));
        }
        return R.ok(pageList);
    }

    /**
     * 生成邀请码
     */
    @PostMapping("/api/v1/invitation/add")
    @ResponseBody
    @ApiOperation(value = "生成邀请码")
    @Transactional
    @AdminOperationLog(
            businessType = AdminBusinessType.INVITATION_CODE_MANAGEMENT,
            operationType = AdminOperationType.ADD,
            content = "生成邀请码",
            appendRequestData = true
    )
    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());
    }

    /**
     * 解绑
     */
@@ -93,14 +141,21 @@
    @ApiImplicitParams({
            @ApiImplicitParam(value = "id", name = "id", dataType = "Long", required = true)
    })
    @AdminOperationLog(
            businessType = AdminBusinessType.INVITATION_CODE_MANAGEMENT,
            operationType = AdminOperationType.EDIT,
            contentExpression = "'解绑邀请码,id=' + #p0",
            appendRequestData = false
    )
    public R unbind(@RequestParam("id") Long id){
        UpdateWrapper<XpInvitationCode> xpInvitationCodeUpdateWrapper = new UpdateWrapper<>();
        xpInvitationCodeUpdateWrapper.eq("id",id);
        xpInvitationCodeUpdateWrapper.set("status",3);
        xpInvitationCodeUpdateWrapper.eq("id", id);
        xpInvitationCodeUpdateWrapper.set("status", 3);
        xpInvitationCodeUpdateWrapper.set("bind_account", null);
        xpInvitationCodeService.update(xpInvitationCodeUpdateWrapper);
        return R.ok();
    }

    @ApiOperation(value = "邀请码导入模板下载")
    @GetMapping("/api/v1/invitation/import-template")
    public void importTemplate() {
@@ -132,12 +187,16 @@
            }
        }
    }

    @ApiOperation(value = "邀请码导入")
    @PostMapping("/base/v1/invitation/importExcel")
    public R<String> importExcel(@RequestPart("file") MultipartFile file) {
        System.out.println("文件名 = " + file.getOriginalFilename());
        System.out.println("文件大小 = " + file.getSize());
        System.out.println("ContentType = " + file.getContentType());
    @AdminOperationLog(
            businessType = AdminBusinessType.INVITATION_CODE_MANAGEMENT,
            operationType = AdminOperationType.IMPORT,
            content = "导入邀请码",
            appendRequestData = false
    )
    public R<String> importExcel(@RequestParam("file") MultipartFile file) {
        ImportParams params = new ImportParams();
        params.setHeadRows(1); //表头行数
        InputStream inputStream = null;
@@ -152,64 +211,148 @@
            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());
            xpInvitationCode.setBindAccount(locationExcel.getBindAccount());
            if (StringUtils.hasLength(locationExcel.getBindTime())){
            if (StringUtils.hasLength(locationExcel.getBindTime())) {
                String replace = locationExcel.getBindTime().replace(".", "-");
                String res = replace.replace("/", "-");
                locationExcel.setBindTime(res);
            }

            xpInvitationCode.setBindTime(DateUtils.stringToLocalDateTime(locationExcel.getBindTime()));
            if (StringUtils.hasLength(locationExcel.getBindAccount())){
                xpInvitationCode.setStatus(2);
            }else{
                xpInvitationCode.setStatus(1);
            UcUserInfo userByMobile = userResourcesClient.getUserByMobile(locationExcel.getBindAccount());
            if(null == userByMobile || StringUtils.isEmpty(userByMobile.getUid())){
                throw new ServiceException("手机号"+locationExcel.getBindAccount()+"查询用户失败");
            }
            XpWhitelist xpWhitelist = new XpWhitelist();
            xpWhitelist.setAccount(locationExcel.getBindAccount());
            xpWhitelist.setAccount(SensitiveDataCryptoUtil.encrypt(locationExcel.getBindAccount()));
            xpWhitelist.setNickName(userByMobile.getNickName());
            xpWhitelist.setAppUserId(userByMobile.getUid());
            xpWhitelist.setStatus(true);
            xpWhitelist.setIsPriorityDispatch(false);
            xpWhitelist.setWhitelistSource(2);
            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")
    @AdminOperationLog(
            businessType = AdminBusinessType.INVITATION_CODE_MANAGEMENT,
            operationType = AdminOperationType.EXPORT,
            content = "导出邀请码",
            appendRequestData = true
    )
    public void exportExcel(@RequestBody InvitationPageListQuery query) {
        List<XpInvitationCode> list = xpInvitationCodeService.exportList(query);
        List<XpInvitationCodeExportExcel> xpWhitelistExportExcels = new ArrayList<>();
@@ -247,17 +390,42 @@
        }

    }

    /**
     * 批量删除
     */
    @GetMapping("/api/v1/invitation/deleteByIds")
    @ResponseBody
    @ApiOperation(value = "批量删除")
    @AdminOperationLog(
            businessType = AdminBusinessType.INVITATION_CODE_MANAGEMENT,
            operationType = AdminOperationType.DELETE,
            contentExpression = "'批量删除邀请码,ids=' + #p0",
            appendRequestData = false
    )
    public R deleteByIds(String ids){
        // 再转化为Long类型
        List<Long> longs = Arrays.stream(ids.split(",")).map(Long::parseLong).collect(Collectors.toList());
        List<XpInvitationCode> xpInvitationCodes = xpInvitationCodeService.listByIds(longs);
        for (XpInvitationCode xpInvitationCode : xpInvitationCodes) {
            XpWhitelist one = xpWhitelistService.getOne(new QueryWrapper<XpWhitelist>().eq("whitelist_source", 2)
                    .eq("status", 1).eq("is_deleted", 0)
                    .eq("account", SensitiveDataCryptoUtil.encrypt(xpInvitationCode.getBindAccount())).last(" limit 1"));
            if(null != one){
                xpWhitelistService.removeById(one);
            }
        }
        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);
    }
}