309 lines
11 KiB
Plaintext
309 lines
11 KiB
Plaintext
|
package com.sercurityControl.proteam.util;
|
|||
|
|
|
|||
|
|
import com.alibaba.fastjson2.JSONObject;
|
|||
|
|
import org.apache.commons.lang3.StringUtils;
|
|||
|
|
import org.apache.poi.hssf.usermodel.*;
|
|||
|
|
import org.apache.poi.ss.usermodel.*;
|
|||
|
|
import org.apache.poi.xssf.usermodel.*;
|
|||
|
|
import org.springframework.web.multipart.MultipartFile;
|
|||
|
|
|
|||
|
|
import java.io.InputStream;
|
|||
|
|
import java.util.*;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @author HeiZi
|
|||
|
|
*/
|
|||
|
|
public class ImportExcelHelper {
|
|||
|
|
|
|||
|
|
static int cell1=1;
|
|||
|
|
static int cell2=2;
|
|||
|
|
static int cell3=3;
|
|||
|
|
static int cell4=4;
|
|||
|
|
static int cell5=5;
|
|||
|
|
static int cell6=6;
|
|||
|
|
|
|||
|
|
public static List<?> readExcel(MultipartFile file, Class<?> mClass) throws Exception {
|
|||
|
|
String fix=".xls";
|
|||
|
|
String fix2=".xlsx";
|
|||
|
|
String fileName = file.getOriginalFilename();
|
|||
|
|
System.out.println("OriginalFilename：" + fileName);
|
|||
|
|
if (!StringUtils.endsWithAny(fileName, fix,fix2)) {
|
|||
|
|
throw new RuntimeException("不支持excel以外的文件导入！");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
InputStream inputStream = file.getInputStream();
|
|||
|
|
String className = mClass.getSimpleName();
|
|||
|
|
System.out.println("className：" + className);
|
|||
|
|
//根据指定的文件输入流导入Excel从而产生Workbook对象
|
|||
|
|
Workbook workbook = null;
|
|||
|
|
Sheet sheet = null;
|
|||
|
|
if (fileName!=null && fileName.endsWith(fix)) {
|
|||
|
|
workbook = new HSSFWorkbook(inputStream);
|
|||
|
|
//获取Excel文档中的第一个表单
|
|||
|
|
sheet = workbook.getSheetAt(0);
|
|||
|
|
if (checkModal(sheet, className)) {
|
|||
|
|
throw new RuntimeException("模板错误，请重新选择模板！");
|
|||
|
|
}
|
|||
|
|
} else if (fileName!=null && fileName.endsWith(fix2)) {
|
|||
|
|
workbook = new XSSFWorkbook(inputStream);
|
|||
|
|
//获取Excel文档中的第一个表单
|
|||
|
|
sheet = workbook.getSheetAt(0);
|
|||
|
|
if (checkModal(sheet, className)) {
|
|||
|
|
throw new RuntimeException("模板错误，请重新选择模板！");
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
switch (className) {
|
|||
|
|
case "VoiInfoEntity":
|
|||
|
|
System.err.println("-----违章依据导入------");
|
|||
|
|
break;
|
|||
|
|
case "BackboneScoreEntity":
|
|||
|
|
System.err.println("-----骨干成绩导入------");
|
|||
|
|
break;
|
|||
|
|
case "BidTowerVo":
|
|||
|
|
System.err.println("-----标段杆塔导入------");
|
|||
|
|
break;
|
|||
|
|
case "TowerVo":
|
|||
|
|
System.err.println("-----杆塔导入------");
|
|||
|
|
break;
|
|||
|
|
case "SupeConsUserVo":
|
|||
|
|
System.err.println("-----监理/施工单位导入------");
|
|||
|
|
break;
|
|||
|
|
default:
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
int sheetCount = 0;
|
|||
|
|
if (workbook != null) {
|
|||
|
|
workbook.getNumberOfSheets();
|
|||
|
|
}
|
|||
|
|
System.out.println("Sheet(表单)数量：" + sheetCount);
|
|||
|
|
//获得最后一条记录得的行号，从0开始
|
|||
|
|
int totalRowNum = 0;
|
|||
|
|
if (sheet != null) {
|
|||
|
|
totalRowNum = sheet.getLastRowNum();
|
|||
|
|
}
|
|||
|
|
System.out.println("总记录数：" + (totalRowNum + 1));
|
|||
|
|
return createBean(sheet, mClass);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public static boolean checkModal(Sheet sheet, String className) {
|
|||
|
|
int colNum = sheet.getRow(0).getLastCellNum();
|
|||
|
|
switch (className) {
|
|||
|
|
case "VoiInfoEntity":
|
|||
|
|
return colNum != 2;
|
|||
|
|
case "BackboneScoreEntity":
|
|||
|
|
return colNum != 12;
|
|||
|
|
case "BidTowerVo":
|
|||
|
|
return colNum != 7;
|
|||
|
|
case "TowerVo":
|
|||
|
|
case "SupeConsUserVo":
|
|||
|
|
return colNum != 4;
|
|||
|
|
default:
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private static List<JSONObject> createBean(Sheet sheet, Class<?> mClass) {
|
|||
|
|
if (sheet == null || sheet.getLastRowNum() < 0) {
|
|||
|
|
return null;
|
|||
|
|
}
|
|||
|
|
List<JSONObject> list = new ArrayList<>();
|
|||
|
|
// 总列数
|
|||
|
|
int last = sheet.getRow(0).getLastCellNum();
|
|||
|
|
System.out.println("列数：" + last);
|
|||
|
|
try {
|
|||
|
|
//对Sheet中的每一行进行迭代
|
|||
|
|
for (Row row : sheet) {
|
|||
|
|
if (row == null){
|
|||
|
|
continue;
|
|||
|
|
}
|
|||
|
|
// 第一行是标题栏
|
|||
|
|
switch (mClass.getSimpleName()) {
|
|||
|
|
case "BackboneScoreEntity":
|
|||
|
|
case "BidTowerVo":
|
|||
|
|
case "TowerVo":
|
|||
|
|
case "SupeConsUserVo":
|
|||
|
|
if (row.getRowNum() < 2){
|
|||
|
|
continue;
|
|||
|
|
}
|
|||
|
|
break;
|
|||
|
|
default:
|
|||
|
|
if (row.getRowNum() < 1) {
|
|||
|
|
continue;
|
|||
|
|
}
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
boolean isBlankRow = true;
|
|||
|
|
for (Cell c : row) {
|
|||
|
|
if (c.getCellType() != CellType.BLANK) {
|
|||
|
|
isBlankRow = false;
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if (isBlankRow) {
|
|||
|
|
// throw new RuntimeException("模板中含有无效的空行，请删除多余的空行后再导入！");
|
|||
|
|
continue;
|
|||
|
|
}
|
|||
|
|
JSONObject obj = new JSONObject();
|
|||
|
|
obj.put("rowNo", row.getRowNum() + 1);
|
|||
|
|
switch (mClass.getSimpleName()) {
|
|||
|
|
case "VoiInfoEntity":
|
|||
|
|
satelliting(2, row);
|
|||
|
|
if (row.getCell(1) != null){
|
|||
|
|
// 违章依据
|
|||
|
|
obj.put("notiInfo", row.getCell(1).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
break;
|
|||
|
|
case "BackboneScoreEntity":
|
|||
|
|
backboneScoreEntity(obj,row);
|
|||
|
|
break;
|
|||
|
|
case "BidTowerVo":
|
|||
|
|
bidTowerVo(obj,row);
|
|||
|
|
break;
|
|||
|
|
case "TowerVo":
|
|||
|
|
satelliting(4, row);
|
|||
|
|
if (row.getCell(1) != null){
|
|||
|
|
// 杆塔编码名称
|
|||
|
|
obj.put("towerName", row.getCell(1).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
if (row.getCell(2) != null){
|
|||
|
|
// 经度{
|
|||
|
|
obj.put("lon", row.getCell(2).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
// 纬度
|
|||
|
|
if (row.getCell(3) != null){
|
|||
|
|
obj.put("lat", row.getCell(3).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
break;
|
|||
|
|
case "SupeConsUserVo":
|
|||
|
|
sueConsUser(obj,row);
|
|||
|
|
break;
|
|||
|
|
default:
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
list.add(obj);
|
|||
|
|
}
|
|||
|
|
} catch (IllegalStateException e) {
|
|||
|
|
e.printStackTrace();
|
|||
|
|
throw new RuntimeException("模板中含有单元格数据格式不正确");
|
|||
|
|
}
|
|||
|
|
return list;
|
|||
|
|
}
|
|||
|
|
public static void sueConsUser(JSONObject obj, Row row){
|
|||
|
|
satelliting(4, row);
|
|||
|
|
if (row.getCell(1) != null){
|
|||
|
|
// 单位名称
|
|||
|
|
obj.put("name", row.getCell(1).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
if (row.getCell(2) != null){
|
|||
|
|
// 单位编码
|
|||
|
|
obj.put("code", row.getCell(2).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
if (row.getCell(3) != null){
|
|||
|
|
// 备注
|
|||
|
|
obj.put("remarks", row.getCell(3).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
public static void backboneScoreEntity(JSONObject obj, Row row){
|
|||
|
|
satelliting(12, row);
|
|||
|
|
if (row.getCell(1) != null){
|
|||
|
|
// 人员编码
|
|||
|
|
obj.put("userCode", row.getCell(1).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (row.getCell(2) != null){
|
|||
|
|
// 姓名
|
|||
|
|
obj.put("userName", row.getCell(2).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (row.getCell(3) != null){
|
|||
|
|
// 身份证号
|
|||
|
|
obj.put("idNumber", row.getCell(3).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (row.getCell(4) != null){
|
|||
|
|
// 所属单位
|
|||
|
|
obj.put("unit", row.getCell(4).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (row.getCell(5) != null){
|
|||
|
|
// 所属部门
|
|||
|
|
obj.put("part", row.getCell(5).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (row.getCell(6) != null){
|
|||
|
|
// 试卷成绩
|
|||
|
|
obj.put("score", row.getCell(6).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (row.getCell(7) != null){
|
|||
|
|
// 合格
|
|||
|
|
obj.put("qualified", row.getCell(7).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (row.getCell(8) != null){
|
|||
|
|
// 考试时间
|
|||
|
|
obj.put("testTime", row.getCell(8).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (row.getCell(9) != null){
|
|||
|
|
// 有效期
|
|||
|
|
obj.put("effectiveTime", row.getCell(9).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (row.getCell(10) != null){
|
|||
|
|
// 工种
|
|||
|
|
obj.put("workType", row.getCell(10).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (row.getCell(11) != null){
|
|||
|
|
// 人员
|
|||
|
|
obj.put("peopleType", row.getCell(11).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
public static void bidTowerVo(JSONObject obj, Row row){
|
|||
|
|
satelliting(7, row);
|
|||
|
|
if (row.getCell(cell1) != null){
|
|||
|
|
// 标段工程编码
|
|||
|
|
obj.put("bidNo", row.getCell(cell1).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
if (row.getCell(cell2) != null){
|
|||
|
|
// 工程编码
|
|||
|
|
obj.put("proNo", row.getCell(cell2).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
if (row.getCell(cell3) != null){
|
|||
|
|
// 标段工程名称
|
|||
|
|
obj.put("proName", row.getCell(cell3).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (row.getCell(cell4) != null){
|
|||
|
|
// 杆塔编码名称)
|
|||
|
|
obj.put("towerName", row.getCell(cell4).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
if (row.getCell(cell5) != null){
|
|||
|
|
// 经度
|
|||
|
|
obj.put("lon", row.getCell(cell5).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
if (row.getCell(cell6) != null){
|
|||
|
|
// 纬度
|
|||
|
|
obj.put("lat", row.getCell(cell6).getStringCellValue());
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private static void satelliting(int j, Row row) {
|
|||
|
|
for (int i = 0; i < j; i++) {
|
|||
|
|
if (row.getCell(i) != null) {
|
|||
|
|
row.getCell(i).setCellType(CellType.STRING);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|