帷幄君成-富创-后端代码
blame | 历史 | 补丁 | 提交 | 提交对比 | 忽略空白
zhanghl
2025-05-23 6da289fff756eaecc600422d77c4afff1ddebab0 
 aps-modules/aps-core/src/main/java/com/aps/core/controller/mainPlan/ApsGasPipelinePredictionController.java


@@ -7,23 +7,28 @@


import com.aps.common.log.annotation.Log;


import com.aps.common.log.enums.BusinessType;


import com.aps.common.security.annotation.RequiresPermissions;


import com.aps.common.security.utils.DictUtils;


import com.aps.core.domain.ApsGasPipelinePrediction;


import com.aps.core.domain.ApsMaterialProductGroupManagement;


import com.aps.core.service.IApsGasPipelinePredictionService;


import com.aps.system.api.domain.SysDictData;


import io.swagger.v3.oas.annotations.Operation;


import io.swagger.v3.oas.annotations.tags.Tag;


import jakarta.servlet.http.HttpServletResponse;


import lombok.SneakyThrows;


import org.apache.commons.io.IOUtils;


import org.apache.poi.ss.usermodel.Row;


import org.apache.poi.ss.usermodel.Sheet;


import org.apache.poi.ss.usermodel.Workbook;


import org.apache.poi.ss.usermodel.WorkbookFactory;


import org.springframework.beans.factory.annotation.Autowired;


import org.springframework.core.io.ByteArrayResource;


import org.springframework.http.HttpHeaders;


import org.springframework.http.MediaType;


import org.springframework.http.ResponseEntity;


import org.springframework.web.bind.annotation.*;


import org.springframework.web.multipart.MultipartFile;




import java.io.InputStream;


import java.io.ByteArrayInputStream;


import java.io.ByteArrayOutputStream;


import java.net.URLEncoder;


import java.nio.charset.StandardCharsets;


import java.util.List;


@@ -57,7 +62,7 @@




    @Operation(summary = "导入手工气体预测数据", description = "批量导入")


    @Log(title = "导入手工气体预测数据", businessType = BusinessType.IMPORT)


    @RequiresPermissions("gasPipeline:prediction:import")


//    @RequiresPermissions("gasPipeline:prediction:import")


    @PostMapping("/importData")


    public AjaxResult importData(MultipartFile file) throws Exception {




@@ -75,6 +80,20 @@


    @GetMapping("/template")


    public ResponseEntity<ByteArrayResource> exportTemplate() {


        byte[] file = IOUtils.resourceToByteArray("/templates/气体预测数据模板v1.0.xlsx");


        Workbook workbook = WorkbookFactory.create(new ByteArrayInputStream(file));


        Sheet sheet = workbook.getSheet("字典-生产基地");


        List<SysDictData> sysDictDataList = DictUtils.getDictCache("aps_factory");


        if (sysDictDataList != null) {


            for (int i = 0; i < sysDictDataList.size(); i++) {


                Row row = sheet.createRow(i + 1);


                row.createCell(0).setCellValue(sysDictDataList.get(i).getDictValue());


                row.createCell(1).setCellValue(sysDictDataList.get(i).getDictLabel());


            }


            ByteArrayOutputStream baos = new ByteArrayOutputStream();


            workbook.write(baos);


            file = baos.toByteArray();


        }


        workbook.close();


        ByteArrayResource resource = new ByteArrayResource(file);


        return ResponseEntity.ok()


                .header("Access-Control-Expose-Headers", HttpHeaders.CONTENT_DISPOSITION)