Java 使用 Poi 导入 Excel 通用（一）

版权声明：转载请注明出处 https://blog.csdn.net/qq_40162735/article/details/84874862

上一篇【Java实现从网页上传导入excel数据到数据库】一文中利用poi 导入excel工具只是一个简版，今天利用反射机制重新写一个通用excel导入工具类，可供大家参考使用

实现思路：

       利用java反射机制把excel读取出来的值放进相对应的实体成员变量中

       工作簿 中的列必须与实体成员变量顺序和类型一致

核心代码：

import java.io.File;
import java.io.IOException;
import java.lang.reflect.Method;
import java.lang.reflect.ParameterizedType;
import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

import org.apache.commons.fileupload.disk.DiskFileItem;
import org.apache.commons.io.FileUtils;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFDateUtil;
import org.apache.poi.hssf.usermodel.HSSFFormulaEvaluator;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.formula.functions.T;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.FormulaEvaluator;
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.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFDataFormat;
import org.apache.poi.xssf.usermodel.XSSFFormulaEvaluator;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.commons.CommonsMultipartFile;

import com.mysql.jdbc.Field;

/**
 * @author DevinLiu
 * 通用excel导入数据库
 */
public class ExcelUtil {
	
    private final static String excel2003L =".xls";    //2003- 版本的excel  
    private final static String excel2007U =".xlsx";   //2007+ 版本的excel  


	/**
	 * 获取实体对象返回属性名称
	 * @param obj 实体对象
	 * @return
	 * @throws Exception
	 */
	public java.lang.reflect.Field[] findEntityAllTypeName(Object obj)throws Exception{

		Class<? extends Object> cls = obj.getClass();

		return cls.getDeclaredFields();
	}
	
	/**
	 * 根据文件选择excel版本
	 * @return
	 * @throws Exception 
	 */
	public Workbook chooseWorkbook(MultipartFile file) throws Exception{
		
			Workbook workbook = null;
		
			//把MultipartFile转化为File
			CommonsMultipartFile cmf = (CommonsMultipartFile)file;
			DiskFileItem dfi = (DiskFileItem) cmf.getFileItem();
			File fo = dfi.getStoreLocation();
			
			String filename = file.getOriginalFilename();
			String fileType = (filename.substring(filename.lastIndexOf("."), filename.length())).toLowerCase();
			
			if(excel2003L.equals(fileType)){  
       		 	    workbook = new HSSFWorkbook(FileUtils.openInputStream(fo));  //2003- 
                        }else if(excel2007U.equals(fileType)){  
                            workbook = new XSSFWorkbook(FileUtils.openInputStream(fo));  //2007+  
                        }else{  
                            throw new Exception("解析的文件格式有误！");  
                        } 
			return workbook;
	}


	/**
	 * 公共的导入excel方法
	 * @param file 文件
	 * @param sheetname 工作簿名称
	 * @param obj 实体类
	 * @return
	 * @throws IOException 
	 */
	public  List<Object> importBaseExcel(MultipartFile file,String sheetname,Object obj) throws IOException{

		Workbook workbook = null;

		try {
			//读取文件内容
			workbook = this.chooseWorkbook(file);

			//获取工作表
			Sheet sheet = workbook.getSheet(sheetname);

			//获取sheet中第一行行号
			int firstRowNum = sheet.getFirstRowNum();
			//获取sheet中最后一行行号
			int lastRowNum = sheet.getLastRowNum();

			//获取该实体所有定义的属性 返回Field数组
			java.lang.reflect.Field[] entityName = this.findEntityAllTypeName(obj);

			String classname =  obj.getClass().getName();
			Class<?> clazz = Class.forName(classname);

			List<Object> list = new ArrayList<Object>();

			//循环插入数据
			for(int i=firstRowNum+1;i<=lastRowNum;i++){

				Row row = sheet.getRow(i);

				//可以根据该类名生成Java对象
				Object pojo =  clazz.newInstance();

				//除自增编号外，实体字段匹配sheet列 
				for(int j = 0;j < entityName.length-3;j++){

					//获取属性的名字,将属性的首字符大写，方便构造set方法
					String name = "set"+entityName[j+1].getName().substring(0, 1).toUpperCase().concat(entityName[j+1].getName().substring(1).toLowerCase())+"";
					//获取属性的类型
					String type = entityName[j+1].getGenericType().toString();

					Method m = null;
					//getMethod只能调用public声明的方法，而getDeclaredMethod基本可以调用任何类型声明的方法
					m = obj.getClass().getDeclaredMethod(name,entityName[j+1].getType());
					
					Cell pname = row.getCell(j);
					//根据属性类型装入值
					switch (type) {
						case "char":
						case "java.lang.Character":
						case "class java.lang.String":
							m.invoke(pojo,getVal(pname));
							break;
						case "int":
						case "class java.lang.Integer":
							m.invoke(pojo,Integer.valueOf(getVal(pname)));
							break;
						case "class java.util.Date":
							m.invoke(pojo,getVal(pname));
							break;
						case "float":
						case "double":
						case "java.lang.Double":
						case "java.lang.Float":
						case "java.lang.Long":
						case "java.lang.Short":
						case "java.math.BigDecimal":
							m.invoke(pojo,Double.valueOf(getVal(pname)));
							break;
						default:
							break;
					}
				}
				list.add(pojo);
			}
			return list;
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			workbook.close();
		}
		return null;
	}

	/** 
	 * 处理类型
	 * @param cell 
	 * @return 
	 */  
	public static String getVal(Cell cell) {  
		if (null != cell) {

			switch (cell.getCellType()) {
			case XSSFCell.CELL_TYPE_NUMERIC: // 数字

				String val = cell.getNumericCellValue()+"";
				int index = val.indexOf(".");

				if(Integer.valueOf(val.substring(index+1)) == 0){
					DecimalFormat df = new DecimalFormat("0");//处理科学计数法
					return df.format(cell.getNumericCellValue());
				}
				return cell.getNumericCellValue()+"";//double
			case XSSFCell.CELL_TYPE_STRING: // 字符串
				return cell.getStringCellValue() + "";
			case XSSFCell.CELL_TYPE_BOOLEAN: // Boolean
				return cell.getBooleanCellValue() + "";
			case XSSFCell.CELL_TYPE_FORMULA: // 公式

				try{
					if(HSSFDateUtil.isCellDateFormatted(cell)){
						Date date = cell.getDateCellValue();
						return (date.getYear() + 1900) + "-" + (date.getMonth() + 1) +"-" + date.getDate();
					}else{
						return String.valueOf((int)cell.getNumericCellValue());
					}
				}catch (IllegalStateException e) {
					return  String.valueOf(cell.getRichStringCellValue());
				}
			case XSSFCell.CELL_TYPE_BLANK: // 空值
				return "";
			case XSSFCell.CELL_TYPE_ERROR: // 故障
				return "";
			default:
				return "未知类型   ";
			}
		} else {
			return "";
		}
	}

}

Java 使用 Poi 导出 Excel 通用（二）​​​​​​​