工厂模式 实现读取excel工具类

package com.**.cpic.util.excel;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;

import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
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.XSSFWorkbook;

import com.alibaba.fastjson.JSONObject;
import com.**.cpic.entity.Staff;

import lombok.extern.slf4j.Slf4j;

@Slf4j
public class ExcelUtil {

	public List<ExcelRowSerializable> read(File file,RowBeanFactory<ExcelRowSerializable> beanFactory) {

		Workbook workbook = getWorkbook(file);
		Sheet sheet = workbook.getSheetAt(0);
		// 获取文档中已保存数据的行数
		int rowNum = sheet.getPhysicalNumberOfRows();
		// 获取第一行
		Row row = sheet.getRow(0);
		// 获取当前行已保存数据的最大列数
		int colnum = row.getPhysicalNumberOfCells();
		
		List<ExcelRowSerializable> datas = new ArrayList<>(rowNum-1);
		
		for (int i = 1; i < rowNum; i++) {
			row = sheet.getRow(i);
			if (null != row) {
				ExcelRowSerializable rowBean = beanFactory.createBean();//创建对象
				datas.add(rowBean);
				for (int j = 0; j < colnum; j++) {
					Cell cell = row.getCell(j);
					rowBean.write(j, cell.getStringCellValue());//根据序列化设置参数值
				}
			}
		}
		return datas;
	}

	private static Workbook getWorkbook(File file) {// 根据后缀获取Excel表格
		Workbook workbook = null;
		String fileName = file.getName();
		String suffix = fileName.substring(fileName.lastIndexOf(".") + 1);
		try (InputStream intpuStream = new FileInputStream(file)) {
			if ("xls".equals(suffix)) {
				workbook = new HSSFWorkbook(intpuStream);
			} else if ("xlsx".equals(suffix)) {
				workbook = new XSSFWorkbook(intpuStream);
			}
		} catch (FileNotFoundException e) {
			log.error("read excel error {}",e);
		} catch (IOException e) {
			log.error("read excel error {}",e);
		}
		return workbook;
	}

	public static void main(String args[]) {
		ExcelUtil util = new ExcelUtil();
		util.read(new File("C:\\workdoc","1.xlsx"),new RowBeanFactory<ExcelRowSerializable>() {
			@Override
			public ExcelRowSerializable createBean() {
				return new Staff();
			}
		}).forEach(staff->{
			Staff tempStaff = (Staff) staff;
			System.out.println(JSONObject.toJSONString(tempStaff));
		});
	}
}

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值