Java常用工具类封装——封装对excel的操作

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.text.DecimalFormat;
import java.util.LinkedList;
import java.util.List;

import javax.servlet.http.HttpServletResponse;

import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFRichTextString;
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.openxml4j.exceptions.InvalidFormatException;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.DateUtil;
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.ss.usermodel.WorkbookFactory;

/**
* 封装对excel的操作，包括本地读写excel和流中输出excel,支持office 2007。<br/>
* 依赖于poi-3.9-20121203.jar,poi-ooxml-3.9-20121203.jar,poi-ooxml-schemas-3.9-
* 20121203.jar,dom4j-1.6.1.jar<br/>
* 有参构造函数参数为excel的全路径<br/>
*
* @author jiangshuai
* @date 2017年04月24日
*/
public class ExcelUtil {

// excel文件路径
private String path = "";

// 写入excel时，是否自动扩展列宽度来符合内容。
private boolean autoColumnWidth = false;

/**
* 无参构造函数 默认
*/
public ExcelUtil() {
}

/**
* 有参构造函数
*
* @param path
*            excel路径
*/
public ExcelUtil(String path) {
this.path = path;
}

/**
* 读取某个工作簿上的所有单元格的值。
*
* @param sheetOrder
*            工作簿序号，从0开始。
* @return List<Object[]> 所有单元格的值。
* @throws IOException
*             加载excel文件IO异常。
* @throws FileNotFoundException
*             excel文件没有找到异常。
* @throws InvalidFormatException
* @author jiangshuai
* @date 2017年04月24日
*/
public List<Object[]> read(int sheetOrder) throws FileNotFoundException,
IOException, InvalidFormatException {
FileInputStream fis = new FileInputStream(path);
Workbook workbook = WorkbookFactory.create(fis);
if (fis != null) {
fis.close();
}
Sheet sheet = workbook.getSheetAt(sheetOrder);
// 用来记录excel值
List<Object[]> valueList = new LinkedList<Object[]>();
// 循环遍历每一行、每一列。
for (Row row : sheet) {
// 每一行
Object[] rowObject = null;
for (Cell cell : row) {
// cell.getCellType是获得cell里面保存的值的type
switch (cell.getCellType()) {
case Cell.CELL_TYPE_BOOLEAN:
// 得到Boolean对象的方法
rowObject = CollectionUtil.addObjectToArray(rowObject,
cell.getBooleanCellValue());
break;
case Cell.CELL_TYPE_NUMERIC:
// 先看是否是日期格式
if (DateUtil.isCellDateFormatted(cell)) {
// 读取日期格式
rowObject = CollectionUtil.addObjectToArray(rowObject,
cell.getDateCellValue());
} else {
DecimalFormat df = new DecimalFormat();
// 单元格的值,替换掉,
String value = df.format(cell.getNumericCellValue())
.replace(",", "");
// 读取数字
rowObject = CollectionUtil.addObjectToArray(rowObject,
value);
}
break;
case Cell.CELL_TYPE_FORMULA:
// 读取公式
rowObject = CollectionUtil.addObjectToArray(rowObject,
cell.getCellFormula());
break;
case Cell.CELL_TYPE_STRING:
// 读取String
rowObject = CollectionUtil.addObjectToArray(rowObject, cell
.getRichStringCellValue().toString());
break;
}
}
// 将这行添加到list。
valueList.add(rowObject);
}
return valueList;
}

/**
* 读取某个工作簿上的某个单元格的值。
*
* @param sheetOrder
*            工作簿序号,从0开始。
* @param colum
*            列数 从1开始
* @param row
*            行数 从1开始
* @return 单元格的值。
* @throws Exception
*             加载excel异常。
* @author jiangshuai
* @date 2017年04月24日
*/
public String read(int sheetOrder, int colum, int row) throws Exception {
FileInputStream fis = new FileInputStream(path);
Workbook workbook = WorkbookFactory.create(fis);
if (fis != null) {
fis.close();
}
Sheet sheet = workbook.getSheetAt(sheetOrder);
Row rows = sheet.getRow(row - 1);
Cell cell = rows.getCell(colum - 1);
String content = cell.getStringCellValue();
return content;
}

/**
* 在指定的工作簿、行、列书写值。
*
* @param sheetOrder
*            工作簿序号，基于0.
* @param colum
*            列 基于1
* @param row
*            行 基于1
* @param content
*            将要被书写的内容。
* @throws Exception
*             书写后保存异常。
* @author jiangshuai
* @date 2017年04月24日
*/
public void write(int sheetOrder, int colum, int row, String content)
throws Exception {
FileInputStream fis = new FileInputStream(path);
Workbook workbook = WorkbookFactory.create(fis);
if (fis != null) {
fis.close();
}
Sheet sheet = workbook.getSheetAt(sheetOrder);
Row rows = sheet.createRow(row - 1);
Cell cell = rows.createCell(colum - 1);
cell.setCellValue(content);
FileOutputStream fileOut = new FileOutputStream(path);
workbook.write(fileOut);
fileOut.close();

}

/**
* 得到一个工作区最后一条记录的序号，相当于这个工作簿共多少行数据。
*
* @param sheetOrder
*            工作区序号
* @return int 序号。
* @throws IOException
*             根据excel路径加载excel异常。
* @throws InvalidFormatException
* @author jiangshuai
* @date 2017年04月24日
*/
public int getSheetLastRowNum(int sheetOrder) throws IOException,
InvalidFormatException {
FileInputStream fis = new FileInputStream(path);
Workbook workbook = WorkbookFactory.create(fis);
if (fis != null) {
fis.close();
}
Sheet sheet = workbook.getSheetAt(sheetOrder);
return sheet.getLastRowNum();
}

/**
* 在磁盘生成一个含有内容的excel,路径为path属性
*
* @param sheetName
*            导出的sheet名称
* @param fieldName
*            列名数组
* @param data
*            数据组
* @throws IOException
* @author jiangshuai
* @date 2017年04月24日
*/
public void makeExcel(String sheetName, String[] fieldName,
List<Object[]> data) throws IOException {
// 在内存中生成工作薄
HSSFWorkbook workbook = makeWorkBook(sheetName, fieldName, data);
// 截取文件夹路径
String filePath = path.substring(0, path.lastIndexOf("\\"));
// 如果路径不存在，创建路径
File file = new File(filePath);
// System.out.println(path+"-----------"+file.exists());
if (!file.exists())
file.mkdirs();
FileOutputStream fileOut = new FileOutputStream(path);
workbook.write(fileOut);
fileOut.close();
}

/**
* 在输出流中导出excel。
*
* @param excelName
*            导出的excel名称 包括扩展名
* @param sheetName
*            导出的sheet名称
* @param fieldName
*            列名数组
* @param data
*            数据组
* @param response
*            response
* @throws IOException
*             转换流时IO错误
* @author jiangshuai
* @date 2017年04月24日
*/
public void makeStreamExcel(String excelName, String sheetName,
String[] fieldName, List<Object[]> data,
HttpServletResponse response) throws IOException {
OutputStream os = null;
response.reset(); // 清空输出流
os = response.getOutputStream(); // 取得输出流
response.setHeader("Content-disposition", "attachment; filename="
+ new String(excelName.getBytes(), "ISO-8859-1")); // 设定输出文件头
response.setContentType("application/msexcel"); // 定义输出类型
// 在内存中生成工作薄
HSSFWorkbook workbook = makeWorkBook(sheetName, fieldName, data);
os.flush();
workbook.write(os);
}

/**
* 根据条件，生成工作薄对象到内存。
*
* @param sheetName
*            工作表对象名称
* @param fieldName
*            首列列名称
* @param data
*            数据
* @return HSSFWorkbook
* @author jiangshuai
* @date 2017年04月24日
*/
private HSSFWorkbook makeWorkBook(String sheetName, String[] fieldName,
List<Object[]> data) {
// 用来记录最大列宽,自动调整列宽。
Integer collength[] = new Integer[fieldName.length];

// 产生工作薄对象
HSSFWorkbook workbook = new HSSFWorkbook();
// 产生工作表对象
HSSFSheet sheet = workbook.createSheet();
// 为了工作表能支持中文,设置字符集为UTF_16
workbook.setSheetName(0, sheetName);
// 产生一行
HSSFRow row = sheet.createRow(0);
// 产生单元格
HSSFCell cell;
// 写入各个字段的名称
for (int i = 0; i < fieldName.length; i++) {
// 创建第一行各个字段名称的单元格
cell = row.createCell((short) i);
// 设置单元格内容为字符串型
cell.setCellType(HSSFCell.CELL_TYPE_STRING);
// 为了能在单元格中输入中文,设置字符集为UTF_16
// cell.setEncoding(HSSFCell.ENCODING_UTF_16);
// 给单元格内容赋值
cell.setCellValue(new HSSFRichTextString(fieldName[i]));
// 初始化列宽
collength[i] = fieldName[i].getBytes().length;
}
// 临时单元格内容
String tempCellContent = "";
// 写入各条记录,每条记录对应excel表中的一行
for (int i = 0; i < data.size(); i++) {
Object[] tmp = data.get(i);
// 生成一行
row = sheet.createRow(i + 1);
for (int j = 0; j < tmp.length; j++) {
cell = row.createCell((short) j);
// 设置单元格字符类型为String
cell.setCellType(HSSFCell.CELL_TYPE_STRING);
tempCellContent = (tmp[j] == null) ? "" : tmp[j].toString();
cell.setCellValue(new HSSFRichTextString(tempCellContent));

// 如果自动调整列宽度。
if (autoColumnWidth) {
if (j >= collength.length) { // 标题列数小于数据列数时。
collength = CollectionUtil.addObjectToArray(collength,
tempCellContent.getBytes().length);
} else {
// 如果这个内容的宽度大于之前最大的，就按照这个设置宽度。
if (collength[j] < tempCellContent.getBytes().length) {
collength[j] = tempCellContent.getBytes().length;
}
}
}
}
}

// 自动调整列宽度。
if (autoColumnWidth) {
// 调整列为这列文字对应的最大宽度。
for (int i = 0; i < fieldName.length; i++) {
sheet.setColumnWidth(i, collength[i] * 2 * 256);
}
}
return workbook;
}

/**
* 功能：设置写入excel时，是否自动扩展列宽度来符合内容，默认为false。
*
* @author jiangshuai
* @date 2017年04月24日
* @param autoColumnWidth
*            true或者false
*/
public void setAutoColumnWidth(boolean autoColumnWidth) {
this.autoColumnWidth = autoColumnWidth;
}
}