Java 读取excel文件并导入数据库

&波吉&

已于 2022-10-16 11:24:03 修改

阅读量6.9k

点赞数 3

分类专栏： Spring 文章标签： java 数据库

于 2022-07-27 00:31:21 首次发布

本文链接：https://blog.csdn.net/weixin_48913290/article/details/126005640

版权

Spring 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

导入：

先导入依赖：

<!-- 文件上传 -->
<dependency>
    <groupId>org.apache.httpcomponents</groupId>
    <artifactId>httpmime</artifactId>
	<version>4.5.7</version>
</dependency>
<!-- JSON -->
<dependency>
    <groupId>com.alibaba</groupId>
    <artifactId>fastjson</artifactId>
    <version>1.2.41</version>
</dependency>
<!-- POI -->
<dependency>
    <groupId>org.apache.poi</groupId>
    <artifactId>poi-ooxml</artifactId>
    <version>3.16</version>
</dependency>

1、获取想要上传excel文件的路径

2、获得路径后，通过路径定位生成数据流，使用FileInputStream方法读取路径对应文件的原始字节流。

//根据路径生成 FileInputStream字节流
FileInputStream inputStream = new FileInputStream(new File(fileName));

3、使用ExcelUtils工具类读取字节流 + 路径。

//通过ExcelUtils工具将Excel数据存入到list中，工具代码下面细讲,这一步读取Excel已经完成了，如果不想进行插入数据库操作，可以直接拿着list用啦。
List<List<Object>> list = ExcelUtils.getListByExcel(inputStream,fileName);

4、ExcelUtils工具类对数据进行处理，因为表格有行和列，所以处理完数据使用二维集合List<List>对数据进行存储。

ExcelUtils类讲解：

public class ExcelUtils {
    private final static String excel2003L =".xls";    //2003- 版本的excel
    private final static String excel2007U =".xlsx";  //2007版本

    /**
     * @Description：获取IO流中的数据，组装成List<List<Object>>对象
     * @param in,fileName
     * @return
     * @throws
     */
    public static List<List<Object>> getListByExcel(InputStream in, String fileName) throws Exception{
        List<List<Object>> list = null;

        //创建Excel工作薄
        Workbook work = getWorkbook(in,fileName);
        if(null == work){
            throw new Exception("创建Excel工作薄为空！");
        }
        Sheet sheet = null;  //页数
        Row row = null;  //行数
        Cell cell = null;  //列数

        list = new ArrayList<List<Object>>();
        //遍历Excel中所有的sheet
        for (int i = 0; i < work.getNumberOfSheets(); i++) {
            sheet = work.getSheetAt(i);
            if(sheet==null){continue;}

            //遍历当前sheet中的所有行
            for (int j = sheet.getFirstRowNum(); j <= sheet.getLastRowNum(); j++) {
                row = sheet.getRow(j);
                if(row==null){continue;}

                //遍历所有的列
                List<Object> li = new ArrayList<Object>();
                for (int y = row.getFirstCellNum(); y < row.getLastCellNum(); y++) {
                    cell = row.getCell(y);
                    li.add(getValue(cell));
                }
                list.add(li);
            }
        }

        return list;

    }

    /**
     * @Description：根据文件后缀，自适应上传文件的版本
     * @param inStr,fileName
     * @return
     * @throws Exception
     */
    public static  Workbook getWorkbook(InputStream inStr,String fileName) throws Exception{
        Workbook wb = null;
        String fileType = fileName.substring(fileName.lastIndexOf("."));
        if(excel2003L.equals(fileType)){
            wb = new HSSFWorkbook(inStr);  //2003-
        }else if(excel2007U.equals(fileType)){
            wb = new XSSFWorkbook(inStr);  //2007+
        }else{
            throw new Exception("解析的文件格式有误！");
        }
        return wb;
    }

