“mitao”
2025-12-22 c30851026169acc0f330825d90baf435edb4341d
ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TAppUserController.java
@@ -1,28 +1,36 @@
package com.ruoyi.web.controller.api;
import cn.afterturn.easypoi.excel.ExcelExportUtil;
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 com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.extension.conditions.update.LambdaUpdateChainWrapper;
import com.ruoyi.common.annotation.Log;
import com.ruoyi.common.basic.PageInfo;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.common.core.exception.ServiceException;
import com.ruoyi.common.enums.BusinessType;
import com.ruoyi.common.utils.WebUtils;
import com.ruoyi.framework.web.service.TokenService;
import com.ruoyi.system.dto.BatchUpdateIntegralDto;
import com.ruoyi.system.dto.UpdateAppUserDto;
import com.ruoyi.system.export.AppUserGoldImport;
import com.ruoyi.system.export.AppUserImport;
import com.ruoyi.system.model.TAppUser;
import com.ruoyi.system.model.TCompany;
import com.ruoyi.system.model.TGoldChangeRecord;
import com.ruoyi.system.export.BigGiftPackageImport;
import com.ruoyi.system.model.*;
import com.ruoyi.system.query.AppUserListQuery;
import com.ruoyi.system.query.BigGiftPackageQuery;
import com.ruoyi.system.service.TAppUserService;
import com.ruoyi.system.service.TCompanyService;
import com.ruoyi.system.service.TGoldChangeRecordService;
import com.ruoyi.system.service.TIntegralChangeRecordService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.apache.poi.ss.usermodel.Workbook;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.transaction.annotation.Propagation;
@@ -33,14 +41,18 @@
import org.springframework.web.multipart.MultipartFile;
import javax.annotation.Resource;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.net.URLEncoder;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
/**
 * <p>
@@ -57,7 +69,7 @@
    @Resource
    private TAppUserService  appUserService;
    private TAppUserService appUserService;
    @Autowired
@@ -72,6 +84,9 @@
    @Autowired
    private TGoldChangeRecordService goldChangeRecordService;
    @Autowired
    private TIntegralChangeRecordService integralChangeRecordService;
    @ApiOperation(value = "用户管理分页列表")
    @PostMapping(value = "/pageList")
@@ -85,7 +100,7 @@
    @Log(title = "小程序用户-修改状态", businessType = BusinessType.UPDATE)
    public R pageList(@RequestParam Integer id) {
        TAppUser appUser = appUserService.getById(id);
        appUser.setStatus(appUser.getStatus()==1?2:1);
        appUser.setStatus(appUser.getStatus() == 1 ? 2 : 1);
        appUserService.updateById(appUser);
        return R.ok();
    }
@@ -96,7 +111,7 @@
    @Log(title = "小程序用户-修改用户信息", businessType = BusinessType.UPDATE)
    public R updateAppUser(@RequestBody UpdateAppUserDto dto) {
        if(dto.getCode()!=null && !dto.getCode().isEmpty()){
        if (dto.getCode() != null && !dto.getCode().isEmpty()) {
            long count = appUserService.count(new LambdaQueryWrapper<TAppUser>().eq(TAppUser::getCode, dto.getCode()).ne(TAppUser::getId, dto.getId()));
            if (count > 0) {
                return R.fail("推广号已存在");
@@ -104,21 +119,21 @@
        }
        TAppUser appUser = appUserService.getById(dto.getId());
        String code = appUser.getCode();
        if(dto.getCode()!=null && !dto.getCode().isEmpty() && !code.equals(dto.getCode())){
        if (dto.getCode() != null && !dto.getCode().isEmpty() && !code.equals(dto.getCode())) {
            appUser.setCode(dto.getCode());
            // 移除之前的二维码
            redisTemplate.delete("USER_WX_IMG" + appUser.getId());
        }
        if(dto.getPromotionStatus()!=null){
        if (dto.getPromotionStatus() != null) {
            appUser.setPromotionStatus(dto.getPromotionStatus());
        }
        if(dto.getUserType()!=null){
        if (dto.getUserType() != null) {
            appUser.setUserType(dto.getUserType());
        }
        if(dto.getReferrerUserId()!=null){
        if (dto.getReferrerUserId() != null) {
            appUser.setReferrerUserId(dto.getReferrerUserId());
        }
        if(dto.getCompanyId()!=null){
        if (dto.getCompanyId() != null) {
            appUser.setCompanyId(dto.getCompanyId());
        }
        appUserService.updateById(appUser);
@@ -129,7 +144,7 @@
    @ApiOperation(value = "查看详情")
    @GetMapping(value = "/detail")
    public R<TAppUser> detail(@RequestParam String id) {
        return R.ok(appUserService.getById( id));
        return R.ok(appUserService.getById(id));
    }
@@ -165,26 +180,26 @@
        JSONObject result = new JSONObject();
        int i = 1;
        for (AppUserGoldImport appUserGoldImport : imports) {
            if (!StringUtils.hasLength(appUserGoldImport.getUserPhone())){
                result.append("第"+i+"行:[", "用户手机号"+"]不能为空");
            if (!StringUtils.hasLength(appUserGoldImport.getUserPhone())) {
                result.append("第" + i + "行:[", "用户手机号" + "]不能为空");
                continue;
            }
            if (!StringUtils.hasLength(appUserGoldImport.getGold())){
                result.append("第"+i+"行:[", "金币"+"]不能为空");
            if (!StringUtils.hasLength(appUserGoldImport.getGold())) {
                result.append("第" + i + "行:[", "金币" + "]不能为空");
                continue;
            }
            TAppUser user = appUserService.getOne(new LambdaQueryWrapper<TAppUser>().eq(TAppUser::getPhone, appUserGoldImport.getUserPhone()).ne(TAppUser::getStatus, 3).last("LIMIT 1"));
            if (user==null){
                result.append("第"+i+"行:[", "用户不存在"+"]");
            if (user == null) {
                result.append("第" + i + "行:[", "用户不存在" + "]");
                continue;
            }
            BigDecimal totalPrice= BigDecimal.ZERO;
            BigDecimal totalPrice = BigDecimal.ZERO;
            try {
                totalPrice= new BigDecimal(appUserGoldImport.getGold());
                totalPrice = new BigDecimal(appUserGoldImport.getGold());
                user.setGoldBalance(user.getGoldBalance().add(totalPrice).setScale(2, RoundingMode.HALF_UP));
            } catch (Exception e) {
                result.append("第"+i+"行:[", "金币"+"]格式错误");
                result.append("第" + i + "行:[", "金币" + "]格式错误");
                continue;
            }
@@ -197,15 +212,14 @@
            goldChangeRecord.setCreateUserId(userId);
            goldChangeRecordService.save(goldChangeRecord);
            appUserService.updateById( user);
            appUserService.updateById(user);
            i++;
        }
        if(!result.isEmpty()){
        if (!result.isEmpty()) {
            return R.fail(result.toString());
        }
        return R.ok();
    }
    @Log(title = "小程序用户-用户导入", businessType = BusinessType.IMPORT)
@@ -238,46 +252,46 @@
        JSONObject result = new JSONObject();
        int i = 1;
        for (AppUserImport appUserImport : imports) {
            if (!StringUtils.hasLength(appUserImport.getUsername())){
                result.append("第"+i+"行:[", "用户名"+"]不能为空");
            if (!StringUtils.hasLength(appUserImport.getUsername())) {
                result.append("第" + i + "行:[", "用户名" + "]不能为空");
                continue;
            }
            if (!StringUtils.hasLength(appUserImport.getPhone())){
                result.append("第"+i+"行:[", "联系电话"+"]不能为空");
            if (!StringUtils.hasLength(appUserImport.getPhone())) {
                result.append("第" + i + "行:[", "联系电话" + "]不能为空");
                continue;
            }
            if (!StringUtils.hasLength(appUserImport.getSex())){
                result.append("第"+i+"行:[", "性别"+"]不能为空");
            if (!StringUtils.hasLength(appUserImport.getSex())) {
                result.append("第" + i + "行:[", "性别" + "]不能为空");
                continue;
            }
            if (!StringUtils.hasLength(appUserImport.getBirth())){
                result.append("第"+i+"行:[", "出生日期"+"]不能为空");
            if (!StringUtils.hasLength(appUserImport.getBirth())) {
                result.append("第" + i + "行:[", "出生日期" + "]不能为空");
                continue;
            }
            if (!StringUtils.hasLength(appUserImport.getRole())){
                result.append("第"+i+"行:[", "角色"+"]不能为空");
            if (!StringUtils.hasLength(appUserImport.getRole())) {
                result.append("第" + i + "行:[", "角色" + "]不能为空");
                continue;
            }
            if (!StringUtils.hasLength(appUserImport.getCode())){
                result.append("第"+i+"行:[", "推广号"+"]不能为空");
            if (!StringUtils.hasLength(appUserImport.getCode())) {
                result.append("第" + i + "行:[", "推广号" + "]不能为空");
                continue;
            }
            TAppUser user = appUserService.getOne(new LambdaQueryWrapper<TAppUser>().eq(TAppUser::getPhone, appUserImport.getPhone()).ne(TAppUser::getStatus, 3).last("LIMIT 1"));
            if (user!=null){
                result.append("第"+i+"行:[", "联系电话已存在"+"]");
            if (user != null) {
                result.append("第" + i + "行:[", "联系电话已存在" + "]");
                continue;
            }
            TAppUser user1 = appUserService.getOne(new LambdaQueryWrapper<TAppUser>().eq(TAppUser::getCode, appUserImport.getCode()).ne(TAppUser::getStatus, 3).last("LIMIT 1"));
            if (user1!=null){
                result.append("第"+i+"行:[", "推广号已存在"+"]");
            if (user1 != null) {
                result.append("第" + i + "行:[", "推广号已存在" + "]");
                continue;
            }
            String sex = appUserImport.getSex();
            Integer gender = "男".equals(sex) ? 1 : "女".equals(sex) ? 2 : null;
            if(gender==null){
                result.append("第"+i+"行:[", "性别"+"]格式错误");
            if (gender == null) {
                result.append("第" + i + "行:[", "性别" + "]格式错误");
                continue;
            }
@@ -286,14 +300,14 @@
            try {
                birthday = LocalDate.parse(birth);
            } catch (Exception e) {
                result.append("第"+i+"行:[", "出生日期"+"]格式错误");
                result.append("第" + i + "行:[", "出生日期" + "]格式错误");
                continue;
            }
            String role = appUserImport.getRole();
            Integer userType = "游客".equals(role) ? 1 : "营销员".equals(role) ? 2 : "合伙人".equals(role) ? 3 : null;
            if(userType==null){
                result.append("第"+i+"行:[", "角色"+"]格式错误");
            if (userType == null) {
                result.append("第" + i + "行:[", "角色" + "]格式错误");
                continue;
            }
@@ -305,28 +319,114 @@
            tAppUser.setBirthday(birthday);
            tAppUser.setUserType(userType);
            if(appUserImport.getReferrerCode()!=null){
            if (appUserImport.getReferrerCode() != null) {
                TAppUser referrerUser = appUserService.getOne(new LambdaQueryWrapper<TAppUser>().eq(TAppUser::getCode, appUserImport.getReferrerCode()).ne(TAppUser::getStatus, 3).last("LIMIT 1"));
                if(referrerUser!=null){
                if (referrerUser != null) {
                    tAppUser.setReferrerUserId(referrerUser.getId());
                    tAppUser.setReferrerStatus(1);
                    tAppUser.setReferrerTime(LocalDateTime.now());
                }
            }
            if(appUserImport.getDealerCode()!=null){
            if (appUserImport.getDealerCode() != null) {
            }
            TCompany one = companyService.getOne(new LambdaQueryWrapper<TCompany>().eq(TCompany::getCode, appUserImport.getDealerCode()).ne(TCompany::getStatus, 3).last("LIMIT 1"));
            if(one!=null){
            if (one != null) {
                tAppUser.setCompanyId(one.getId());
            }
            appUserService.save( user);
            appUserService.save(user);
            i++;
        }
        if(!result.isEmpty()){
        if (!result.isEmpty()) {
            return R.fail(result.toString());
        }
        return R.ok();
    }
    @ApiOperation(value = "小程序用户导出")
    @PostMapping(value = "/exportAppUser")
    @Log(title = "小程序用户-导出", businessType = BusinessType.EXPORT)
    public void exportAppUser(@RequestBody AppUserListQuery query) {
        List<TAppUser> list = appUserService.exportAppUser(query);
        Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), TAppUser.class, list);
        HttpServletResponse response = WebUtils.response();
        assert response != null;
        response.setContentType("application/vnd.ms-excel");
        response.setCharacterEncoding("utf-8");
        ServletOutputStream outputStream = null;
        try {
            String fileName = URLEncoder.encode("小程序用户导出信息.xls", "utf-8");
            response.setHeader("Content-Disposition", "attachment;filename=" + fileName);
            response.setContentType("application/vnd.ms-excel;charset=UTF-8");
            response.setHeader("Pragma", "no-cache");
            response.setHeader("Cache-Control", "no-cache");
            outputStream = response.getOutputStream();
            workbook.write(outputStream);
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                outputStream.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
    @ApiOperation(value = "一键清空金币")
    @GetMapping(value = "/clearGold")
    @Log(title = "小程序用户-一键清空金币", businessType = BusinessType.UPDATE)
    @Transactional(rollbackFor = Exception.class, propagation = Propagation.REQUIRES_NEW)
    public R clearGold() {
        Long userId = tokenService.getLoginUser().getUserId();
        List<TAppUser> list = appUserService.list();
        ArrayList<TGoldChangeRecord> tGoldChangeRecords = new ArrayList<>();
        for (TAppUser user : list) {
            TGoldChangeRecord tGoldChangeRecord = new TGoldChangeRecord();
            tGoldChangeRecord.setAppUserId(user.getId());
            tGoldChangeRecord.setPrice(user.getGoldBalance());
            tGoldChangeRecord.setBalance(BigDecimal.ZERO);
            tGoldChangeRecord.setType(2);
            tGoldChangeRecord.setCreateUserId(userId);
            tGoldChangeRecords.add(tGoldChangeRecord);
        }
        goldChangeRecordService.saveBatch(tGoldChangeRecords);
        appUserService.update(new LambdaUpdateWrapper<TAppUser>().set(TAppUser::getGoldBalance, 0));
        return R.ok();
    }
    @ApiOperation(value = "批量修改积分")
    @PostMapping(value = "/batchUpdateGold")
    @Log(title = "小程序用户-批量修改积分", businessType = BusinessType.UPDATE)
    @Transactional(rollbackFor = Exception.class, propagation = Propagation.REQUIRES_NEW)
    public R batchUpdateGold(@RequestBody BatchUpdateIntegralDto dto) {
        Long userId = tokenService.getLoginUser().getUserId();
        List<Integer> userIds = dto.getUserIds();
        List<TAppUser> tAppUsers = appUserService.listByIds(userIds);
        List<TIntegralChangeRecord> tIntegralChangeRecords = new ArrayList<>();
        for (TAppUser tAppUser : tAppUsers) {
            TIntegralChangeRecord tIntegralChangeRecord = new TIntegralChangeRecord();
            tIntegralChangeRecord.setAppUserId(tAppUser.getId());
            tIntegralChangeRecord.setOldBalance(new BigDecimal(tAppUser.getAccumulativeIntegral()));
            tIntegralChangeRecord.setPrice(new BigDecimal(dto.getIntegral()));
            tIntegralChangeRecord.setCreateUserId(userId);
            tIntegralChangeRecords.add(tIntegralChangeRecord);
            tAppUser.setAccumulativeIntegral(dto.getIntegral());
        }
        integralChangeRecordService.saveBatch(tIntegralChangeRecords);
        appUserService.updateBatchById(tAppUsers);
        return R.ok();
    }
}