xuhy
2025-11-12 4f78db00a0aa292655e216c507589df8dd65d7b7
ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TCourseOrderController.java
@@ -1,13 +1,20 @@
package com.ruoyi.web.controller.api;
import cn.afterturn.easypoi.excel.ExcelExportUtil;
import cn.afterturn.easypoi.excel.entity.ExportParams;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.ruoyi.common.annotation.Log;
import com.ruoyi.common.basic.PageInfo;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.common.core.domain.entity.SysUser;
import com.ruoyi.common.enums.BusinessType;
import com.ruoyi.common.utils.DateUtils;
import com.ruoyi.common.utils.WebUtils;
import com.ruoyi.framework.web.service.TokenService;
import com.ruoyi.system.dto.AuditCourseOrderRefundDTO;
import com.ruoyi.system.export.TCourseOrderExport;
import com.ruoyi.system.export.TRefundOrderExport;
import com.ruoyi.system.model.TCourse;
import com.ruoyi.system.model.TCourseOrder;
import com.ruoyi.system.query.TCourseOrderQuery;
@@ -18,12 +25,18 @@
import com.ruoyi.system.vo.TCourseVO;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.apache.poi.ss.usermodel.Workbook;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.List;
import java.util.Objects;
/**
 * <p>
@@ -137,5 +150,77 @@
        return R.ok(courseService.removeByIds(ids));
    }
    @Log(title = "课程订单管理信息-课程报名记录导出", businessType = BusinessType.EXPORT)
    @ApiOperation(value = "课程报名记录导出")
    @PostMapping("/exportCourseOrder")
    public void exportCourseOrder(@RequestBody TCourseOrderQuery query){
        List<TCourseOrderExport> courseOrderExports = courseOrderService.exportCourseOrder(query);
        for (TCourseOrderExport courseOrderExport : courseOrderExports) {
            if(Objects.nonNull(courseOrderExport.getPayTime())){
                courseOrderExport.setPayTimeStr(DateUtils.localDateTimeToString(courseOrderExport.getPayTime()));
            }
        }
        Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), TCourseOrderExport.class, courseOrderExports);
        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();
            }
        }
    }
    @Log(title = "课程订单管理信息-退款记录导出", businessType = BusinessType.EXPORT)
    @ApiOperation(value = "退款记录导出")
    @PostMapping("/exportRefundOrder")
    public void exportRefundOrder(@RequestBody TCourseOrderQuery query){
        List<TRefundOrderExport> refundOrderExports = courseOrderService.exportRefundOrder(query);
        for (TRefundOrderExport courseOrderExport : refundOrderExports) {
            if(Objects.nonNull(courseOrderExport.getPayTime())){
                courseOrderExport.setPayTimeStr(DateUtils.localDateTimeToString(courseOrderExport.getPayTime()));
            }
            if(Objects.nonNull(courseOrderExport.getRequestRefundTime())){
                courseOrderExport.setRequestRefundTimeStr(DateUtils.localDateTimeToString(courseOrderExport.getRequestRefundTime()));
            }
        }
        Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), TRefundOrderExport.class, refundOrderExports);
        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();
            }
        }
    }
}