上传excel文件,读取内容,增加事务写入数据库

package com.inspur.icpmg.itss.asset.dao.impl;

import com.inspur.icpmg.util.DBHelper;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
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 org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.transaction.TransactionStatus;
import org.springframework.transaction.support.DefaultTransactionDefinition;

import java.io.*;

/**
 * Create by wjup on 2019/5/31 17:00
 */
public class ImportExcel {
    /**
     * 上传的excel文件和文件名称
     *
     * form表单上传框name为acUpload
     * acUploadFileName:xxFileName格式自动获取上传的文件名
     *
     */
    private File acUpload;
    private String acUploadFileName;

    public String getAcUploadFileName() {
        return acUploadFileName;
    }
    public void setAcUploadFileName(String acUploadFileName) {
        this.acUploadFileName = acUploadFileName;
    }
    public File getAcUpload() {
        return acUpload;
    }
    public void setAcUpload(File acUpload) {
        this.acUpload = acUpload;
    }
    private static JdbcTemplate jdbcTemplate = new JdbcTemplate(DBHelper.getDataSource());// 数据库连接池

    public void file() throws IOException {
        InputStream in = new FileInputStream(acUpload);
        if (acUploadFileName.endsWith("xls") || acUploadFileName.endsWith("xlsx")) {
            Workbook wb = null;
            if (acUploadFileName.endsWith("xls")) {
                //2003
                wb = new HSSFWorkbook(in);
            } else if (acUploadFileName.endsWith("xlsx")) {
                //2007
                wb = new XSSFWorkbook(in);
            }
            // 获取excel表第一张sheet页
            Sheet sheet1 = wb.getSheetAt(0);
            importExcel(sheet1);
        }
    }


    public void importExcel(Sheet sheet) {
        //添加事务
        DefaultTransactionDefinition def = new DefaultTransactionDefinition();
        DataSourceTransactionManager transactionManager = new DataSourceTransactionManager(jdbcTemplate.getDataSource());
        TransactionStatus status = transactionManager.getTransaction(def);
        Boolean Iscommit = false;
        String name = "";

        for (Row row : sheet) {
            if (row.getRowNum() >= 1) {
                boolean flag = false;
                for (int c = row.getFirstCellNum(); c < row.getLastCellNum(); c++) {
                    if (row.getCell(c) != null && row.getCell(c).getCellType() != HSSFCell.CELL_TYPE_BLANK) {
                        flag = true;
                    }
                }
                if (flag) {
                    if (row.getCell(0) != null) {
                        name = row.getCell(0).toString().trim();
                    } else {
                        String msg = "不能为空";
                        Iscommit = true;
                    }
                }
            }
        }

        String sql = "insert into user values("+name+")";

        if (Iscommit) {
            transactionManager.rollback(status);
        } else {
            try {
                // 出现异常回滚事务
                jdbcTemplate.batchUpdate(sql);
            } catch (Exception ex) {
                transactionManager.rollback(status);
            } finally {
                transactionManager.commit(status);
            }
        }
    }


}

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值