Spring JdbcTemplate批量操作数据库实例

Spring JdbcTemplate使用实例
Spring NamedParameterJdbcTemplate的使用实例
前面我们用实例演示了如何使用JdbcTemplate和NamedParameterJdbcTemplate来对数据库的增删查改,但是那些实例中都是针对单条记录进行操作,当我们需要对一批数据进行相同的SQL操作该怎么办呢?总不能用for循环来遍历n次吧,那也太低效了。

JdbcTemplate和NamedParameterJdbcTemplate当然支持对数据库的批量操作了,通过以下实例你将了解到如何相对高效地执行批量的SQL语句。

首先,我们需要在service的实现类中声明两个jdbcTemplate,后面需要使用它们进行操作。

    @Autowired
    private JdbcTemplate jdbcTemplate;
    @Autowired
    private NamedParameterJdbcTemplate template;

现在我们针对JdbcTemplate来演示如何进行数据库SQL的批量操作:

    @Override
    public int[] batchUpdateGradeByName(List students) {
        String sql = "update test_student set grade = ? where name = ?";
        return this.jdbcTemplate.batchUpdate(sql, new BatchPreparedStatementSetter() {
            @Override
            public void setValues(PreparedStatement ps, int i) throws SQLException {
                ps.setString(1, students.get(i).getGrade());
                ps.setString(2, students.get(i).getName());
            }

            @Override
            public int getBatchSize() {
                return students.size();
            }
        });
    }
    @Test
    public void batchUpdateGradeByName() {
        Student s1 = new Student();
        s1.setGrade("6");
        s1.setName("李白");
        Student s2 = new Student();
        s2.setGrade("4");
        s2.setName("杜甫");

        List studentList = new ArrayList<>();
        studentList.add(s1);
        studentList.add(s2);

        int[] nums = this.studentService.batchUpdateGradeByName(studentList);
        assertEquals(new Integer("2"), new Integer(nums[0]));
        assertEquals(new Integer("1"), new Integer(nums[1]));
    }

在以上的代码中,使用jdbcTemplate.batchUpdate时,第二个参数是一个BatchPreparedStatementSetter的匿名内部类实例,这个实例中我们必须要实现setValuesgetBatchSize方法,其作用分别是替换SQL中占位符的值,以及获取迭代执行次数。

以上方法可能会有些繁琐,当然jdbcTemplate.batchUpdate不止这一种使用方法,其第二个参数也能支持使用List,下面是对应的代码实例:

    @Override
    public int[] batchUpdateGradeByName(List student) {
        String sql = "update test_student set grade = ? where name = ?";
        List parameterList = new ArrayList<>();
        for(Student s : student){
            Object[] params = new Object[]{s.getGrade(), s.getName()};
            parameterList.add(params);
        }

        return this.jdbcTemplate.batchUpdate(sql, parameterList);
    }

在看完JdbcTemplate的使用方法后,我们再来看看使用NamedParameterJdbcTemplate的操作:

    @Override
    public int[] batchUpdateGradeByName(List student) {
        String sql = "update test_student set grade = :grade where name = :name";
        SqlParameterSource[] parameterSources = SqlParameterSourceUtils.createBatch(student);
        return this.template.batchUpdate(sql, parameterSources);
    }

这里我们使用了SqlParameterSource来存放参数值,代码量明显少了很多,值得推荐。

最后一个例子来演示下,倘若我们有海量的数据需要批量操作到数据库的场景。虽说JdbcTemplate和NamedParameterJdbcTemplate都支持批量操作,但是也扛不住千、万级别的数据同时更新呀,除非再次使用for循环,拆成若干次来更新?可以是可以,但是也太low了,其实JdbcTemplate内部是支持指定数据量来分多次批量更新的。

    @Override
    public int[][] batchUpdateGradeByName(List student) {
        String sql = "update test_student set grade = ? where name = ?";
        return this.jdbcTemplate.batchUpdate(sql, student, 100, new ParameterizedPreparedStatementSetter() {
            @Override
            public void setValues(PreparedStatement ps, Student student) throws SQLException {
                ps.setString(1, student.getGrade());
                ps.setString(2, student.getName());
            }
        });
    }

最后说明一点,这些操作返回的都是int类型的数组,代表着这批量执行过程中,每一条SQL影响的记录数,最后一个例子是分次批量执行,因此返回的是一个int类型的二维数组。

总结下来,使用JdbcTemplate和NamedParameterJdbcTemplate操作数据库确实不像ORM框架那样简洁便利,估计真实生产上用的人会很少吧。但无论如何,这是学习Spring必不可少的一个环节,了解一下还是很有必要的。

你可能感兴趣的:(Spring JdbcTemplate批量操作数据库实例)