liujie
2026-06-09 c96fbe6b0aeb413a17c429133ee6d81d138144b9
xp-robotaxi-operation-boot/src/main/java/com/xiaopeng/operation/controller/XpInvitationCodeController.java
@@ -10,16 +10,23 @@
import cn.hutool.core.util.DesensitizedUtil;
import cn.hutool.core.util.PhoneUtil;
import cn.hutool.core.io.resource.ClassPathResource;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
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.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;
@@ -33,6 +40,7 @@
import org.apache.poi.ss.usermodel.Workbook;
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;
@@ -52,6 +60,9 @@
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>
@@ -76,6 +87,8 @@
    private SysGroupMenuClient groupMenuClient;
    @Resource
    private TokenService tokenService;
    @Autowired
    private UserResourcesClient userResourcesClient;
    private static final Pattern CHINESE_PATTERN = Pattern.compile("\\p{IsHan}");
    private static final long INVITATION_CODE_LIMIT = 500L;
@@ -97,7 +110,13 @@
    @PostMapping("/api/v1/invitation/add")
    @ResponseBody
    @ApiOperation(value = "生成邀请码")
    public R add(@RequestBody InvitationAddDto dto) {
    @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)
@@ -122,7 +141,13 @@
    @ApiImplicitParams({
            @ApiImplicitParam(value = "id", name = "id", dataType = "Long", required = true)
    })
    public R unbind(@RequestParam("id") Long id) {
    @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);
@@ -165,6 +190,12 @@
    @ApiOperation(value = "邀请码导入")
    @PostMapping("/base/v1/invitation/importExcel")
    @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); //表头行数
@@ -280,11 +311,14 @@
                String res = replace.replace("/", "-");
                locationExcel.setBindTime(res);
            }
            xpInvitationCode.setBindTime(DateUtils.stringToLocalDateTime(locationExcel.getBindTime()));
            xpInvitationCode.setStatus(2);
            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);
@@ -313,6 +347,12 @@
    @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<>();
@@ -357,9 +397,24 @@
    @GetMapping("/api/v1/invitation/deleteByIds")
    @ResponseBody
    @ApiOperation(value = "批量删除")
    public R deleteByIds(String ids) {
    @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();
    }