资质导出

dsk-module/dsk-biz-api/src/main/java/com/dsk/search/controller/ExportController.java

@@ -2,13 +2,19 @@ package com.dsk.search.controller;
 
 import com.alibaba.fastjson2.JSONObject;
 import com.dsk.common.core.domain.AjaxResult;
+import com.dsk.common.utils.poi.ExcelUtil;
 import com.dsk.jsk.domain.bo.ComposeQueryDto;
 import com.dsk.search.service.BusinessOpportunityRadarService;
 import com.dsk.search.service.ExportService;
+import com.dsk.system.domain.vo.AptitudeVo;
+import com.dsk.system.domain.vo.SysUserImportVo;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
 
 import javax.annotation.Resource;
+import javax.servlet.http.HttpServletResponse;
+import java.util.ArrayList;
+import java.util.List;
 
 
 /**
@@ -25,6 +31,18 @@ public class ExportController {
     @Resource
     private ExportService exportService;
 
+    /**
+     * 获取导入模板
+     */
+    @PostMapping("/aptitude/importTemplate")
+    public void importTemplate(HttpServletResponse response) {
+        List<Object> list = new ArrayList<>();
+        AptitudeVo aptitudeVo = new AptitudeVo();
+        aptitudeVo.setCompanyName("重庆市轨道交通（集团）有限公司");
+        list.add(aptitudeVo);
+        ExcelUtil.exportTemplate(list, "企业数据", "excel/userImportTemplate.xlsx", response);
+    }
+
     /*
      * 批量资质导出
      */

dsk-module/dsk-biz-api/src/main/java/com/dsk/search/service/ExportService.java

 package com.dsk.search.service;
 
 import cn.hutool.core.bean.BeanUtil;
+import cn.hutool.core.map.MapUtil;
 import com.alibaba.fastjson2.JSONObject;
+import com.dsk.common.constant.Constants;
 import com.dsk.common.core.domain.AjaxResult;
 import com.dsk.common.excel.ExcelUtils;
 import com.dsk.common.exception.ServiceException;
+import com.dsk.jsk.domain.bo.ComposeQueryDto;
 import com.dsk.jsk.service.EnterpriseService;
+import com.dsk.system.domain.SysUserFileRecord;
 import com.dsk.system.domain.vo.AptitudeVo;
+import com.dsk.system.service.ISysUserFileRecordService;
 import com.dsk.system.utils.DskOpenApiUtil;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.context.annotation.Bean;
+import org.springframework.stereotype.Component;
 import org.springframework.stereotype.Service;
 import org.springframework.web.multipart.MultipartFile;
 
@@ -28,8 +36,13 @@ public class ExportService {
 
     @Autowired
     private DskOpenApiUtil dskOpenApiUtil;
+    @Autowired
+    private ISysUserFileRecordService sysUserFileRecordService;
+    @Value("dsk.exportBackUrl")
+    private String EXPORT_BACK_URL;
 
     public AjaxResult aptitudeFile(MultipartFile file) throws Exception {
+        ComposeQueryDto composeQueryDto = new ComposeQueryDto();
         //识别Excel内容
         List<AptitudeVo> companyNameList = new ExcelUtils<>(AptitudeVo.class).importExcel(file.getInputStream(), 2);
         if (companyNameList.isEmpty()) {
@@ -42,9 +55,19 @@ public class ExportService {
             }
             list.add(companyNameList.get(i).getCompanyName());
         }
-        Map<String, Object> params = new HashMap<>();
-        params.put("keyword", String.join(",",list));
-        dskOpenApiUtil.requestBody("/nationzj/marketAnalysis/areaGroupByProvince", params);
+        composeQueryDto.setKeyword(String.join(",",list));
+        SysUserFileRecord fileRecord = new SysUserFileRecord();
+        fileRecord.setFileName("批量导出".concat(Constants.SUFFIX_XLSX));
+        fileRecord.setType(1);
+        //生成中
+        fileRecord.setStatus(2);
+        fileRecord.setRemark("集团招标业绩导出");
+        Long fileId = sysUserFileRecordService.exportAdd(fileRecord);
+        composeQueryDto.setExportUniqueCode(fileId.toString());
+        //回调函数
+        composeQueryDto.setExportBackUrl(EXPORT_BACK_URL);
+
+        dskOpenApiUtil.requestBody("/nationzj/marketAnalysis/areaGroupByProvince", new JSONObject());
         return AjaxResult.success();
     }