山西直流电源系统
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
whyclxw
35 分钟以前 e196f42e9a977c0b2f8452dce875b110fb76e92f 
 src/main/java/com/whyc/service/DeviceSpareService.java


@@ -1,5 +1,8 @@


package com.whyc.service;




import cn.afterturn.easypoi.excel.ExcelExportUtil;


import cn.afterturn.easypoi.excel.entity.TemplateExportParams;


import com.baomidou.mybatisplus.core.conditions.Wrapper;


import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;


import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;


import com.baomidou.mybatisplus.core.toolkit.Wrappers;


@@ -13,17 +16,22 @@


import com.whyc.util.ThreadLocalUtil;


import org.apache.commons.lang.StringUtils;


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


import org.apache.poi.xssf.usermodel.*;


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


import org.springframework.core.io.ClassPathResource;


import org.springframework.stereotype.Service;


import org.springframework.transaction.annotation.Transactional;


import org.springframework.web.multipart.MultipartFile;




import javax.annotation.Resource;


import javax.servlet.http.HttpServletResponse;


import java.io.File;


import java.io.FileOutputStream;


import java.io.IOException;


import java.util.ArrayList;


import java.util.Date;


import java.util.List;


import java.net.URLEncoder;


import java.time.LocalDateTime;


import java.time.format.DateTimeFormatter;


import java.util.*;




@Service


public class DeviceSpareService {


@@ -66,10 +74,10 @@


        if (file != null && file.size() > 0) {


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


                MultipartFile multipartFile = file.get(i);


                String fileName = multipartFile.getOriginalFilename();


                //String fileName = multipartFile.getOriginalFilename();


                //将fileName中可能存在的,去掉


                fileName = fileName.replace(",","");


                String filePath = fileDirPath + File.separator + timeFormat+"_"+fileName;


                //fileName = fileName.replace(",","");


                String filePath = fileDirPath + File.separator + spare.getName()+"_"+spare.getModel()+"_"+spare.getVersion() + "_" + (i+1) + "_"+ timeFormat+".png";




                multipartFile.transferTo(new File(filePath));


                String split = "pis_file"+File.separator+"deviceSpare";


@@ -229,13 +237,143 @@


            spare.setSupplier(supplier);




            spareList.add(spare);


            //第七列为图片 TODO 待处理


            /*Cell cell = row.getCell(7);


            CellType cellType = cell.getCellType();


            System.out.println("-");*/


        }


        addBatch(spareList);




