工业测量

src/main/java/com/xkrs/service/impl/FileServiceImpl.java

@@ -0,0 +1,97 @@
+package com.xkrs.service.impl;
+
+import com.xkrs.dao.DataDictDao;
+import com.xkrs.dao.DataSourceDao;
+import com.xkrs.dao.FileDao;
+import com.xkrs.encapsulation.PromptMessageEnum;
+import com.xkrs.model.entity.FileEntity;
+import com.xkrs.model.qo.FileQo;
+import com.xkrs.service.FileService;
+import com.xkrs.util.ExcelUploadUtil;
+import com.xkrs.util.ExportExcel;
+import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
+import org.springframework.context.i18n.LocaleContextHolder;
+import org.springframework.stereotype.Service;
+import org.springframework.web.multipart.MultipartFile;
+
+import javax.annotation.Resource;
+import javax.servlet.http.HttpServletResponse;
+import javax.transaction.Transactional;
+import java.io.IOException;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Locale;
+import java.util.Map;
+
+import static com.xkrs.encapsulation.OutputEncapsulation.outputEncapsulationObject;
+
+/**
+ * @Author: XinYi Song
+ * @Date: 2022/1/20 15:24
+ */
+@Service
+public class FileServiceImpl implements FileService {
+
+    @Resource
+    private FileDao fileDao;
+
+    @Resource
+    private DataDictDao dataDictDao;
+
+    @Resource
+    private DataSourceDao dataSourceDao;
+
+    /**
+     * 添加模板信息
+     * @param fileQo
+     * @return
+     */
+    @Override
+    public String insertFileExcel(FileQo fileQo, MultipartFile fileExcel) throws IOException {
+        Locale locale = LocaleContextHolder.getLocale();
+        FileEntity fileEntity = fileDao.findByFileModelNumberAndFileBatchNumberAndFileMachineCodeAndFileProductNumber(fileQo.getFileModelNumber(), fileQo.getFileBatchNumber(), fileQo.getFileMachineCode(), fileQo.getFileProductNumber());
+        if(fileEntity != null){
+            return outputEncapsulationObject(PromptMessageEnum.DATA_EXIT,"已存在该产品的模板!",locale);
+        }
+        if(fileExcel == null){
+            return outputEncapsulationObject(PromptMessageEnum.DATA_NONE,"模板不能为空！",locale);
+        }
+        String file = ExcelUploadUtil.memoryFile(fileExcel);
+        FileEntity fileEntity1 = new FileEntity();
+        fileEntity1.setFileModelNumber(fileQo.getFileModelNumber());
+        fileEntity1.setFileBatchNumber(fileQo.getFileBatchNumber());
+        fileEntity1.setFileMachineCode(fileQo.getFileMachineCode());
+        fileEntity1.setFileProductNumber(fileQo.getFileProductNumber());
+        fileEntity1.setFileUploadPath(file);
+
+        fileDao.save(fileEntity1);
+        return outputEncapsulationObject(PromptMessageEnum.SUCCESS,"添加成功！",locale);
+    }
+
+    /**
+     * 导出excel
+     * @param fileQo
+     * @return
+     */
+    @Transactional(rollbackOn = Exception.class)
+    @Override
+    public String exportExcel(FileQo fileQo) throws IOException, InvalidFormatException {
+        Locale locale = LocaleContextHolder.getLocale();
+        FileEntity entity = fileDao.findByFileModelNumberAndFileBatchNumberAndFileMachineCodeAndFileProductNumber(fileQo.getFileModelNumber(), fileQo.getFileBatchNumber(), fileQo.getFileMachineCode(), fileQo.getFileProductNumber());
+        if(entity == null){
+            return outputEncapsulationObject(PromptMessageEnum.DATA_NONE,"暂时没有该产品的模板，请添加！",locale);
+        }
+        String fileUploadPath = entity.getFileUploadPath();
+        List<Map<String, String>> maps = dataSourceDao.selectDataNameAndData(fileQo.getFileModelNumber(), fileQo.getFileBatchNumber(), fileQo.getFileMachineCode(), fileQo.getFileProductNumber());
+        if(maps == null || maps.size() == 0){
+            return outputEncapsulationObject(PromptMessageEnum.DATA_NONE,"暂时没有该产品的信息！",locale);
+        }
+        Map<String,String> map = new HashMap<String,String>();
+        for(Map<String,String> stringMap : maps){
+            map.put(stringMap.get("dataname"),stringMap.get("numericalvalue"));
+        }
+        String fill = ExportExcel.exportToProveExcel(map,fileUploadPath);
+        fileDao.updateFileDownloadPath(fileQo.getFileModelNumber(), fileQo.getFileBatchNumber(), fileQo.getFileMachineCode(), fileQo.getFileProductNumber(),fill);
+        return outputEncapsulationObject(PromptMessageEnum.SUCCESS,fill,locale);
+    }
+}