导入结算表

oa-back/ruoyi-admin/src/main/java/com/ruoyi/web/controller/oa/CmcSettleController.java

@@ -1,17 +1,33 @@
 package com.ruoyi.web.controller.oa;
 
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.InputStream;
 import java.math.BigDecimal;
 import java.text.ParseException;
 import java.text.SimpleDateFormat;
+import java.util.ArrayList;
 import java.util.Calendar;
 import java.util.List;
+import java.util.Objects;
 import javax.servlet.http.HttpServletResponse;
 
 import com.alibaba.fastjson2.JSONArray;
 import com.alibaba.fastjson2.JSONObject;
+import com.ruoyi.common.utils.SnowFlake;
+import com.ruoyi.file.domain.FilesAchievement;
+import com.ruoyi.file.domain.FilesStorage;
 import com.ruoyi.oa.domain.CmcSettle;
 import com.ruoyi.oa.service.ICmcSettleSummaryService;
+import org.apache.poi.hssf.usermodel.HSSFSheet;
+import org.apache.poi.hssf.usermodel.HSSFWorkbook;
+import org.apache.poi.ss.usermodel.CellType;
+import org.apache.poi.xssf.usermodel.XSSFCell;
+import org.apache.poi.xssf.usermodel.XSSFSheet;
+import org.apache.poi.xssf.usermodel.XSSFWorkbook;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.PutMapping;
@@ -27,6 +43,7 @@ import com.ruoyi.common.enums.BusinessType;
 import com.ruoyi.oa.service.ICmcSettleService;
 import com.ruoyi.common.utils.poi.ExcelUtil;
 import com.ruoyi.common.core.page.TableDataInfo;
+import org.springframework.web.multipart.MultipartFile;
 
 /**
  * cmc结算审批Controller
@@ -41,9 +58,8 @@ public class CmcSettleController extends BaseController
     @Autowired
     private ICmcSettleService cmcSettleService;
 
-    @Autowired
-    private ICmcSettleSummaryService cmcSettleSummaryService;
-
+    @Value("${cmc.profile}")
+    private String profile;
     /**
      * 查询cmc结算审批列表
      */
@@ -67,6 +83,43 @@ public class CmcSettleController extends BaseController
         util.exportExcel(response, list, "cmc结算审批数据");
     }
 
+    /**
+     * 上传结算excel文件
+     */
+    @PostMapping("/uploadSheet")
+    public AjaxResult uploadSettleSheet(MultipartFile file) throws Exception {
+        if (file.isEmpty()) {
+            return AjaxResult.error("文件内容为空！");
+        }
+        else {
+            File profilePath = new File(profile + "/upload/导入数据");
+            if (!profilePath.exists())
+                profilePath.mkdirs();
+            File transferFile = new File( profilePath + "/" + file.getOriginalFilename());
+            file.transferTo(transferFile);
+            InputStream in = new FileInputStream(transferFile);
+            XSSFSheet sheetAt = null;
+            int rowNumber = 0;
+            if (file.getOriginalFilename() != null) {
+                sheetAt = new XSSFWorkbook(in).getSheetAt(0);
+                rowNumber = sheetAt.getPhysicalNumberOfRows();
+            }
+            JSONArray jsonArray = new JSONArray();
+            JSONObject jsonObject = new JSONObject();
+            for (int i = 1; i < rowNumber - 1;) {
+                i++;
+                jsonObject.put("priceId", getStringCellValue(sheetAt.getRow(i).getCell(0)));
+                jsonObject.put("content", getStringCellValue(sheetAt.getRow(i).getCell(1)));
+                jsonObject.put("workload", getStringCellValue(sheetAt.getRow(i).getCell(4)));
+                jsonObject.put("groundType", getStringCellValue(sheetAt.getRow(i).getCell(5)));
+                jsonObject.put("coefficient", getStringCellValue(sheetAt.getRow(i).getCell(6)));
+                jsonObject.put("remark", getStringCellValue(sheetAt.getRow(i).getCell(7)));
+                jsonArray.add(jsonObject);
+            }
+            return AjaxResult.success(jsonArray);
+        }
+    }
+
     /**
      * 获取cmc结算审批详细信息
      */
@@ -202,4 +255,17 @@ public class CmcSettleController extends BaseController
         }
         typeAmountObject.put("其他结算", oAmount);
     }
+
+    public String getStringCellValue(XSSFCell xssfCell) {
+        if (xssfCell == null) {
+            return "";
+        }
+        if (xssfCell.getCellType() == CellType.NUMERIC) {
+            return String.valueOf(xssfCell.getNumericCellValue());
+        } else if (xssfCell.getCellType() == CellType.BOOLEAN) {
+            return String.valueOf(xssfCell.getBooleanCellValue());
+        } else {
+            return xssfCell.getStringCellValue();
+        }
+    }
 }