        return null;


        }




        //第七列为图片-浮动式图片


        //获取绘图对象中的所有图形


        XSSFDrawing drawing = (XSSFDrawing) sheet.getDrawingPatriarch();


        if (drawing == null) {


            drawing = (XSSFDrawing) sheet.createDrawingPatriarch();


        }


        List<XSSFShape> shapes = drawing.getShapes();


        for (XSSFShape shape : shapes) {


            if (shape instanceof XSSFPicture) {


                XSSFPicture pic = (XSSFPicture) shape;


                XSSFClientAnchor anchor = pic.getClientAnchor();


                //获取图片所在的起始行


                int rowIndex = anchor.getRow1();


                byte[] data = pic.getPictureData().getData();




                DeviceSpare spare = spareList.get(rowIndex - 1);


                //对存储路径进行定义


                Date now = new Date();


                String timeFormat = ThreadLocalUtil.format(ThreadLocalUtil.TIME_YYYY_MM_DD_HH_MM_SS_UNION, now);


                String dirMonth = ThreadLocalUtil.format(ThreadLocalUtil.TIME_YYYY_MM, now);


                String fileDirPath = CommonUtil.getRootFile() + "deviceSpare" + File.separator + dirMonth;


                File fileDir = new File(fileDirPath);


                //如果文件夹不存在则创建


                if (!fileDir.exists()) {


                    fileDir.mkdirs();


                }


                String filePath = fileDirPath + File.separator + spare.getName()+"_"+spare.getModel()+"_"+spare.getVersion() + "_"+ timeFormat+".png";




                // 保存图片到本地


                try (FileOutputStream fos = new FileOutputStream(filePath)) {


                    fos.write(data);


                }


                String split = "pis_file"+File.separator+"deviceSpare";


                String picUrl = File.separator + filePath.substring(filePath.indexOf(split));


                spare.setPictureUrl(picUrl);


            }


        }




        //新增列表


        List<DeviceSpare> spareListNew = new ArrayList<>();


        //更新列表


        List<DeviceSpare> spareListUpdate = new ArrayList<>();


        //查询库中的所有设备器件


        List<DeviceSpare> deviceSpareListInDB = mapper.selectList((Wrapper<DeviceSpare>) CommonUtil.nullObject);


        for (DeviceSpare spare : spareList){


            boolean isExist = false;


            for (DeviceSpare spareInDB : deviceSpareListInDB){


                if (spare.getName().equals(spareInDB.getName())


                        && spare.getModel().equals(spareInDB.getModel())


                        && spare.getVersion().equals(spareInDB.getVersion())


                        && spare.getBrand().equals(spareInDB.getBrand())


                        && spare.getType().equals(spareInDB.getType())


                        && spare.getSupplier().equals(spareInDB.getSupplier())


                ){


                    isExist = true;


                    spare.setId(spareInDB.getId());


                    spare.setQuantity(spareInDB.getQuantity()+spare.getQuantity());


                    if (spare.getPictureUrl() != null && spareInDB.getPictureUrl()!=null){


                        spare.setPictureUrl(spareInDB.getPictureUrl()+","+spare.getPictureUrl());


                    }


                    spareListUpdate.add(spare);


                }


            }


            if(!isExist){


                //不存在则新增


                spareListNew.add(spare);


            }


        }


        //更新


        if(spareListUpdate.size()>0) {


            updateQuantityAndPictureBatch(spareListUpdate);


        }


        //新增


        if(spareListNew.size()>0) {


            addBatch(spareListNew);


        }




        return new Response().setII(1,"导入完成");


    }




    private void updateQuantityAndPictureBatch(List<DeviceSpare> spareListUpdate) {


        mapper.updateQuantityAndPictureBatch(spareListUpdate);


    }




    public void excelExport(HttpServletResponse response) {


        //查询所有的设备器件


        List<DeviceSpare> deviceSpareList = mapper.selectList((Wrapper<DeviceSpare>) CommonUtil.nullObject);


        //获取导出模板地址


        ClassPathResource classPathResource = new ClassPathResource("excel_templates/template_device_spare.xlsx");


        String path = classPathResource.getPath();


        TemplateExportParams templateExportParams1 = new TemplateExportParams(path,true);


        // 构建导出数据模型


        Map<String, Object> map = new HashMap<>();


        map.put("deviceSpareList", deviceSpareList); // 假设模板中使用 ${deviceSpareList} 作为变量名




        Workbook wb = ExcelExportUtil.exportExcel(templateExportParams1, map);


        try {


            LocalDateTime now = LocalDateTime.now();


            String fileName = "维修管理器件库存_"+now.format(DateTimeFormatter.ofPattern("yyyyMMddHHmmss"))+".xlsx";


            response.setContentType("application/vnd.ms-excel");


            response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileName, "utf-8"));


            response.flushBuffer();


            wb.write(response.getOutputStream());


        } catch (IOException e) {


            e.printStackTrace();


        }




    }




    public Response andOrChangePicture(Integer id, MultipartFile file) throws IOException {


        DeviceSpare spare = mapper.selectById(id);


        //对存储路径进行定义


        Date now = new Date();


        String timeFormat = ThreadLocalUtil.format(ThreadLocalUtil.TIME_YYYY_MM_DD_HH_MM_SS_UNION, now);


        String dirMonth = ThreadLocalUtil.format(ThreadLocalUtil.TIME_YYYY_MM, now);


        String fileDirPath = CommonUtil.getRootFile() + "deviceSpare" + File.separator + dirMonth;


        File fileDir = new File(fileDirPath);


        //如果文件夹不存在则创建


        if (!fileDir.exists()) {


            fileDir.mkdirs();


        }


        String filePath = fileDirPath + File.separator + spare.getName()+"_"+spare.getModel()+"_"+spare.getVersion() + "_"+ timeFormat+".png";




        // 保存图片到本地


        file.transferTo(new File(filePath));


        String split = "pis_file"+File.separator+"deviceSpare";


        String picUrl = File.separator + filePath.substring(filePath.indexOf(split));


        //更新图片


        UpdateWrapper<DeviceSpare> update = Wrappers.update();


        update.eq("id",id);


        update.set("picture_url",picUrl);


        mapper.update((DeviceSpare) CommonUtil.nullObject,update);


        //记录变更


        deviceSpareLogService.add(id,2,"更换图片",now);


        return new Response().setII(1,"新增或者替换图片完成");


    }


}