excel导入数据自动封装到bean对象

package io.renren;

import com.alibaba.fastjson.JSONObject;
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 java.io.File;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.Map;

/**
 * 导入excel封装数据到bean对象,模板要和javabean对应
 */
public class POITest {

    public static void main(String[] args) {
//        Class<TestDO> class = new TestDO().getClass();
        TestDO t = new TestDO();
        importUtils("F:\\test1.xlsx",t);

    }

    public static <T>T importUtils(String EXCEL_PATH, T t){
        Map<Integer, String> COLUMN_MAP = new HashMap<>();
        File file = new File(EXCEL_PATH);
        Workbook workbook = null;
        try {
            workbook = new XSSFWorkbook(file);

            int numberOfSheets = workbook.getNumberOfSheets();
            for (int i = 0; i < numberOfSheets; i++) {
                if (i > 0) {
                    break;
                }
                Sheet sheet = workbook.getSheetAt(i);
                int physicalNumberOfRows = sheet.getPhysicalNumberOfRows();
                for (int j = 0; j < physicalNumberOfRows; j++) {
                    Row row = sheet.getRow(j);
                    if (row != null) {
                        int physicalNumberOfCells = row.getPhysicalNumberOfCells();
                        if (j == 0) {
                            for (int k = 0; k < physicalNumberOfCells; k++) {
//                        String cellValue = (String) ExcelUtil.getCellValue(row, k);
                                String cellValue = String.valueOf(row.getCell(k));
                                TestEnum shoppingDetailEnum = TestEnum.get(cellValue);
                                COLUMN_MAP.put(k, shoppingDetailEnum.getFieldName());
                            }
                            continue;// 标题行
                        }

//                        TestDO shoppingDetailEntity = new TestDO();
                        Field declaredField = null;
                        for (int k = 0; k < physicalNumberOfCells; k++) {
                            String fieldName = COLUMN_MAP.get(k);
                            try {
                                declaredField = t.getClass().getDeclaredField(fieldName);
                            } catch (NoSuchFieldException ex) {
                                // 子类不存在该变量则从父类获取变量
//                        declaredField = shoppingDetailEntity.getClass().getSuperclass().getDeclaredField(fieldName);
                                ex.printStackTrace();
                            }

                            Object cellValue = row.getCell(k);
                            String firstLetter = fieldName.substring(0, 1).toUpperCase();
                            String setMethodName = "set" + firstLetter
                                    + fieldName.substring(1);
                            Method setMethod = t.getClass().getMethod(setMethodName,
                                    new Class[] { declaredField.getType() });

                            switch (declaredField.getType().toString()){
                                case "float":
                                    cellValue = Float.parseFloat(String.valueOf(cellValue));
                                    break;
                                case "double":
                                    cellValue = Double.parseDouble(String.valueOf(cellValue));
                                    break;
                                case "int":
                                    cellValue = Integer.parseInt(String.valueOf(cellValue));
                                    break;
                                case "class java.lang.String":
                                    cellValue = String.valueOf(cellValue);
                                    break;
                            }
                            setMethod.invoke(t, new Object[] { cellValue });//调用对象的setXXX方法
                        }
                        // 开始插入数据
                        System.out.println(JSONObject.toJSONString(t));
                    }
                }
            }
        } catch (Exception ex) {
            ex.printStackTrace();
        }
        return  t;
    }
}

===================================================================

package io.renren;

import lombok.Data;

@Data
public class TestDO {

    private String school;
    private String name;
    private int age;
    private float hign;
    private String sex;

}

==================================================================

package io.renren;

public enum  TestEnum {

    SCHOOL("学校", "school"),
    NAME("姓名", "name"),
    AGE("年龄", "age"),
    HIGN("身高", "hign"),
    SEX("性别", "sex");

    private String columnName;

    private String fieldName;

    TestEnum(String columnName, String fieldName) {
        this.columnName = columnName;
        this.fieldName = fieldName;
    }

    public static TestEnum get(String columnName) {
        TestEnum[] values = TestEnum.values();
        for (TestEnum shoppingDetailEnum : values) {
            if (shoppingDetailEnum.getColumnName().equals(columnName)) {
                return shoppingDetailEnum;
            }
        }
        return null;
    }

    public String getColumnName() {
        return columnName;
    }

    public String getFieldName() {
        return fieldName;
    }

}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值