.

src/main/java/com/xkrs/controller/FileController.java

@@ -0,0 +1,82 @@
+package com.xkrs.controller;
+
+import com.xkrs.dao.FileDao;
+import com.xkrs.encapsulation.PromptMessageEnum;
+import com.xkrs.model.qo.FileQo;
+import com.xkrs.service.FileService;
+import com.xkrs.util.ExcelUploadUtil;
+import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
+import org.springframework.context.i18n.LocaleContextHolder;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.multipart.MultipartFile;
+
+import javax.annotation.Resource;
+import javax.transaction.Transactional;
+import java.io.IOException;
+import java.util.Locale;
+
+import static com.xkrs.encapsulation.OutputEncapsulation.outputEncapsulationObject;
+
+/**
+ * @Author: XinYi Song
+ * @Date: 2022/1/20 16:01
+ */
+@RestController
+public class FileController {
+
+    @Resource
+    private FileService fileService;
+
+    @Resource
+    private FileDao fileDao;
+
+    /**
+     * 上传模板信息
+     *
+     * @param fileQo
+     * @param fileExcel
+     * @return
+     * @throws IOException
+     */
+    @PostMapping("/insertFileExcel")
+    public String insertFileExcel(FileQo fileQo, @RequestParam("fileExcel") MultipartFile fileExcel) throws IOException {
+        return fileService.insertFileExcel(fileQo, fileExcel);
+    }
+
+    /**
+     * 修改上传的模板信息
+     *
+     * @param fileQo
+     * @param fileExcel
+     * @return
+     * @throws IOException
+     */
+    @PostMapping("/updateFileUploadPath")
+    @Transactional(rollbackOn = Exception.class)
+    public String updateFileUploadPath(FileQo fileQo, @RequestParam("fileExcel") MultipartFile fileExcel) throws IOException {
+        String lotNo = fileQo.getLotNo();
+        String machineNo = fileQo.getMachineNo();
+        String materialNo = fileQo.getMaterialNo();
+        String modelNo = fileQo.getModelNo();
+        String machineTypeNo = fileQo.getMachineTypeNo();
+        String craftItemName = fileQo.getCraftItemName();
+        Locale locale = LocaleContextHolder.getLocale();
+        String file = ExcelUploadUtil.memoryFile(fileExcel, 1);
+        fileDao.updateFileUploadPath(lotNo, machineNo, materialNo, modelNo, machineTypeNo, craftItemName, file);
+        return outputEncapsulationObject(PromptMessageEnum.SUCCESS, "修改成功！", locale);
+    }
+
+    /**
+     * 导出excel
+     *
+     * @param fileQo
+     * @return
+     */
+    @PostMapping("/exportExcel")
+    public String exportExcel(@RequestBody FileQo fileQo) throws IOException, InvalidFormatException {
+        return fileService.exportExcel(fileQo);
+    }
+}