liujie
2026-07-22 fdd6c6769ee253cd20392383aa52a3d73f933619
ruoyi-service/ruoyi-order/src/main/java/com/ruoyi/order/controller/TVipOrderController.java
@@ -6,6 +6,7 @@
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.ruoyi.account.api.dto.GiveVipDto;
import com.ruoyi.account.api.dto.UserListQueryDto;
import com.ruoyi.account.api.feignClient.AppUserClient;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.common.core.utils.DateUtils;
@@ -19,17 +20,24 @@
import com.ruoyi.order.api.model.TExchangeOrder;
import com.ruoyi.order.api.model.TShoppingOrder;
import com.ruoyi.order.api.model.TVipOrder;
import com.ruoyi.order.api.model.TVipOrderRefund;
import com.ruoyi.order.api.query.ChargingOrderQuery;
import com.ruoyi.order.api.query.ShoppingOrderQuery;
import com.ruoyi.order.api.query.VipShoppingOrderQuery;
import com.ruoyi.order.api.vo.ChargingOrderVO;
import com.ruoyi.order.dto.TEnterpriseExport;
import com.ruoyi.order.export.EnterpriseChargingDetailExport;
import com.ruoyi.order.export.TVipOrderExport;
import com.ruoyi.order.service.TChargingOrderService;
import com.ruoyi.order.service.TVipOrderRefundService;
import com.ruoyi.order.service.TVipOrderService;
import com.ruoyi.other.api.domain.TEnterpriseUserApplication;
import com.ruoyi.other.api.feignClient.OtherClient;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.apache.poi.ss.usermodel.Workbook;
import org.omg.CORBA.PRIVATE_MEMBER;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.*;
@@ -38,6 +46,8 @@
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
@@ -61,6 +71,10 @@
    private AppUserClient appUserClient;
    @Resource
    private OtherClient otherClient;
    @Resource
    private TVipOrderRefundService vipOrderRefundService;
    @Resource
    private TChargingOrderService chargingOrderService;
    
    
    
@@ -73,6 +87,7 @@
            if (data.isEmpty()){
                return AjaxResult.success(new PageInfo<TVipOrder>());
            }
            query.setUserIds(data);
        }
        PageInfo<TVipOrder> res = vipOrderService.pageList(query);
        return AjaxResult.success(res);
@@ -126,15 +141,43 @@
    @PostMapping("/callBack")
    public R callBack(@RequestParam("code")String code,@RequestParam("outTradeNo")String outTradeNo){
        TVipOrder one = vipOrderService.lambdaQuery().eq(TVipOrder::getCode, code).one();
        one.setSerialNumber(outTradeNo);
        one.setPaymentStatus(2);
        vipOrderService.updateById(one);
        if (one == null) {
            return R.fail("会员订单不存在");
        }
        boolean updated = vipOrderService.update(new LambdaUpdateWrapper<TVipOrder>()
                .eq(TVipOrder::getId, one.getId())
                .eq(TVipOrder::getPaymentStatus, 1)
                .set(TVipOrder::getSerialNumber, outTradeNo)
                .set(TVipOrder::getPaymentStatus, 2)
                .set(TVipOrder::getPayTime, LocalDateTime.now()));
        if (!updated) {
            return R.ok();
        }
        GiveVipDto giveVipDto = new GiveVipDto();
        giveVipDto.setVipId(one.getVipId());
        giveVipDto.setType(one.getVipType());
        giveVipDto.setUserIds(one.getAppUserId().toString());
        appUserClient.giveVip(giveVipDto);
        return R.ok();
    }
    @PostMapping("/vipWxRefund")
    public void vipWxRefund(@RequestParam("out_refund_no") String out_refund_no,
                            @RequestParam("refund_id") String refund_id,
                            @RequestParam("tradeState") String tradeState,
                            @RequestParam("success_time") String success_time){
        if("SUCCESS".equals(tradeState)){
            TVipOrderRefund vipOrderRefund = vipOrderRefundService.getOne(new LambdaUpdateWrapper<TVipOrderRefund>().eq(TVipOrderRefund::getRefundCode, out_refund_no));
            vipOrderRefund.setRefundSerialNumber(refund_id);
            vipOrderRefund.setRefundStatus(2);
            vipOrderRefund.setRefundTime(LocalDateTime.now());
            vipOrderRefundService.updateById(vipOrderRefund);
            TVipOrder vipOrder = vipOrderService.getById(vipOrderRefund.getVipOrderId());
            vipOrder.setRefundStatus(2);
            vipOrderService.updateById(vipOrder);
        }
    }