    /**
     * @Description：对表格中数值进行格式化
     * @param cell
     * @return
     */
    //解决excel类型问题，获得数值
    public static String getValue(Cell cell) {
        String value = "";
        if(null==cell){
            return value;
        }
        switch (cell.getCellType()) {
            //数值型
            case NUMERIC:
                if (DateUtil.isCellDateFormatted(cell)) {
                    //如果是date类型则 ，获取该cell的date值
                    Date date = DateUtil.getJavaDate(cell.getNumericCellValue());
                    SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd");
                    value = format.format(date);;
                }else {// 纯数字
                    BigDecimal big=new BigDecimal(cell.getNumericCellValue());
                    value = big.toString();
                    //解决1234.0  去掉后面的.0
                    if(null!=value&&!"".equals(value.trim())){
                        String[] item = value.split("[.]");
                        if(1<item.length&&"0".equals(item[1])){
                            value=item[0];
                        }
                    }
                }
                break;
            //字符串类型
            case STRING:
                value = cell.getStringCellValue();
                break;
            // 公式类型
            case FORMULA:
                //读公式计算值
                value = String.valueOf(cell.getNumericCellValue());
                if (value.equals("NaN")) {// 如果获取的数据值为非法值,则转换为获取字符串
                    value = cell.getStringCellValue();
                }
                break;
            // 布尔类型
            case BOOLEAN:
                value = " "+ cell.getBooleanCellValue();
                break;
            default:
                value = cell.getStringCellValue();
        }
        if("null".endsWith(value.trim())){
            value="";
        }
        return value;
    }
}

5、已经获取到了Excel表格中所有的数据，包含所有的行和列，都在List<List>中，这一步所有的数据获取完了

6、有了数据，要将数据存储到数据库。

7、需要创建数据库字段的实体类，使用@Excel注解将表格和数据库字段名一一对应

@Data
@TableName("t_student")//数据库名字
public class Student{

        @TableId(type = IdType.AUTO)
        @Excel(name = "序号", cellType = Excel.ColumnType.NUMERIC)
        private Long id;

        @Excel(name = "学生ID")
        private Long id;

        @Excel(name = "名称")
        private String name;

        @Excel(name = "性别")
        private String sex;

        @Excel(name = "年龄")
        private String age;

      
        
}

8、通过双重 for循环遍历List<List>,内层List为列，外层List为行，如下图将第一行数据单独拿出来，作为判断字段，如果实体类等@Excel注解属性和Excel表格第一行名字相同，对实体类进行set赋值

//定义Excel第一行的属性
List<Object> firstRows = null;
//获取第一行属性 放入firstRows中。
if(list != null && list.size() > 0){
	firstRows = list.get(0);
}
//遍历除第一行以外的Excel表格中的值
for (int i = 1; i < list.size(); i++) {
    //rows是某一行，i = 1 为第二行，	i = 2 为第三行
    List<Object> rows = list.get(i);
    //数据库数据的实体类
    Student student =new student();
    //遍历这一行所有的值
    for (int j = 0; j < rows.size(); j++){
    //某一行的某一列 j为列的坐标    
    String cellVal = (String) rows.get(j);
     //对实体类进行赋值,使用setFieldValueByFieldName方法，下面详细讲代码
    TestExcel.setFieldValueByFieldName(questionBank, firstRows.get(j).toString().trim(), 	 cellVal);
                    }

                }
     // 使用spring插入到数据库
     questionBankService.save(questionBank);
            }

9、从第二行开始，每遍历完一行，将获得一个实体类对象，因为是结合数据库写的实体类，通过springboot可以直接插入数据库。

public class TestExcel {

    /**
     * 用反射方法，获取列名，比较，为属性赋值
     * @param object
     * @param fieldName
     * @param val
     */
    public static void setFieldValueByFieldName(Object object, String fieldName, Object val) {
        try {
            //通过反射获取所有实体类所有定义的方法,object为数据库实体类
            Field[] fields = object.getClass().getDeclaredFields();
            //遍历fields
            for (int i = 0; i < fields.length; i++) {
                Field field = fields[i];
                //读取注释，如@Excel(name = "科目")
                Excel annotation = field.getAnnotation(Excel.class);
                //因为private类型，所有要设置允许
                field.setAccessible(true);
                //annotation.name() 注解里的属性,如果annotation没写注解，读取原始名字如subject
                if(annotation == null){
                    if(fieldName.equals(field.getName())){
                        if(field.getType() == Integer.class){
                            field.set(object, Integer.valueOf(val.toString()));
                        }else if(field.getType() == Long.class){
                            field.set(object, Long.valueOf(val.toString()));
                        }else {
                            field.set(object, val);
                        }
                        return;
                    }  
                }else {//设置了注解，并且表格里的Excel字段值和注解的name值相同，则为相应字段赋值
                    if(fieldName.equals(annotation.name())){
                        //进行类型判断，因为实体类中变量类型不同。
                        if(field.getType() == Integer.class){
                            field.set(object, Integer.valueOf(val.toString()));
                        }else if(field.getType() == Long.class){
                            field.set(object, Long.valueOf(val.toString()));
                        }else {
                            field.set(object, val);
                        }
                        return;
                    }
                }

            }
        }catch (Exception e){
            e.printStackTrace();
        }
    }

}

10、通过读取成功插入到数据库