@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;
}
|