@@ -181,7 +224,119 @@
    }
    @RequiresPermissions(value = {"/vipOrder", "/vipOrder/export"}, logical = Logical.OR)
    @ApiOperation(tags = {"管理后台-会员订单"}, value = "导出")
    @PutMapping(value = "/exportVipOrder")
    @Log(title = "会员订单导出", businessType = BusinessType.EXPORT)
    public void exportVipOrder(@RequestBody VipShoppingOrderQuery query) {
        if (StringUtils.hasLength(query.getPhone())) {
            List<Long> data = appUserClient.getUserIdsByPhone(query.getPhone()).getData();
            if (data.isEmpty()) {
                writeVipOrderWorkbook(new ArrayList<>());
                return;
            }
            query.setUserIds(data);
        }
        query.setPageCurr(1);
        query.setPageSize(100000);
        List<TVipOrder> list = vipOrderService.pageList(query).getRecords();
        List<TVipOrderExport> exports = new ArrayList<>();
        for (TVipOrder vipOrder : list) {
            TVipOrderExport export = new TVipOrderExport();
            BeanUtils.copyProperties(vipOrder, export);
            export.setCreateTime(DateUtils.localDateTimeToString(vipOrder.getCreateTime()));
            exports.add(export);
        }
        writeVipOrderWorkbook(exports);
    }
    @RequiresPermissions(value = {"/vipList", "/vipList/export"}, logical = Logical.OR)
    @ApiOperation(tags = {"管理后台-会员列表"}, value = "导出会员充电订单")
    @PutMapping(value = "/exportVipOrderByVipUser")
    @Log(title = "会员列表导出会员充电订单", businessType = BusinessType.EXPORT)
    public void exportVipOrderByVipUser(@RequestBody UserListQueryDto query) {
        List<Long> userIds = appUserClient.getVipUserIds(query).getData();
        if (userIds == null || userIds.isEmpty()) {
            writeVipChargingOrderWorkbook(new ArrayList<>());
            return;
        }
        ChargingOrderQuery orderQuery = new ChargingOrderQuery();
        orderQuery.setUserIds(userIds);
        orderQuery.setPageCurr(1);
        orderQuery.setPageSize(100000);
        List<ChargingOrderVO> list = chargingOrderService.chargingOrderList(orderQuery).getRecords();
        List<EnterpriseChargingDetailExport> exports = new ArrayList<>();
        for (ChargingOrderVO order : list) {
            EnterpriseChargingDetailExport export = new EnterpriseChargingDetailExport();
            export.setCode(order.getCode());
            export.setOrderType(Integer.valueOf(1).equals(order.getOrderType()) ? "小程序" : "刷卡");
            export.setOrderClassification("充电订单");
            export.setSiteName(order.getSiteName());
            export.setTerminalName(order.getTerminalName());
            export.setStatus(getChargingOrderStatus(order.getStatus()));
            export.setChargingSecond(order.getChargingSecond());
            export.setTotalAmount(order.getOrderAmount());
            export.setCouponDiscountAmount(order.getCouponDiscountAmount());
            export.setVipDiscountAmount(order.getVipDiscountAmount());
            export.setPaymentAmount(order.getPaymentAmount());
            export.setRefundAmount(order.getRefundAmount());
            export.setPhone(order.getPhone());
            export.setOrderSource(getChargingOrderSource(order.getOrderSource()));
            export.setStartTime(DateUtils.localDateTimeToString(order.getStartTime()));
            export.setEndTime(DateUtils.localDateTimeToString(order.getEndTime()));
            exports.add(export);
        }
        writeVipChargingOrderWorkbook(exports);
    }
    private String getChargingOrderStatus(Integer status) {
        if (status == null) return "未知";
        String[] names = {"未知", "等待中", "启动中", "充电中", "停止中", "已结束"};
        return status >= 0 && status < names.length ? names[status] : "未知";
    }
    private String getChargingOrderSource(Integer source) {
        if (Integer.valueOf(1).equals(source)) return "明星充电";
        if (Integer.valueOf(2).equals(source)) return "快电";
        if (Integer.valueOf(3).equals(source)) return "新电途";
        return "";
    }
    private void writeVipChargingOrderWorkbook(List<EnterpriseChargingDetailExport> exports) {
        Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), EnterpriseChargingDetailExport.class, exports);
        writeWorkbook(workbook, "会员充电订单导出.xls");
    }
    private void writeVipOrderWorkbook(List<TVipOrderExport> exports) {
        Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), TVipOrderExport.class, exports);
        writeWorkbook(workbook, "会员订单导出.xls");
    }
    private void writeWorkbook(Workbook workbook, String exportFileName) {
        HttpServletResponse response = WebUtils.response();
        response.setCharacterEncoding("utf-8");
        ServletOutputStream outputStream = null;
        try {
            String fileName = URLEncoder.encode(exportFileName, "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 {
                workbook.close();
                if (outputStream != null) {
                    outputStream.close();
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
    }