1. 程式人生 > 程式設計 >spring中使用mybatis實現批量插入的示例程式碼

spring中使用mybatis實現批量插入的示例程式碼

有3種實現方式:foreach,spring事務,以及ExecutorType.BATCH.

1. foreach方式

這種方式實際是對SQL語句進行拼接,生成一個長長的SQL,對很多變數進行繫結。如果資料量不大(1000個以內),可以用這種方式。如果資料量太大,可能資料庫會報錯。

定義介面

public interface StudentMapper05 {
  public void insertStudent(List<Student> studentList);
}

定義mapper

適用於Oracle資料庫

<insert id="insertStudent">
  BEGIN
  <foreach collection="list" item="student" index="index" separator="">
    INSERT INTO test_student(ID,NAME,BRANCH,PERCENTAGE,PHONE,EMAIL) 
    VALUES
    (SEQ_ID.nextval,#{student.name},#{student.branch},#{student.percentage},#{student.phone},#{student.email});
  </foreach>
  END;
</insert>

這個mapper的含義,就是把上送的studentList拼接成一個長SQL,拼成的SQL類似:

BEGIN
INSERT INTO test_student(ID,EMAIL) VALUES (SEQ_ID.nextval,?,?);
INSERT INTO test_student(ID,?);
...
END;

studentList有幾個,就會生成多少個insert語句拼接到一起,每個?都會進行變數繫結,所以當studentList中資料量較多時,生成的SQL會很長,導致資料庫執行報錯。

dao

public class StudentDao05 {
  private StudentMapper05 studentMapper; // 省略getter和setter
  
  public void insertStudentList(List<Student> studentList) {
    studentMapper.insertStudent(studentList);
  }
}

beans

mybatis-spring-05.xml:

<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
  <property name="dataSource" ref="oracleDataSource" />
  <property name="configLocation" value="classpath:mybatis/config/mybatis-config-05.xml"/>
</bean>
<bean id="studentMapper05" class="org.mybatis.spring.mapper.MapperFactoryBean">
  <property name="mapperInterface" value="com.ws.experiment.spring.mybatis.mapper.StudentMapper05" />
  <property name="sqlSessionFactory" ref="sqlSessionFactory" />
</bean>
<bean id="studentDao05" class="com.ws.experiment.spring.mybatis.dao.StudentDao05">
  <property name="studentMapper" ref="studentMapper05" />
</bean>

main函式

public static void main(String[] args) {
  String[] configFiles = new String[]{"spring-beans-config.xml","mybatis/mybatis-spring-05.xml"};  // 分別配置datasource和mybatis相關bean
  ApplicationContext context = new ClassPathXmlApplicationContext(configFiles);
  
  StudentDao05 studentDao = (StudentDao05)context.getBean("studentDao05");
  
  int counts[] = new int[]{10,50,100,200,500,1000,2000,3000,5000,8000};
  for (int count : counts) {
    List<Student> studentList = new ArrayList<>();
    for (int i = 0; i < count; i++) {
      Student st = new Student();
      st.setName("name");
      st.setBranch("");
      st.setEmail("");
      st.setPercentage(0);
      st.setPhone(0);
      studentList.add(st);
    }
    long startTime = System.currentTimeMillis();
    studentDao.insertStudentList(studentList);
    long endTime = System.currentTimeMillis();
    System.out.println("插入" + count + "筆資料耗時: " + (endTime - startTime) +" ms");
  }
}

測試結果

插入100筆資料耗時: 197 ms
插入200筆資料耗時: 232 ms
插入500筆資料耗時: 421 ms
插入1000筆資料耗時: 650 ms
插入2000筆資料耗時: 1140 ms
插入3000筆資料耗時: 27113 ms
插入5000筆資料耗時: 98213 ms
插入8000筆資料耗時: 301101 ms

2. 藉助spring事務

藉助spring事務,插入一組資料

開啟spring事務

<bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
  <property name="dataSource" ref="oracleDataSource" />
</bean>

<tx:annotation-driven transaction-manager="transactionManager" />

定義介面

public interface StudentMapper06 {
  public void insertStudent(@Param("student") Student student);
}

mapper

<insert id="insertStudent">
  INSERT INTO test_student(ID,EMAIL) 
  VALUES
  (SEQ_ID.nextval,#{student.email})
</insert>

dao

public class StudentDao06 {
  private StudentMapper06 studentMapper; // 省略getter和setter
  
  @Transactional // spring事務控制
  public void insertStudentList(List<Student> students) {
    for (Student student : students) {
      studentMapper.insertStudent(student);
    }
  }
}

beans

<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
  <property name="dataSource" ref="oracleDataSource" />
  <property name="configLocation" value="classpath:mybatis/config/mybatis-config-06.xml"/>
</bean>
<bean id="studentMapper06" class="org.mybatis.spring.mapper.MapperFactoryBean">
  <property name="mapperInterface" value="com.ws.experiment.spring.mybatis.mapper.StudentMapper06" />
  <property name="sqlSessionFactory" ref="sqlSessionFactory" />
</bean>
<bean id="studentDao06" class="com.ws.experiment.spring.mybatis.dao.StudentDao06">
  <property name="studentMapper" ref="studentMapper06" />
</bean>

main

測試結果

batchInsert001插入10筆資料耗時: 602 ms
batchInsert001插入50筆資料耗時: 196 ms
batchInsert001插入100筆資料耗時: 284 ms
batchInsert001插入200筆資料耗時: 438 ms
batchInsert001插入500筆資料耗時: 944 ms
batchInsert001插入1000筆資料耗時: 1689 ms
batchInsert001插入2000筆資料耗時: 3138 ms
batchInsert001插入3000筆資料耗時: 4427 ms
batchInsert001插入5000筆資料耗時: 7368 ms
batchInsert001插入8000筆資料耗時: 11832 ms

3. 使用ExecutorType.BATCH

基本原理是SqlSession sqlSession = sqlSessionFactory.openSession(ExecutorType.BATCH,false);,設定BATCH方式的sqlSession

有三種設定方式:

3.1 在mybatis的config檔案中設定

SqlSessionFactoryBean中可以配置配置檔案:

<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
  <property name="dataSource" ref="oracleDataSource" />
  <property name="configLocation" value="classpath:mybatis/config/mybatis-config-06.xml"/>
</bean>

這個mybatis配置檔案中,設定BATCH方式:

<configuration>
  <settings>
    <!-- 預設開啟BATCH的Executor -->
    <setting name="defaultExecutorType" value="BATCH" />
  </settings>
  <mappers>
    <mapper class="com.ws.experiment.spring.mybatis.mapper.StudentMapper06" />
  </mappers>
</configuration>

這樣,預設開啟的sqlSession就都是BATCH方式的。再與spring的事務結合(參看上一節中的spring事務設定),就可以實現批量插入。

測試結果:

batchInsert001插入10筆資料耗時: 565 ms
batchInsert001插入50筆資料耗時: 117 ms
batchInsert001插入100筆資料耗時: 98 ms
batchInsert001插入200筆資料耗時: 106 ms
batchInsert001插入500筆資料耗時: 145 ms
batchInsert001插入1000筆資料耗時: 132 ms
batchInsert001插入2000筆資料耗時: 154 ms
batchInsert001插入3000筆資料耗時: 163 ms
batchInsert001插入5000筆資料耗時: 200 ms
batchInsert001插入8000筆資料耗時: 250 ms

3.2 自己建立sqlSession,手工commit

SqlSessionFactory sqlSessionFactory = (SqlSessionFactory)context.getBean("sqlSessionFactory");
SqlSession sqlSession = sqlSessionFactory.openSession(ExecutorType.BATCH,false);
StudentMapper06 studentMapper = sqlSession.getMapper(StudentMapper06.class);
for (int i = 0; i < count; i++) {
  Student st = new Student();
  st.setName("name");
  ...
  studentMapper.insertStudent(st);
}
sqlSession.commit();
sqlSession.clearCache();
sqlSession.close();

測試結果:

batchInsert002插入10筆資料耗時: 568 ms
batchInsert002插入50筆資料耗時: 157 ms
batchInsert002插入100筆資料耗時: 132 ms
batchInsert002插入200筆資料耗時: 135 ms
batchInsert002插入500筆資料耗時: 148 ms
batchInsert002插入1000筆資料耗時: 139 ms
batchInsert002插入2000筆資料耗時: 151 ms
batchInsert002插入3000筆資料耗時: 139 ms
batchInsert002插入5000筆資料耗時: 207 ms
batchInsert002插入8000筆資料耗時: 299 ms

3.3 使用sqlSessionTemplate在XML檔案中建立bean

建立一個SqlSessionTemplate,然後注入到MapperFactoryBean中,生成對應的mapper:

<!-- 以ExecutorType.BATCH方式插入資料庫 -->
<bean id="batchSqlSessionTemplate" class="org.mybatis.spring.SqlSessionTemplate">
  <constructor-arg name="sqlSessionFactory" ref="sqlSessionFactory" />
  <constructor-arg name="executorType" value="BATCH" />
</bean>
<bean id="studentMapper06_batch" class="org.mybatis.spring.mapper.MapperFactoryBean">
  <property name="mapperInterface" value="com.ws.experiment.spring.mybatis.mapper.StudentMapper06" />
  <property name="sqlSessionTemplate" ref="batchSqlSessionTemplate" />
</bean>
<bean id="studentDao06_batch" class="com.ws.experiment.spring.mybatis.dao.StudentDao06">
  <property name="studentMapper" ref="studentMapper06_batch" />
</bean>

與spring的事務結合後(參看上一節中的spring事務設定),就可以實現批量插入

測試結果

batchInsert003插入10筆資料耗時: 651 ms
batchInsert003插入50筆資料耗時: 133 ms
batchInsert003插入100筆資料耗時: 124 ms
batchInsert003插入200筆資料耗時: 129 ms
batchInsert003插入500筆資料耗時: 144 ms
batchInsert003插入1000筆資料耗時: 179 ms
batchInsert003插入2000筆資料耗時: 229 ms
batchInsert003插入3000筆資料耗時: 241 ms
batchInsert003插入5000筆資料耗時: 216 ms
batchInsert003插入8000筆資料耗時: 259 ms

到此這篇關於spring中使用mybatis實現批量插入的示例程式碼的文章就介紹到這了,更多相關spring mybatis批量插入內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們!