package org.jeecg.modules.tms.controller;
|
|
import java.io.File;
|
import java.io.InputStream;
|
import java.util.ArrayList;
|
import java.util.Arrays;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
import java.io.IOException;
|
import java.io.UnsupportedEncodingException;
|
import java.net.URLDecoder;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
|
import cn.hutool.core.collection.CollectionUtil;
|
import com.alibaba.fastjson.JSONObject;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import org.apache.poi.ss.usermodel.*;
|
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
|
import org.jeecg.common.api.vo.Result;
|
import org.jeecg.common.constant.CommonConstant;
|
import org.jeecg.common.system.query.QueryGenerator;
|
import org.jeecg.common.util.PmsUtil;
|
import org.jeecg.modules.tms.entity.BaseTools;
|
import org.jeecg.modules.tms.entity.ParaCommonTool;
|
import org.jeecg.modules.tms.entity.ToolsClassify;
|
import org.jeecg.modules.tms.service.IBaseToolsService;
|
import org.jeecg.modules.tms.service.IParaCommonToolService;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import lombok.extern.slf4j.Slf4j;
|
import org.jeecg.modules.tms.service.IToolsClassifyService;
|
import org.jeecg.modules.tms.utils.ExcelUtils;
|
import org.jeecgframework.poi.excel.ExcelImportUtil;
|
import org.jeecgframework.poi.excel.def.NormalExcelConstants;
|
import org.jeecgframework.poi.excel.entity.ExportParams;
|
import org.jeecgframework.poi.excel.entity.ImportParams;
|
import org.jeecgframework.poi.excel.view.JeecgEntityExcelView;
|
import org.jeecg.common.system.base.controller.JeecgController;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
import org.springframework.web.multipart.MultipartHttpServletRequest;
|
import org.springframework.web.servlet.ModelAndView;
|
import com.alibaba.fastjson.JSON;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import org.jeecg.common.aspect.annotation.AutoLog;
|
|
/**
|
* @Description: tms_para_common_tool
|
* @Author: jeecg-boot
|
* @Date: 2025-05-12
|
* @Version: V1.0
|
*/
|
@Api(tags="tms_para_common_tool")
|
@RestController
|
@RequestMapping("/tms/paraCommonTool")
|
@Slf4j
|
public class ParaCommonToolController extends JeecgController<ParaCommonTool, IParaCommonToolService> {
|
@Autowired
|
private IParaCommonToolService paraCommonToolService;
|
@Autowired
|
private IToolsClassifyService toolsClassifyService;
|
@Autowired
|
private IBaseToolsService baseToolsService;
|
|
/**
|
* 分页列表查询
|
*
|
* @param paraCommonTool
|
* @param pageNo
|
* @param pageSize
|
* @param req
|
* @return
|
*/
|
//@AutoLog(value = "tms_para_common_tool-分页列表查询")
|
@ApiOperation(value="tms_para_common_tool-分页列表查询", notes="tms_para_common_tool-分页列表查询")
|
@GetMapping(value = "/list")
|
public Result<IPage<ParaCommonTool>> queryPageList(ParaCommonTool paraCommonTool,
|
@RequestParam(name="pageNo", defaultValue="1") Integer pageNo,
|
@RequestParam(name="pageSize", defaultValue="10") Integer pageSize,
|
HttpServletRequest req) {
|
QueryWrapper<ParaCommonTool> queryWrapper = QueryGenerator.initQueryWrapper(paraCommonTool, req.getParameterMap());
|
Page<ParaCommonTool> page = new Page<ParaCommonTool>(pageNo, pageSize);
|
IPage<ParaCommonTool> pageList = paraCommonToolService.page(page, queryWrapper);
|
return Result.OK(pageList);
|
}
|
|
/**
|
* 添加
|
*
|
* @param paraCommonTool
|
* @return
|
*/
|
@AutoLog(value = "tms_para_common_tool-添加")
|
@ApiOperation(value="tms_para_common_tool-添加", notes="tms_para_common_tool-添加")
|
//@RequiresPermissions("org.jeecg.modules:tms_para_common_tool:add")
|
@PostMapping(value = "/add")
|
public Result<String> add(@RequestBody ParaCommonTool paraCommonTool) {
|
paraCommonToolService.save(paraCommonTool);
|
return Result.OK("添加成功！");
|
}
|
|
/**
|
* 编辑
|
*
|
* @param paraCommonTool
|
* @return
|
*/
|
@AutoLog(value = "tms_para_common_tool-编辑")
|
@ApiOperation(value="tms_para_common_tool-编辑", notes="tms_para_common_tool-编辑")
|
//@RequiresPermissions("org.jeecg.modules:tms_para_common_tool:edit")
|
@RequestMapping(value = "/edit", method = {RequestMethod.PUT,RequestMethod.POST})
|
public Result<String> edit(@RequestBody ParaCommonTool paraCommonTool) {
|
paraCommonToolService.updateById(paraCommonTool);
|
return Result.OK("编辑成功!");
|
}
|
|
/**
|
* 通过id删除
|
*
|
* @param id
|
* @return
|
*/
|
@AutoLog(value = "tms_para_common_tool-通过id删除")
|
@ApiOperation(value="tms_para_common_tool-通过id删除", notes="tms_para_common_tool-通过id删除")
|
//@RequiresPermissions("org.jeecg.modules:tms_para_common_tool:delete")
|
@DeleteMapping(value = "/delete")
|
public Result<String> delete(@RequestParam(name="id",required=true) String id) {
|
paraCommonToolService.removeById(id);
|
return Result.OK("删除成功!");
|
}
|
|
/**
|
* 批量删除
|
*
|
* @param ids
|
* @return
|
*/
|
@AutoLog(value = "tms_para_common_tool-批量删除")
|
@ApiOperation(value="tms_para_common_tool-批量删除", notes="tms_para_common_tool-批量删除")
|
//@RequiresPermissions("org.jeecg.modules:tms_para_common_tool:deleteBatch")
|
@DeleteMapping(value = "/deleteBatch")
|
public Result<String> deleteBatch(@RequestParam(name="ids",required=true) String ids) {
|
this.paraCommonToolService.removeByIds(Arrays.asList(ids.split(",")));
|
return Result.OK("批量删除成功!");
|
}
|
|
/**
|
* 通过id查询
|
*
|
* @param id
|
* @return
|
*/
|
//@AutoLog(value = "tms_para_common_tool-通过id查询")
|
@ApiOperation(value="tms_para_common_tool-通过id查询", notes="tms_para_common_tool-通过id查询")
|
@GetMapping(value = "/queryById")
|
public Result<ParaCommonTool> queryById(@RequestParam(name="id",required=true) String id) {
|
ParaCommonTool paraCommonTool = paraCommonToolService.getById(id);
|
if(paraCommonTool==null) {
|
return Result.error("未找到对应数据");
|
}
|
return Result.OK(paraCommonTool);
|
}
|
|
/**
|
* 导出excel
|
*
|
* @param request
|
* @param paraCommonTool
|
*/
|
//@RequiresPermissions("org.jeecg.modules:tms_para_common_tool:exportXls")
|
@RequestMapping(value = "/exportXls")
|
public ModelAndView exportXls(HttpServletRequest request, ParaCommonTool paraCommonTool) {
|
return super.exportXls(request, paraCommonTool, ParaCommonTool.class, "tms_para_common_tool");
|
}
|
|
/**
|
* 通过excel导入数据
|
*
|
* @param request
|
* @param response
|
* @return
|
*/
|
//@RequiresPermissions("tms_para_common_tool:importExcel")
|
@RequestMapping(value = "/importExcel", method = RequestMethod.POST)
|
public Result<?> importExcel(HttpServletRequest request, HttpServletResponse response) {
|
MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
|
Map<String, MultipartFile> fileMap = multipartRequest.getFileMap();
|
// 错误信息
|
List<String> errorMessage = new ArrayList<>();
|
int successLines = 0, errorLines = 0, dataLines = 0;
|
List<ParaCommonTool> paraCommonToolImportList = new ArrayList<>();
|
try {
|
for (Map.Entry<String, MultipartFile> entity : fileMap.entrySet()) {
|
// 获取上传文件对象
|
MultipartFile file = entity.getValue();
|
InputStream inputStream = file.getInputStream();
|
Workbook workbook = new XSSFWorkbook(inputStream);
|
Sheet sheet = workbook.getSheetAt(0);
|
for (Row row : sheet) {
|
ParaCommonTool paraCommonTool = new ParaCommonTool();
|
int rowNum = row.getRowNum();
|
int currentRowNum = rowNum + 1;
|
//表头行和空行略过
|
if (rowNum == 0 || ExcelUtils.isRowEmpty(row)) {
|
continue;
|
}
|
dataLines++;
|
Cell cell1 = row.getCell(1);
|
if (cell1 == null || cell1.getCellType() == CellType.BLANK) {
|
errorMessage.add("第" + currentRowNum + "行，数据导入失败,一级分类名称为空! ");
|
errorLines++;
|
continue;
|
}
|
Cell cell2 = row.getCell(2);
|
if (cell2 == null || cell2.getCellType() == CellType.BLANK) {
|
errorMessage.add("第" + currentRowNum + "行，数据导入失败,二级分类名称为空! ");
|
errorLines++;
|
continue;
|
}
|
Cell cell3 = row.getCell(3);
|
if (cell3 == null || cell3.getCellType() == CellType.BLANK) {
|
errorMessage.add("第" + currentRowNum + "行，数据导入失败,三级分类名称为空! ");
|
errorLines++;
|
continue;
|
}
|
Cell cell4 = row.getCell(4);
|
if (cell4 == null || cell4.getCellType() == CellType.BLANK) {
|
errorMessage.add("第" + currentRowNum + "行，数据导入失败,四级分类名称为空! ");
|
errorLines++;
|
continue;
|
}
|
Cell cell9 = row.getCell(9);
|
if (cell9 == null || cell9.getCellType() == CellType.BLANK) {
|
errorMessage.add("第" + currentRowNum + "行，数据导入失败,型号/图号为空! ");
|
errorLines++;
|
continue;
|
}
|
//根据四级分类名称和型号/图号查询工具信息
|
List<ToolsClassify> toolClassifyList = toolsClassifyService.list(new LambdaQueryWrapper<ToolsClassify>()
|
.eq(ToolsClassify::getTypeName, ExcelUtils.getCellValueAsString(cell4))
|
.eq(ToolsClassify::getStatus, CommonConstant.STATUS_1));
|
if (CollectionUtil.isEmpty(toolClassifyList)) {
|
errorMessage.add("第" + currentRowNum + "行，数据导入失败,系统不存在该四级分类名称! ");
|
errorLines++;
|
continue;
|
} else {
|
ToolsClassify toolsClassify = toolClassifyList.get(0);
|
List<BaseTools> baseToolsList = baseToolsService.list(new LambdaQueryWrapper<BaseTools>()
|
.eq(BaseTools::getClassifyId, toolsClassify.getId())
|
.eq(BaseTools::getToolModel, ExcelUtils.getCellValueAsString(cell9)));
|
if (CollectionUtil.isEmpty(baseToolsList)) {
|
errorMessage.add("第" + currentRowNum + "行，数据导入失败,系统不存在对应的工具信息! ");
|
errorLines++;
|
continue;
|
} else {
|
BaseTools baseTools = baseToolsList.get(0);
|
List<ParaCommonTool> paraCommonToolList = paraCommonToolService.list(new LambdaQueryWrapper<ParaCommonTool>()
|
.eq(ParaCommonTool::getToolCode, baseTools.getId())
|
.eq(ParaCommonTool::getToolModel, ExcelUtils.getCellValueAsString(cell9)));
|
if (CollectionUtil.isEmpty(paraCommonToolList)) {
|
paraCommonTool.setClassifyId(toolsClassify.getId());
|
paraCommonTool.setToolCode(baseTools.getId());
|
paraCommonTool.setToolModel(ExcelUtils.getCellValueAsString(cell9));
|
} else {
|
paraCommonTool = paraCommonToolList.get(0);
|
}
|
}
|
}
|
|
Cell cell0 = row.getCell(0);//序号(或企业内原代码)
|
paraCommonTool.setSignCode(ExcelUtils.getCellValueAsString(cell0));
|
Cell cell5 = row.getCell(5);//物品中文名称
|
if (cell5 == null || cell5.getCellType() == CellType.BLANK) {
|
errorMessage.add("第" + currentRowNum + "行，数据导入失败,物品中文名称为空! ");
|
errorLines++;
|
continue;
|
} else {
|
paraCommonTool.setChineseName(ExcelUtils.getCellValueAsString(cell5));
|
}
|
Cell cell6 = row.getCell(6);//物品外文名称
|
paraCommonTool.setForeignLanguageName(ExcelUtils.getCellValueAsString(cell6));
|
Cell cell7 = row.getCell(7);//标准级别
|
paraCommonTool.setStandardLevel(ExcelUtils.getCellValueAsString(cell7));
|
Cell cell8 = row.getCell(8);//标准号
|
paraCommonTool.setStandardCode(ExcelUtils.getCellValueAsString(cell8));
|
|
paraCommonToolImportList.add(paraCommonTool);
|
successLines++;
|
}
|
}
|
paraCommonToolService.saveOrUpdateBatch(paraCommonToolImportList);
|
if (successLines == dataLines) {
|
return Result.OK("共" + dataLines + "行数据全部导入成功！");
|
} else {
|
JSONObject result = new JSONObject(5);
|
result.put("successCount", successLines);
|
result.put("msg", "总上传行数：" + dataLines + "，成功导入行数：" + successLines);
|
result.put("errorMsg", errorMessage);
|
String fileUrl = PmsUtil.saveErrorMsgByList(errorMessage, "userImportExcelErrorLog");
|
int lastIndex = fileUrl.lastIndexOf(File.separator);
|
String fileName = fileUrl.substring(lastIndex + 1);
|
result.put("fileUrl", "/sys/common/static/" + fileUrl);
|
result.put("fileName", fileName);
|
Result res = Result.ok(result);
|
res.setCode(201);
|
res.setMessage("文件导入成功，但有错误。");
|
return res;
|
}
|
} catch (IOException e) {
|
throw new RuntimeException(e);
|
}
|
}
|
|
}
|
