package com.whyc.service;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.github.pagehelper.PageHelper;
|
import com.github.pagehelper.PageInfo;
|
import com.whyc.dto.FileUrlDTO;
|
import com.whyc.dto.Response;
|
import com.whyc.mapper.ProductMapper;
|
import com.whyc.pojo.DocUser;
|
import com.whyc.pojo.Product;
|
import com.whyc.pojo.ProductBom;
|
import com.whyc.util.ActionUtil;
|
import com.whyc.util.CommonUtil;
|
import com.whyc.util.FileUtil;
|
import com.whyc.util.Zip4jUtil;
|
import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
|
import org.apache.poi.ss.usermodel.*;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.io.*;
|
import java.text.SimpleDateFormat;
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.LinkedList;
|
import java.util.List;
|
|
@Service
|
public class ProductService {
|
@Autowired(required = false)
|
private ProductMapper mapper;
|
|
//查询出所有的产品信息(分页加模糊查询<产品的编码，型号，名字，定制表编号>
|
public Response getAllProduct(String parentCode, String parentName, String parentModel, String customCode, int pageCurr, int pageSize) {
|
PageHelper.startPage(pageCurr,pageSize);
|
QueryWrapper wrapper=new QueryWrapper();
|
if(parentCode!=null&&!parentCode.isEmpty()){
|
wrapper.like("parent_code",parentCode);
|
}
|
if(parentName!=null&&!parentName.isEmpty()){
|
wrapper.like("parent_name",parentName);
|
}
|
if(parentModel!=null&&!parentModel.isEmpty()){
|
wrapper.like("parent_model",parentModel);
|
}
|
if(customCode!=null&&!customCode.isEmpty()){
|
wrapper.like("custom_code",customCode);
|
}
|
wrapper.orderByAsc("id");
|
List list=mapper.selectList(wrapper);
|
PageInfo pageInfo=new PageInfo(list);
|
return new Response().setII(1,list.size()>0?true:false,pageInfo,"返回产品信息");
|
}
|
|
public Product getVersion(String parentCode, String customCode) {
|
QueryWrapper<Product> query = Wrappers.query();
|
query.eq("parent_code",parentCode).eq("custom_code",customCode).last(" limit 1");
|
return mapper.selectOne(query);
|
}
|
|
public Product getVersion(Integer productId) {
|
QueryWrapper<Product> query = Wrappers.query();
|
query.eq("product_id",productId);
|
return mapper.selectOne(query);
|
}
|
|
public int updateVersion(String parentCode, String customCode, Integer nextVersion) {
|
UpdateWrapper<Product> update = Wrappers.update();
|
update.set("version",nextVersion).eq("parent_code",parentCode).eq("custom_code",customCode);
|
mapper.update(null,update);
|
|
QueryWrapper<Product> query = Wrappers.query();
|
query.select("id").eq("parent_code",parentCode).eq("custom_code",customCode);
|
return mapper.selectOne(query).getId();
|
}
|
|
public Product getById(Integer id) {
|
return mapper.selectById(id);
|
}
|
|
public void deleteByParentCodeAndCustomCode(String parentCode, String customCode) {
|
UpdateWrapper<Product> update = Wrappers.update();
|
update.eq("parent_code",parentCode).eq("custom_code",customCode);
|
mapper.delete(update);
|
}
|
|
public void insert(Product product) {
|
mapper.insert(product);
|
}
|
|
public Response zipParse(MultipartFile file) throws IOException, InvalidFormatException {
|
List<ProductBom> list = new LinkedList<>();
|
|
Product product = new Product();
|
List<ProductBom> bomList = new LinkedList<>();
|
Response response = new Response();
|
//检查是否为zip
|
boolean isZip = Zip4jUtil.checkZipFileParam(file);
|
if (!isZip) {
|
return response.set(1, false, "上传的文件格式不是zip");
|
}
|
//zip存储路径:doc_file/product_submit/${username}/{dateFormat}/${timeStamp}
|
String rootFile = CommonUtil.getRootFile();
|
DocUser user = ActionUtil.getUser();
|
String dateFormat = new SimpleDateFormat("YYYY-MM").format(new Date());
|
long timeStamp = System.currentTimeMillis();
|
String filePath = rootFile + File.separator + "product_submit" + File.separator + user.getName() + File.separator + dateFormat + File.separator + timeStamp;
|
File parentFile = new File(filePath);
|
String originalFilename = file.getOriginalFilename();
|
File zipFile = new File(filePath + File.separator + originalFilename);
|
if (!zipFile.exists()) {
|
zipFile.mkdirs();
|
}
|
file.transferTo(zipFile);
|
//解压文件夹
|
Zip4jUtil.unPackZip(zipFile, null, filePath);
|
//遍历解压后的文件夹路径,解析excel
|
System.out.println(filePath);
|
List<String> fileList = new ArrayList<>();
|
//图纸筛选
|
fileList = FileUtil.getStaticFilePath(parentFile, fileList);
|
List<FileUrlDTO> dwgExistsList = getDwgList(fileList);
|
for (int i = 0; i < fileList.size(); i++) {
|
String fileTempUrl = fileList.get(i);
|
//查询需上传的bom数据
|
if (fileTempUrl.contains(".xls") && fileTempUrl.contains("(BOM)")) {
|
Workbook workbook = null;
|
InputStream inputStream = new FileInputStream(new File(fileTempUrl));
|
workbook = WorkbookFactory.create(inputStream);
|
inputStream.close();
|
List<? extends PictureData> allPictures = workbook.getAllPictures();
|
//取第一个sheet表
|
Sheet sheet = workbook.getSheetAt(0);
|
int lastRowNum = sheet.getLastRowNum();
|
//取第三行,并以第三行开始
|
Row row2 = sheet.getRow(1);
|
short lastCellNum = row2.getLastCellNum();
|
for (int k = 2; k < lastRowNum + 1; k++) {
|
if(k==2){
|
product.setParentCode(sheet.getRow(2).getCell(1).getStringCellValue());
|
product.setParentName(sheet.getRow(2).getCell(2).getStringCellValue());
|
product.setParentModel(sheet.getRow(2).getCell(3).getStringCellValue());
|
}
|
ProductBom bom = new ProductBom();
|
for (int j = 0; j < lastCellNum; j++) {
|
Row row = sheet.getRow(k);
|
Cell cell = row.getCell(j);
|
String cellValue = null;
|
Double cellValueInt = 0d;
|
if (j == 9 || j == 0) {
|
cellValueInt = cell.getNumericCellValue();
|
} else if (j != 15) {
|
cellValue = cell.getStringCellValue();
|
}
|
switch (j) {
|
case 0: {
|
bom.setId(cellValueInt.intValue());
|
}
|
break;
|
case 4: {
|
bom.setCategory(cellValue);
|
}
|
break;
|
case 5: {
|
bom.setSubCode(cellValue);
|
}
|
break;
|
case 6: {
|
bom.setSubName(cellValue);
|
}
|
break;
|
case 7: {
|
bom.setSubModel(cellValue);
|
//判断图纸查看是否存在
|
bom.setDwgExist(0);
|
String dwgFileName = cellValue + ".dwg";
|
for (int n = 0; n < dwgExistsList.size(); n++) {
|
FileUrlDTO fileUrlDTO = dwgExistsList.get(n);
|
if (dwgFileName.toUpperCase().equals(fileUrlDTO.getFileName().toUpperCase())) {
|
//存在
|
bom.setDwgExist(1);
|
bom.setDwgUrl(fileUrlDTO.getHttpFileUrl());
|
break;
|
}
|
}
|
}
|
break;
|
case 8: {
|
bom.setUnit(cellValue);
|
}
|
break;
|
case 9: {
|
bom.setQuantity(cellValueInt.intValue());
|
}
|
break;
|
case 10: {
|
bom.setProducer(cellValue);
|
}
|
break;
|
case 11: {
|
bom.setMaterial(cellValue);
|
}
|
break;
|
case 12: {
|
bom.setThickness(cellValue);
|
}
|
break;
|
case 13: {
|
bom.setSurfaceDetail(cellValue);
|
}
|
break;
|
case 14: {
|
bom.setNotes(cellValue);
|
}
|
break;
|
case 15: {
|
//图片,从0开始,到图片size为止
|
int m = k - 2;
|
if (m < allPictures.size()) {
|
PictureData pictureData = allPictures.get(m);
|
//图片存储 product_submit/username/2022-07/
|
String approvingPath = rootFile + File.separator + "product_submit" + File.separator + user.getName() + File.separator + dateFormat + File.separator + timeStamp;
|
File provingFile = new File(approvingPath);
|
if (!provingFile.exists()) {
|
provingFile.mkdirs();
|
}
|
String suffix = pictureData.suggestFileExtension();
|
String picturePath = approvingPath + File.separator + bom.getSubModel() + "." + suffix;
|
String picturePathFront = "doc_file" + File.separator + "product_submit" + File.separator + user.getName() + File.separator + dateFormat + File.separator + timeStamp + File.separator + bom.getSubModel() + "." + suffix;
|
byte[] data = pictureData.getData();
|
FileOutputStream fileOutputStream = null;
|
File pictureFile = new File(picturePath);
|
fileOutputStream = new FileOutputStream(pictureFile);
|
fileOutputStream.write(data);
|
|
bom.setPictureUrl(picturePathFront);
|
}
|
}
|
break;
|
}
|
}
|
bomList.add(bom);
|
}
|
product.setBomList(bomList);
|
}
|
}
|
return response.setII(1, true, product, filePath);
|
|
}
|
|
private List<FileUrlDTO> getDwgList(List<String> fileList) {
|
List<FileUrlDTO> list = new LinkedList<>();
|
fileList.forEach(fileUrl->{
|
if(fileUrl.substring(fileUrl.lastIndexOf(".")+1).equals("dwg")) {
|
FileUrlDTO dto = new FileUrlDTO();
|
dto.setFileName(fileUrl.substring(fileUrl.lastIndexOf(File.separator) + 1));
|
dto.setHttpFileUrl(fileUrl.substring(fileUrl.lastIndexOf("doc_file" + File.separator + "product_approving")));
|
list.add(dto);
|
}
|
});
|
return list;
|
}
|
}
|
