How to use

getBatchCount

method

in

com.querydsl.sql.dml.SQLInsertClause

long size = insert.getBatchCount();
if (size > 0) {
  if (createSetId()) {

  /**
   * 批量插入数据,可以携带ID,但没有ID的不会进行ID赋值
   *
   * @param entityList 实体
   * @return 变更条数
   */
  default long createWithId(@Nonnull Collection<T> entityList) {
    SQLInsertClause insert = sql().insert(root());
    entityList.forEach(t -> {
      beforeCreate(t);
      validOnCreate(t);
      insert.populate(t, DefaultMapper.WITH_NULL_BINDINGS).addBatch();
    });
    long size = insert.getBatchCount();
    if (size > 0) {
      insert.execute();
    }
    insert.clear();
    return size;
  }
}

@Test
public void clear() {
  QEmployee emp1 = new QEmployee("emp1");
  SQLInsertClause insert = new SQLInsertClause(null, SQLTemplates.DEFAULT, emp1);
  insert.set(emp1.id, 1);
  insert.addBatch();
  assertEquals(1, insert.getBatchCount());
  insert.clear();
  assertEquals(0, insert.getBatchCount());
}

@Test
public void insert_batch() {
  SQLInsertClause insert = insert(survey)
      .set(survey.id, 5)
      .set(survey.name, "55")
      .addBatch();
  assertEquals(1, insert.getBatchCount());
  insert.set(survey.id, 6)
      .set(survey.name, "66")
      .addBatch();
  assertEquals(2, insert.getBatchCount());
  assertEquals(2, insert.execute());
  assertEquals(1L, query().from(survey).where(survey.name.eq("55")).fetchCount());
  assertEquals(1L, query().from(survey).where(survey.name.eq("66")).fetchCount());
}

@Test
public void insert_batch_to_bulk() {
  SQLInsertClause insert = insert(survey);
  insert.setBatchToBulk(true);
  insert.set(survey.id, 5)
      .set(survey.name, "55")
      .addBatch();
  assertEquals(1, insert.getBatchCount());
  insert.set(survey.id, 6)
      .set(survey.name, "66")
      .addBatch();
  assertEquals(2, insert.getBatchCount());
  assertEquals(2, insert.execute());
  assertEquals(1L, query().from(survey).where(survey.name.eq("55")).fetchCount());
  assertEquals(1L, query().from(survey).where(survey.name.eq("66")).fetchCount());
}