/*

        批量插入和返回每条插入记录的id
     */
    @Insert("INSERT into village(name,district) VALUES(#{villageName}, #{district})")
    @Options(useGeneratedKeys = true, keyColumn = "id", keyProperty = "vid")
    void insertVillage(Village village);

/*


      @Test
  void testInsertWithMapper() {
    try (SqlSession sqlSession = sqlSessionFactory.openSession(ExecutorType.BATCH)) {
      Mapper userMapper = sqlSession.getMapper(Mapper.class);
      User user1 = new User(null, "Pocoyo");
      userMapper.insert(user1);
      User user2 = new User(null, "Valentina");
      userMapper.insert(user2);
      sqlSession.flushStatements();
      assertEquals(Integer.valueOf(50), user1.getId());
      assertEquals(Integer.valueOf(50), user2.getId());
      sqlSession.commit();
    }

    try (SqlSession sqlSession = sqlSessionFactory.openSession()) {
      List<User> users = sqlSession.selectList("select");
      Assertions.assertEquals(2, users.size());
    }
  }



     */

    @Test
    public void insertBatchVillage() {

        ArrayList<Village> villages = new ArrayList<>();

        for (int i = 0; i < 2; i++) {

            Village village = new Village();

//            village.setVid(4 + i);

            village.setVillageName("小村庄" + i);

            village.setDistrict("100" + i + "0");

            villages.add(village);
        }

        try (SqlSession sqlSession = factory.openSession(ExecutorType.BATCH)) {
            VillageMapper villageMapper = sqlSession.getMapper(VillageMapper.class);
            villages.forEach(new Consumer<Village>() {
                @Override
                public void accept(Village village) {
                    villageMapper.insertVillage(village);
                }
            });
            sqlSession.flushStatements();
            sqlSession.commit();
        }

        System.out.println(new Gson().toJson(villages));



    }