feat: 供应商导入数据、导出模板功能
This commit is contained in:
@@ -10,10 +10,13 @@ import org.nl.wms.basedata_manage.service.IMdCsSupplierbaseService;
|
||||
import org.nl.wms.basedata_manage.service.dao.MdCsSupplierbase;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.http.HttpStatus;
|
||||
import org.springframework.http.MediaType;
|
||||
import org.springframework.http.ResponseEntity;
|
||||
import org.springframework.validation.annotation.Validated;
|
||||
import org.springframework.web.bind.annotation.*;
|
||||
import org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
import javax.servlet.http.HttpServletResponse;
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
|
||||
@@ -61,4 +64,17 @@ public class SupplierController {
|
||||
return new ResponseEntity<>(HttpStatus.OK);
|
||||
}
|
||||
|
||||
@PostMapping("/importExcel")
|
||||
@Log("导入单位")
|
||||
public ResponseEntity<Object> importExcel(@RequestPart("file") MultipartFile file) {
|
||||
iMdCsSupplierbaseService.importExcel(file);
|
||||
return new ResponseEntity<>(HttpStatus.NO_CONTENT);
|
||||
}
|
||||
|
||||
@Log("下载导入模板")
|
||||
@GetMapping(value = "/downloadImportSuppTemplate", produces = MediaType.APPLICATION_OCTET_STREAM_VALUE)
|
||||
public void downloadImportSuppTemplate(HttpServletResponse response) {
|
||||
iMdCsSupplierbaseService.downloadImportSuppTemplate(response);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@@ -4,7 +4,9 @@ import com.baomidou.mybatisplus.core.metadata.IPage;
|
||||
import com.baomidou.mybatisplus.extension.service.IService;
|
||||
import org.nl.common.domain.query.PageQuery;
|
||||
import org.nl.wms.basedata_manage.service.dao.MdCsSupplierbase;
|
||||
import org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
import javax.servlet.http.HttpServletResponse;
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
|
||||
@@ -43,4 +45,10 @@ public interface IMdCsSupplierbaseService extends IService<MdCsSupplierbase> {
|
||||
* @param ids 供应商标识集合
|
||||
*/
|
||||
void delete(Set<String> ids);
|
||||
|
||||
void importExcel(MultipartFile file);
|
||||
|
||||
void downloadImportSuppTemplate(HttpServletResponse response);
|
||||
|
||||
MdCsSupplierbase getByCode(String code);
|
||||
}
|
||||
|
||||
@@ -8,17 +8,27 @@ import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
||||
import com.baomidou.mybatisplus.core.metadata.IPage;
|
||||
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
||||
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
||||
import org.apache.poi.ss.usermodel.*;
|
||||
import org.nl.common.domain.query.PageQuery;
|
||||
import org.nl.common.exception.BadRequestException;
|
||||
import org.nl.common.utils.FileUtil;
|
||||
import org.nl.common.utils.IdUtil;
|
||||
import org.nl.common.utils.SecurityUtils;
|
||||
import org.nl.wms.basedata_manage.enums.BaseDataEnum;
|
||||
import org.nl.wms.basedata_manage.service.IMdCsSupplierbaseService;
|
||||
import org.nl.wms.basedata_manage.service.dao.MdCsSupplierbase;
|
||||
import org.nl.wms.basedata_manage.service.dao.MdPbMeasureunit;
|
||||
import org.nl.wms.basedata_manage.service.dao.mapper.MdCsSupplierbaseMapper;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.springframework.transaction.annotation.Transactional;
|
||||
import org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
import javax.servlet.http.HttpServletResponse;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.math.BigDecimal;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
|
||||
@@ -93,4 +103,67 @@ public class MdCsSupplierbaseServiceImpl extends ServiceImpl<MdCsSupplierbaseMap
|
||||
public void delete(Set<String> ids) {
|
||||
this.baseMapper.deleteBatchIds(ids);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void importExcel(MultipartFile file) {
|
||||
// 1. 校验文件
|
||||
if (file.isEmpty()) {
|
||||
throw new IllegalArgumentException("上传文件不能为空");
|
||||
}
|
||||
|
||||
try (InputStream inputStream = file.getInputStream();
|
||||
Workbook workbook = WorkbookFactory.create(inputStream)) {
|
||||
DataFormatter formatter = new DataFormatter();
|
||||
// 2. 获取第一个Sheet
|
||||
Sheet sheet = workbook.getSheetAt(0);
|
||||
|
||||
// 3. 从第二行开始遍历 (索引1)
|
||||
List<MdCsSupplierbase> dataList = new ArrayList<>();
|
||||
for (int rowIndex = 1; rowIndex <= sheet.getLastRowNum(); rowIndex++) {
|
||||
Row row = sheet.getRow(rowIndex);
|
||||
if (row == null) {
|
||||
continue; // 跳过空行
|
||||
}
|
||||
Cell cell1 = row.getCell(0);
|
||||
MdCsSupplierbase byCode = this.getByCode(formatter.formatCellValue(cell1));
|
||||
if (ObjectUtil.isNotEmpty(byCode)) {
|
||||
throw new BadRequestException("物料[" + cell1 + "]已存在!");
|
||||
}
|
||||
|
||||
// 4. 读取6个字段
|
||||
MdCsSupplierbase entity = new MdCsSupplierbase();
|
||||
entity.setSupp_id(IdUtil.getStringId());
|
||||
entity.setSupp_code(formatter.formatCellValue(cell1));
|
||||
entity.setSupp_name(String.valueOf(row.getCell(1)));
|
||||
entity.setIs_used("1");
|
||||
entity.setCreate_id(SecurityUtils.getCurrentUserId());
|
||||
entity.setCreate_name(SecurityUtils.getCurrentNickName());
|
||||
entity.setCreate_time(DateUtil.now());
|
||||
dataList.add(entity);
|
||||
}
|
||||
|
||||
// 6. 批量保存
|
||||
this.saveBatch(dataList);
|
||||
|
||||
} catch (IOException e) {
|
||||
throw new RuntimeException("文件读取失败: " + e.getMessage());
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException("导入失败: " + e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void downloadImportSuppTemplate(HttpServletResponse response) {
|
||||
try {
|
||||
byte[] byteByTemplate = FileUtil.getByteByTemplate("excel/supplier_template.xlsx");
|
||||
FileUtil.download("供应商导入模板.xlsx", byteByTemplate, response);
|
||||
} catch (Exception e) {
|
||||
log.error(">>> 下载导入模板失败:", e);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public MdCsSupplierbase getByCode(String code) {
|
||||
return getOne(new LambdaQueryWrapper<MdCsSupplierbase>().eq(MdCsSupplierbase::getSupp_code, code));
|
||||
}
|
||||
}
|
||||
|
||||
@@ -125,7 +125,7 @@ public class MdPbMeasureunitServiceImpl extends ServiceImpl<MdPbMeasureunitMappe
|
||||
|
||||
try (InputStream inputStream = file.getInputStream();
|
||||
Workbook workbook = WorkbookFactory.create(inputStream)) {
|
||||
|
||||
DataFormatter formatter = new DataFormatter();
|
||||
// 2. 获取第一个Sheet
|
||||
Sheet sheet = workbook.getSheetAt(0);
|
||||
|
||||
@@ -137,15 +137,15 @@ public class MdPbMeasureunitServiceImpl extends ServiceImpl<MdPbMeasureunitMappe
|
||||
continue; // 跳过空行
|
||||
}
|
||||
Cell cell1 = row.getCell(0);
|
||||
MdPbMeasureunit byCode = this.getByCode(String.valueOf(cell1), false);
|
||||
MdPbMeasureunit byCode = this.getByCode(formatter.formatCellValue(cell1), false);
|
||||
if (ObjectUtil.isNotEmpty(byCode)) {
|
||||
throw new BadRequestException("物料[" + cell1 + "]已存在!");
|
||||
throw new BadRequestException("物料[" + formatter.formatCellValue(cell1) + "]已存在!");
|
||||
}
|
||||
|
||||
// 4. 读取6个字段
|
||||
MdPbMeasureunit entity = new MdPbMeasureunit();
|
||||
entity.setMeasure_unit_id(IdUtil.getStringId());
|
||||
entity.setUnit_code(String.valueOf(cell1));
|
||||
entity.setUnit_code(formatter.formatCellValue(cell1));
|
||||
entity.setUnit_name(String.valueOf(row.getCell(1)));
|
||||
entity.setQty_precision(new BigDecimal(String.valueOf(row.getCell(2))));
|
||||
entity.setCreate_id(SecurityUtils.getCurrentUserId());
|
||||
|
||||
Binary file not shown.
Reference in New Issue
Block a user