|
@ResponseBody
@RequestMapping(value = "import")
public Object import(MultipartFile file) {
if (file.isEmpty()) {
return ResultObject.failure("文件为空");
}
ResultObject result = new ResultObject();
//记录集合
List<Map> mapList = new ArrayList<Map>();
//校验结果
boolean reqFlag = true;
//回复消息
String reqMsg = "";
//报错消息数
Integer error = 0;
//解析文件
try {
//转换成输入流
InputStream is = file.getInputStream();
//得到excel
Workbook workbook = Workbook.getWorkbook(is);
//得到sheet
Sheet sheet = workbook.getSheet(0);
//得到列数
int colsNum = sheet.getColumns();
//得到行数
int rowsNum = sheet.getRows();
if (rowsNum == 1) return ResultObject.failure("没有数据");
//单元格
Cell cell;
//数据校验
for (int i = 1; i < rowsNum; i++) {//第一行是标题,所以i从1开始
Map<Integer, Object> map = new HashMap<Integer, Object>();
for (int j = 0; j < colsNum; j++) {
cell = sheet.getCell(j, i);//第一个参数是列.第二个参数是行
if (j < 4 && "".equals(cell.getContents())) { //----这里判断必填项(前4列)
reqFlag = false;
reqMsg += "第" + (i + 1) + "行错误,错误信息:" + "必填项缺漏";
reqMsg += "<br>";
error++;
break;
}
String cellString = cell.getContents();
cellString = cellString.trim();
switch (j) {
case 1: {
//进行校验处理,例如手机号
if (!StringUtil.isMobileNo(cellString)) {
reqFlag = false;
reqMsg += "第" + (i + 1) + "行错误,错误信息:" + "联系电话有误";
reqMsg += "<br>";
error++;
} else {
map.put(j, cellString);
}
break;
}
case 2: { ### break; }
case 3: { ### break; }
//无需校验,归入default
default: {
map.put(j, cell.getContents());
}
}
}
if (reqFlag) {//校验通过
mapList.add(map);
}
}
} catch (IOException e) {
e.printStackTrace();
} catch (BiffException e) {
e.printStackTrace();
}
//入库
try {
if (mapList.size() > 0 && reqFlag) {
//此处try,catch应优化为事务处理maplist实现全记录成功或失败
result.setSuccess("提交成功");
} else {
if (error > 10) {//设置要显示的错误数
int index = StringUtil.getIndex(reqMsg, 10, "<br>");
reqMsg = reqMsg.substring(0, index + 4);
reqMsg += "未显示错误数:" + (error - 10) + "条";
reqMsg += "<br>";
}
result.setFailure(reqMsg);
}
} catch (Exception e) {
result.setFailure("入库错误,请联系管理员!");
e.printStackTrace();
}
return result;
}
|