完善导出等接口

xuhy

2024-09-11 9ca97fd558700e7054c5f54192a9db7a1e6b8230

 ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TDataGeneratorController.java

@@ -14,9 +14,7 @@
import com.ruoyi.system.domain.TOrderStockGoods;
import com.ruoyi.system.dto.OrderMealGeneratorDTO;
import com.ruoyi.system.dto.OrderSaleGeneratorDTO;
import com.ruoyi.system.export.TOrderMealExportExcel;
import com.ruoyi.system.export.TOrderSaleAndGoodsExportExcel;
import com.ruoyi.system.export.TOrderStockExportExcel;
import com.ruoyi.system.export.*;
import com.ruoyi.system.query.TDataGeneratorMealQuery;
import com.ruoyi.system.query.TDataGeneratorQuery;
import com.ruoyi.system.query.TDataGeneratorSaleQuery;
@@ -214,7 +212,6 @@
        for (TOrderStockVO orderStockVO : list) {
            TOrderStockExportExcel orderStockExportExcel = new TOrderStockExportExcel();
            BeanUtils.copyProperties(orderStockVO, orderStockExportExcel);
            orderStockExportExcel.setGoodsAmount(orderStockVO.getOrderStockGoods().stream().map(TOrderStockGoods::getSalePrice).reduce(BigDecimal::add).get());
            orderStockExportExcel.setStockTime(DateUtils.localDateToString(orderStockVO.getStockTime()));
            orderStockExportExcels.add(orderStockExportExcel);
        }
@@ -224,7 +221,7 @@
        response.setCharacterEncoding("utf-8");
        ServletOutputStream outputStream = null;
        try {
            String fileName = URLEncoder.encode("销售生成列表.xls", "utf-8");
            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");
@@ -242,6 +239,78 @@
        }
    }

    @ApiOperation( value = "餐饮数据生成列表导出")
    @PostMapping(value = "/mealGeneratorListExport")
    public void mealGeneratorListExport(@RequestBody TDataGeneratorQuery query) {
        List<TDataGeneratorVO> list = orderMealService.mealGeneratorListExport(query);
        List<MealGeneratorListExport> mealGeneratorListExports = new ArrayList<>();
        for (TDataGeneratorVO dataGeneratorVO : list) {
            MealGeneratorListExport mealGeneratorListExport = new MealGeneratorListExport();
            BeanUtils.copyProperties(dataGeneratorVO, mealGeneratorListExport);
            mealGeneratorListExport.setGeneratorTime(DateUtils.localDateTimeToString(dataGeneratorVO.getCreateTime()));
            mealGeneratorListExport.setStartTimeStr(DateUtils.localDateToString(dataGeneratorVO.getStartTime()) + " ~ " + DateUtils.localDateToString(dataGeneratorVO.getEndTime()));
            mealGeneratorListExport.setRevenueRange(dataGeneratorVO.getMinMoney() + " ~ " + dataGeneratorVO.getMaxMoney());
            mealGeneratorListExports.add(mealGeneratorListExport);
        }
        Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), MealGeneratorListExport.class, mealGeneratorListExports);
        HttpServletResponse response = WebUtils.response();
        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 = "销售数据生成列表导出")
    @PostMapping(value = "/saleGeneratorListExport")
    public void saleGeneratorListExport(@RequestBody TDataGeneratorQuery query) {
        List<TDataGeneratorVO> list = orderSaleService.saleGeneratorListExport(query);
        List<SaleGeneratorListExport> saleGeneratorListExports = new ArrayList<>();
        for (TDataGeneratorVO dataGeneratorVO : list) {
            SaleGeneratorListExport saleGeneratorListExport = new SaleGeneratorListExport();
            BeanUtils.copyProperties(dataGeneratorVO, saleGeneratorListExport);
            saleGeneratorListExport.setGeneratorTime(DateUtils.localDateTimeToString(dataGeneratorVO.getCreateTime()));
            saleGeneratorListExport.setStartTimeStr(DateUtils.localDateToString(dataGeneratorVO.getStartTime()) + " ~ " + DateUtils.localDateToString(dataGeneratorVO.getEndTime()));
            saleGeneratorListExports.add(saleGeneratorListExport);
        }
        Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), SaleGeneratorListExport.class, saleGeneratorListExports);
        HttpServletResponse response = WebUtils.response();
        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();
            }
        }
    }

}