diff --git a/nladmin-system/nlsso-server/src/main/java/org/nl/wms/basedata_manage/controller/StructIvtController.java b/nladmin-system/nlsso-server/src/main/java/org/nl/wms/basedata_manage/controller/StructIvtController.java index 998d67a..5496874 100644 --- a/nladmin-system/nlsso-server/src/main/java/org/nl/wms/basedata_manage/controller/StructIvtController.java +++ b/nladmin-system/nlsso-server/src/main/java/org/nl/wms/basedata_manage/controller/StructIvtController.java @@ -1,21 +1,29 @@ package org.nl.wms.basedata_manage.controller; +import com.alibaba.fastjson.JSONObject; +import com.alibaba.fastjson.TypeReference; +import com.baomidou.mybatisplus.core.metadata.IPage; import lombok.RequiredArgsConstructor; import lombok.extern.slf4j.Slf4j; import org.nl.common.base.TableDataInfo; import org.nl.common.domain.query.PageQuery; import org.nl.common.logging.annotation.Log; +import org.nl.common.utils.MapOf; import org.nl.wms.basedata_manage.service.IMdPbStoragevehicleextService; +import org.nl.wms.system_manage.service.columnInfo.ColumnInfoService; +import org.springframework.beans.factory.annotation.Autowired; import org.springframework.http.HttpStatus; import org.springframework.http.ResponseEntity; -import org.springframework.web.bind.annotation.GetMapping; -import org.springframework.web.bind.annotation.RequestMapping; -import org.springframework.web.bind.annotation.RequestParam; -import org.springframework.web.bind.annotation.RestController; +import org.springframework.web.bind.annotation.*; import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.IOException; +import java.util.HashMap; +import java.util.List; import java.util.Map; +import java.util.stream.Collectors; /** @@ -35,9 +43,40 @@ public class StructIvtController { @Resource private final IMdPbStoragevehicleextService iMdPbStoragevehicleextService; + @Autowired + private ColumnInfoService columnInfoService; + @GetMapping @Log("查询库存") public ResponseEntity query(@RequestParam Map whereJson, PageQuery page) { return new ResponseEntity<>(TableDataInfo.build(iMdPbStoragevehicleextService.queryAll(whereJson, page)), HttpStatus.OK); } + + @PostMapping("/exportFile") + public void exportFile(@RequestBody Map whereJson, PageQuery page, HttpServletResponse response) { + page.setPage(0); + page.setSize(99999); + IPage pmFormDataDtoIPage = iMdPbStoragevehicleextService.queryAll(whereJson, page); + List pmFormDataDtos = pmFormDataDtoIPage.getRecords(); + + List maps = pmFormDataDtos.stream().map(jsonObject -> { + try { + return jsonObject.toJavaObject(new TypeReference>() {}); + } catch (Exception e) { + e.printStackTrace(); + return new HashMap(); + } + }).collect(Collectors.toList()); + try { + columnInfoService.exportFile("md_pb_groupplate", maps, response, + null, MapOf.of( + "material_code","物料编号","material_name","物料名称","sect_name","库区名称" + ,"stor_name","仓库名称","struct_name","仓位名称","struct_code","仓位编码" + ) + ); + } catch (IOException e) { + log.info("EXCEL 导出异常,异常原因=【{}】",e.getMessage()); + e.printStackTrace(); + } + } } diff --git a/nladmin-system/nlsso-server/src/main/java/org/nl/wms/basedata_manage/service/dao/mapper/StructattrMapper.xml b/nladmin-system/nlsso-server/src/main/java/org/nl/wms/basedata_manage/service/dao/mapper/StructattrMapper.xml index dddd710..4399aec 100644 --- a/nladmin-system/nlsso-server/src/main/java/org/nl/wms/basedata_manage/service/dao/mapper/StructattrMapper.xml +++ b/nladmin-system/nlsso-server/src/main/java/org/nl/wms/basedata_manage/service/dao/mapper/StructattrMapper.xml @@ -112,7 +112,7 @@