2019独角兽企业重金招聘Python工程师标准>>>
最近工作需要一个读取excel文件的工具,结合网络资源写了一个工具类,初步实现了功能。好,老样子,先描述一下业务需求。(基于JFinal2.1版本)
分析了这个简单的需求,其实要做的事情还很多,下面就分步骤动手开始做:
读取excel数据
因为要写数据库,所以要按map这种键值对来匹配读取的数据
处理读取的数据
写库
读取excel文件数据
(看不明白没关系,先往下看)
/**
* 读取Excel2003-2007文件 .xls
* @param path
* @return
* @throws Exception
*/
public List
设置key
public String[] getHead(){
String[] heads = new String[]{"code","name","cate","price","hot","pop","remark"};
return heads;
}
这里有必要解释一下为啥这么做
首先回头看约定好的excel文件内容,在插入数据库时第一列的【序号】是不需要写库的,只要将之后的每列数据匹配好就行,但是标题是中文不能直接用,因此我们就写上面这个方法来保存我们要用到的key,基本上我们需要拿到这样的数据:
{"code":"0001","name":"牛排","cate":"招牌菜","price":"22.00","hot":"1","pop":"1","remark":"隆重推荐"};
key就是heads数组的值,value就是我们读取的每一列数据。
处理读取的数据
private String getValue(XSSFCell xssfRow) {
if (xssfRow.getCellType() == xssfRow.CELL_TYPE_BOOLEAN) {
return String.valueOf(xssfRow.getBooleanCellValue());
} else if (xssfRow.getCellType() == xssfRow.CELL_TYPE_NUMERIC) {
return String.valueOf(xssfRow.getNumericCellValue());
} else {
return String.valueOf(xssfRow.getStringCellValue());
}
}
好,我们先把所做代码整体放出来
package com.feng.excel;
import java.io.FileInputStream;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import com.jfinal.kit.JsonKit;
public class ReadExcel3 {
public static final String EXCEL2003 = "xls";
public static final String EXCEL2010 = "xlsx";
public static final String EMPTY = "";
public static String getFix(String path){
if(path.trim()!=null){
return path.substring(path.lastIndexOf(".")+1, path.length());
}
return EMPTY;
}
public String[] getHead(){
String[] heads = new String[]{"code","name","cate","price","hot","pop","remark"};
return heads;
}
/**
* 读取Excel2010文件 .xlsx
* @param path
* @return
* @throws Exception
*/
public List> readXlsx(String path) throws Exception{
InputStream is = new FileInputStream(path);
//创建对象
XSSFWorkbook xssfWorkbook = new XSSFWorkbook(is);
List> lists = new ArrayList>();
//获取sheet工作表
for(XSSFSheet xssfSheet : xssfWorkbook){
if(xssfSheet == null){
continue;
}
//遍历除标题外所有的行
for(int rowNum=1;rowNum<=xssfSheet.getLastRowNum();rowNum++){
XSSFRow row = xssfSheet.getRow(rowNum);
if(row == null){
throw new Exception("第"+ (rowNum+1) +"行有空行");
}
int minColIndex = row.getFirstCellNum();
int maxColIndex = row.getLastCellNum();
getListMapBy2010(lists, row, minColIndex, maxColIndex);
}
System.out.println(JsonKit.toJson(lists));
}
return lists;
}
/**
* 读取Excel2003-2007文件 .xls
* @param path
* @return
* @throws Exception
*/
public List> readXls(String path) throws Exception{
InputStream is = new FileInputStream(path);
//创建对象
HSSFWorkbook hssfWorkbook = new HSSFWorkbook(is);
List> lists = new ArrayList>();
//获取sheet工作表
for(int numSheet=0;numSheet> lists, HSSFRow hssfRow, int minColIndex, int maxColIndex) {
Map map = new HashMap();
String[] heads = getHead();
for(int colIndex=minColIndex;colIndex> lists, XSSFRow xssfRow, int minColIndex, int maxColIndex){
Map map = new HashMap();
String[] heads = getHead();
for(int colIndex=minColIndex;colIndex
虽然以上代码已经实现了读取excel文件数据,但是看到一些重复的代码,是不是很恶心?OK,那么我们就开始重构他,让这些代码更简洁!