修改

src/main/java/org/rcisoft/business/cmspriceclick/controller/CmsPriceClickController.java

@@ -3,16 +3,23 @@ package org.rcisoft.business.cmspriceclick.controller;
 
 /*固定导入*/
 
+import cn.afterturn.easypoi.excel.entity.ExportParams;
+import cn.afterturn.easypoi.excel.entity.enmus.ExcelType;
+import com.alipay.api.internal.util.file.IOUtils;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import io.swagger.annotations.ApiImplicitParam;
 import io.swagger.annotations.ApiImplicitParams;
 import io.swagger.annotations.ApiOperation;
+import org.apache.poi.ss.usermodel.Workbook;
+import org.apache.poi.ss.usermodel.WorkbookFactory;
 import org.rcisoft.business.cmspriceclick.entity.CmsPriceClick;
 import org.rcisoft.business.cmspriceclick.entity.CmsPriceClickDTO;
 import org.rcisoft.business.cmspriceclick.service.CmsPriceClickService;
+import org.rcisoft.common.component.Global;
 import org.rcisoft.core.anno.CyOpeLogAnno;
 import org.rcisoft.core.constant.CyMessCons;
 import org.rcisoft.core.controller.CyPaginationController;
+import org.rcisoft.core.exception.CyServiceException;
 import org.rcisoft.core.model.CyPersistModel;
 import org.rcisoft.core.operlog.enums.CyLogTypeEnum;
 import org.rcisoft.core.result.CyResult;
@@ -25,16 +32,21 @@ import org.springframework.web.multipart.MultipartFile;
 
 import javax.servlet.http.HttpServletResponse;
 import javax.validation.Valid;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.net.URLEncoder;
 import java.util.List;
 
+
 /**
  * Created by cy on 2024年6月3日 下午3:35:01.
  */
 @RestController
 @RequestMapping("/cmspriceclick")
 public class CmsPriceClickController extends CyPaginationController<CmsPriceClick> {
-
-
+    private static final String PRICE_CLICK_IMPORT_TEMPLATE_NAME = "PromotionalTemplates.xlsx";
+    @Autowired
+    private Global global;
     @Autowired
     private CmsPriceClickService cmsPriceClickServiceImpl;
 
@@ -127,7 +139,7 @@ public class CmsPriceClickController extends CyPaginationController<CmsPriceClic
     @CyOpeLogAnno(title = "system-文章/视频单价点击量表管理-查询文章/视频单价点击量表", businessType = CyLogTypeEnum.EXPORT)
     @ApiOperation(value = "导出文章/视频单价点击量表信息", notes = "导出文章/视频单价点击量表信息")
     @GetMapping(value = "/export")
-    public void outCmsPriceClick(HttpServletResponse response, CmsPriceClick cmsPriceClick,Integer unitId) {
+    public void outCmsPriceClick(HttpServletResponse response, CmsPriceClick cmsPriceClick, Integer unitId) {
         String excelName = "推广次数信息.xlsx";
         cmsPriceClick.setUnitId(unitId);
         List<CmsPriceClickDTO> cmsPriceClickList = cmsPriceClickServiceImpl.export(cmsPriceClick);
@@ -192,11 +204,11 @@ public class CmsPriceClickController extends CyPaginationController<CmsPriceClic
 
     /**
      * 导入增加
-     * */
+     */
     @CyOpeLogAnno(title = "system-阅读量管理-导入增加", businessType = CyLogTypeEnum.IMPORT)
     @ApiOperation(value = "导入增加", notes = "导入增加")
     @PostMapping(value = "/importAdd")
-    public CyResult importExcelAdd(MultipartFile file,CmsPriceClick cmsPriceClick) {
+    public CyResult importExcelAdd(MultipartFile file, CmsPriceClick cmsPriceClick) {
         //校验文件是否为空
         if (file.isEmpty()) {
             return CyResultGenUtil.builder(new CyPersistModel(0), "导入失败，文件为空",
@@ -208,7 +220,7 @@ public class CmsPriceClickController extends CyPaginationController<CmsPriceClic
             return CyResultGenUtil.builder(new CyPersistModel(0), "导入失败，文件格式不正确",
                     "导入失败，文件格式不正确", null);
         }
-        cmsPriceClickServiceImpl.importExcelAdd(file,cmsPriceClick);
+        cmsPriceClickServiceImpl.importExcelAdd(file, cmsPriceClick);
         return CyResultGenUtil.builder(new CyPersistModel(1),
                 CyMessCons.MESSAGE_ALERT_SUCCESS,
                 CyMessCons.MESSAGE_ALERT_ERROR,
@@ -216,6 +228,36 @@ public class CmsPriceClickController extends CyPaginationController<CmsPriceClic
 
     }
 
+    /**
+     * 推广表导出模板
+     */
+    @CyOpeLogAnno(title = "system-阅读量管理-导出阅读量管理模版", businessType = CyLogTypeEnum.EXPORT)
+    @ApiOperation(value = "导出阅读量管理模版", notes = "导出阅读量管理模版")
+    @GetMapping(value = "/exportTemplate")
+    public void exportTemplate(HttpServletResponse response) {
+        OutputStream os = null;
+        Workbook workbook = null;
+        try {
+            response.setCharacterEncoding("UTF-8");
+            response.setHeader("content-Type", "application/vnd.ms-excel");
+            response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode("推广导入模板.xlsx", "UTF-8"));
+            ExportParams exportParams = new ExportParams("推广导入模板", "推广导入模板");
+            exportParams.setType(ExcelType.XSSF);
+            InputStream resourceAsStream = getClass().getClassLoader().getResourceAsStream(global.getEXCEL_TEMPLATE_LOCATION() + PRICE_CLICK_IMPORT_TEMPLATE_NAME);
+            //workbook = ExcelExportUtil.exportExcel(exportParams,SysUser.class, new ArrayList<>());
+//            String s = global.getEXCEL_TEMPLATE_LOCATION() + STUDENT_IMPORT_TEMPLATE_NAME;
+            workbook = WorkbookFactory.create(resourceAsStream);
+            os = response.getOutputStream();
+            workbook.write(os);
+        } catch (Exception e) {
+            System.out.println("------------------------打印异常日志--------------------------------");
+            System.out.println(e);
+            throw new CyServiceException("下载模版失败");
+        } finally {
+            IOUtils.closeQuietly(workbook);
+            IOUtils.closeQuietly(os);
+        }
+    }
 
 
 }