package cc.mrbird.common.util;
import java.io.InputStream;
import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.apache.poi.hssf.usermodel.HSSFRichTextString;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.RichTextString;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFRichTextString;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
public class ExcelUtil2 {
private final static String excel2003L = ".xls"; // 2003- 版本的excel
private final static String excel2007U = ".xlsx"; // 2007+ 版本的excel
/**
* 将流中的Excel数据转成List<Map>
*
* @param in
* 输入流
* @param fileName
* 文件名（判断Excel版本）
* @param mapping
* 字段名称映射
* @return
* @throws Exception
*/
public static List<Map<String, Object>> parseExcel(InputStream in, String fileName,
Map<String, String> mapping) throws Exception {
// 根据文件名来创建Excel工作薄
Workbook work = getWorkbook(in, fileName);
if (null == work) {
throw new Exception("创建Excel工作薄为空！");
}
Sheet sheet = null;
Row row = null;
Cell cell = null;
// 返回数据
List<Map<String, Object>> ls = new ArrayList<Map<String, Object>>();
// 遍历Excel中所有的sheet
for (int i = 0; i < work.getNumberOfSheets(); i++) {
sheet = work.getSheetAt(i);
if (sheet == null)
continue;
// 取第一行标题
row = sheet.getRow(0);
String title[] = null;
if (row != null) {
title = new String[row.getLastCellNum()];
for (int y = row.getFirstCellNum(); y < row.getLastCellNum(); y++) {
cell = row.getCell(y);
title[y] = (String) getCellValue(cell);
}
} else
continue;
//log.info(JSON.toJSONString(title));
// 遍历当前sheet中的所有行
for (int j = 1; j < sheet.getLastRowNum() + 1; j++) {
row = sheet.getRow(j);
Map<String, Object> m = new HashMap<String, Object>();
// 遍历所有的列
//System.out.println("行："+j);
boolean flag = true;//判断当前格子是否没有任何数据
for (int y = row.getFirstCellNum(); y < row.getLastCellNum(); y++) {
cell = row.getCell(y);
if(cell==null){//格子是空的就创建一个，并填充空的富文本内容
row.createCell(y).setCellValue(getRichText("",fileName));
cell = row.getCell(y);
}
String key = title[y];
// log.info(JSON.toJSONString(key));
//System.out.println(title[y]+"\t\t列："+y);
m.put(mapping.get(key), getCellValue(cell));
if(!"".equals(getCellValue(cell).toString())){
flag = false;
}
}
if(flag){
throw new Exception("当前表格中有空行，请删除后再重新导入！");
}
ls.add(m);
}
}
work.close();
return ls;
}
/**
* 根据文件后缀，自适应创建富文本
* @param inStr
* @param fileName
* @return
* @throws Exception
*/
public static RichTextString getRichText(String inStr,String fileName) throws Exception{
RichTextString rs = null;
String fileType = fileName.substring(fileName.lastIndexOf("."));
if (excel2003L.equals(fileType)) {
rs = new HSSFRichTextString(inStr); // 2003-
} else if (excel2007U.equals(fileType)) {
rs = new XSSFRichTextString(inStr); // 2007+
} else {
throw new Exception("请选择正确的excel文件！");
}
return rs;
}
/**
* 描述：根据文件后缀，自适应上传文件的版本
*
* @param inStr
* ,fileName
* @return
* @throws Exception
*/
public static Workbook getWorkbook(InputStream inStr, String fileName) throws Exception {
Workbook wb = null;
String fileType = fileName.substring(fileName.lastIndexOf("."));
if (excel2003L.equals(fileType)) {
wb = new HSSFWorkbook(inStr); // 2003-
} else if (excel2007U.equals(fileType)) {
wb = new XSSFWorkbook(inStr); // 2007+
} else {
throw new Exception("请选择正确的excel文件！");
}
return wb;
}
/**
* 描述：对表格中数值进行格式化
*
* @param cell
* @return
*/
public static Object getCellValue(Cell cell) {
Object value = null;
DecimalFormat df = new DecimalFormat("0"); // 格式化number String字符
SimpleDateFormat sdf = new SimpleDateFormat("yyy-MM-dd"); // 日期格式化
DecimalFormat df2 = new DecimalFormat("0"); // 格式化数字
// if(cell==null){
// return null;
// }
switch (cell.getCellType()) {
case Cell.CELL_TYPE_STRING:
value = cell.getRichStringCellValue().getString();
break;
case Cell.CELL_TYPE_NUMERIC:
if ("General".equals(cell.getCellStyle().getDataFormatString())) {
value = df.format(cell.getNumericCellValue());
} else if ("m/d/yy".equals(cell.getCellStyle().getDataFormatString())) {
value = sdf.format(cell.getDateCellValue());
} else {
value = df2.format(cell.getNumericCellValue());
}
break;
case Cell.CELL_TYPE_BOOLEAN:
value = cell.getBooleanCellValue();
break;
case Cell.CELL_TYPE_BLANK:
value = "";
break;
default:
break;
}
return value;
}
public static void main(String[] args) throws Exception {
File file = new File("D:\\2.xlsx");
FileInputStream fis = new FileInputStream(file);
Map<String, String> m = new HashMap<String, String>();
m.put("IP","IP");
m.put("Vlan","Vlan");
m.put("Computer Name *","ComputerName");
m.put("Mac Address *","MacAddress");
m.put("Inventory No. *","InventoryNo");
m.put("Type * 类型，是否PLC","Type");
m.put("OS version * 系统版本","OSversion");
m.put("Area*地点","Area");
m.put("Machine Description设备描述","MachineDescription");
m.put("Sepecial Softwares*特殊软件","SepecialSoftwares");
m.put("Responsible*负责人","Responsible");
m.put("Section部门","Section");
m.put("Antivirus*是否安装防病毒","Antivirus");
m.put("Patch updated*是否打补丁","PatchUpdated");
m.put("Secure*是否安全","Secure");
m.put("Block USB","BlockUSB");
m.put("Block CD-ROM","BlockCD-ROM");
m.put("A-SC 可用性","A-SC");
m.put("C-SC 机密性","C-SC");
m.put("I-SC 完整性","I-SC");
m.put("Emergency Plan紧急恢复计划","EmergencyPlan");
m.put("备注","Remark");
List<Map<String, Object>> ls = parseExcel(fis, file.getName(), m);
System.out.println(JSON.toJSONString(ls));
for(int i=0;i<ls.size();i++){
Map<String,Object> map = ls.get(i);
map.get("IP");
System.out.println(map.get("IP"));
}
}
}
来源：oschina
链接：https://my.oschina.net/u/4345031/blog/4170104