[置顶] mybatis批量新增系列之有主键的表的批量新增

前面介绍了无主键的表的批量插入，文章地址：http://blog.csdn.net/zhouxiaoyun0228/article/details/9980181

但是在开发中往往许多的表是需要主键的，因而现在介绍一下有主键的表的批量插入，该主键是数据类型的数字类型。

最开始我是这么想的：主键让他自动添加，因而我们可以在insert中添加一个自增序列号就可以了，如下

 <selectKey resultType="long" keyProperty="ID" order="BEFORE">
     SELECT IBATIS_SEQUENCE.NEXTVAL AS ID FROM dual
    </selectKey>

然后在listStr中添加一个#{obj.ID,jdbsType=DOUBLE}即可，但是发现没用，生成的Id不会自动添加进入List中。

后面有想了一下，其实思路还是没有错的。。我们只需要转化一下，将生成的主键Id值放入自动添加到List集合中即可，代码如下

 for(DataRecord d:list){
   Serializable id = this.generateId();
   d.put(keyName, id);
   back.add(d);
  }

上面红色的标记是是与无主键部分唯一的区别，keyName是传入进来的主键名称，this.generateId();是提供自增长id的方法。

public Serializable generateId() {
  return this.getDao().queryForLong("com.mip.biz.syn.data.batch.demo.getKeyValue", null);
 }

mapper中的代码如下

  <select id="getKeyValue" parameterType="map" resultType="LONG">
     SELECT IBATIS_SEQUENCE.NEXTVAL AS ID FROM dual
   </select>

这样一来也就解决了表有主键的问题。其他的代码没有变化，就是接口变化了一下：

   /**
    * 有主键的表批量插入
    * @param list
    * @param tableName
    * @param keyName  主键名称
    */

public void batchInsertWithKey(List<DataRecord> list,String tableName,String keyName);

通过观察，发现代码好多冗余的，因而我们可以重构下代码，重构后的java代码如下，mapper中的代码不变，只添加了上面提到过的getKeyValue

接口方法：

  /**
     * 无主键的表批量插入
     * @param list
     * @param tableName
     */
 public void batchInsertWithOutKey(List<DataRecord> list,String tableName);
   /**
    * 有主键的表批量插入
    * @param list
    * @param tableName
    * @param keyName  主键名称
    */
 public void batchInsertWithKey(List<DataRecord> list,String tableName,String keyName);

以下是实现方法  this.getDao().这里是内部封装的dao层，可以自己改动一下去实现与mapper交互
 @Override
 public void batchInsertWithKey(List<DataRecord> list,String tableName,String keyName) {
  this.batchInsert(list, tableName, keyName);
 }
 @Override
 public void batchInsertWithOutKey(List<DataRecord> list,String tableName) {
  this.batchInsert(list, tableName, null);
 }
 //查询表所包含的列名，以及该列对应的类型
 private List<DataRecord> getTableColumns(String tableName){
  Map<String,Object> m=new HashMap<String,Object>();
  m.put("tableName", tableName);
  //去数据库中查询
  List<DataRecord> list=this.getDao().queryForDataSet("com.mip.biz.syn.data.batch.demo.getTableColumn", m).getResults();
  return list;
 }
 //改变列名对应的类型
 private String changeColumnType(String dataType){
  if("NUMBER".equals(dataType)){
   return "DOUBLE";
  }else if("VARCHAR2".equals(dataType)){
   return "VARCHAR";
  }else if(dataType.startsWith("TIMESTAMP")){
   return "TIMESTAMP"; 
  }else{
   return dataType;
  }
 }

 //根据是否有主键名改变集合的数据，DataRecord是Map的一个自定义的子类
 private List<Map> changgeList(List<DataRecord> list,String keyName ){
  List<Map> back=new ArrayList<Map>();
  if(keyName==null){
   for(DataRecord d:list){
    back.add(d);
   }
  }else{
   for(DataRecord d:list){
    Serializable id = this.generateId();
    d.put(keyName, id);
    back.add(d);
   }
  }
  return back;
 }
 //批量插入数据
 private void batchInsert(List<DataRecord> list,String tableName,String keyName){
  //如果无数据，不需要执行插入了
  if(list.size()<=0){
   return ;
  }
  //获取表的列名以及类型
  List<DataRecord> columns=this.getTableColumns(tableName);
  if(columns.size()==0){
   return;
  }
  Map<String,String> columnMap=new HashMap<String,String>();
  for(DataRecord dr:columns){
   columnMap.put(dr.getString("COLUMN_NAME"), changeColumnType(dr.getString("DATA_TYPE")));
  }
  List<Map> back=this.changgeList(list, keyName);
  Map dm=back.get(0);
  Set<String> objectKeys = dm.keySet();
  StringBuffer insertSql=new StringBuffer();
  insertSql.append("insert into "+tableName+" (");
  StringBuffer listSql=new StringBuffer();
  listSql.append(" select ");
  for(String col:objectKeys){
   insertSql.append(""+col+",");
   listSql.append("#{obj."+col+",jdbcType="+columnMap.get(col)+"}"+",");
  }
  // 删除最后的半角逗号
  if (insertSql.length() > 0) {
   insertSql.deleteCharAt(insertSql.length() - 1);
  }
  if (listSql.length() > 0) {
   listSql.deleteCharAt(listSql.length() - 1);
  }
  insertSql.append(")");
  listSql.append(" from dual");
  Map<String,Object> batchMap=new HashMap<String,Object>();
  batchMap.put("insertSql", insertSql.toString());
  batchMap.put("listSql", listSql.toString());
  batchMap.put("list", back);
  this.getDao().insertByStatement("com.mip.biz.syn.data.batch.demo.batchDemo", batchMap);
 }
 //获取主键序列号
 public Serializable generateId() {
  return this.getDao().queryForLong("com.mip.biz.syn.data.batch.demo.getKeyValue", null);
 }
 public void test(){
  //test_batch
  List<DataRecord> list=new ArrayList<DataRecord>();
  for(int i=0;i<10;i++){
   DataRecord dr=new DataRecord();
   dr.put("USERNAME","NAME_"+i);
   list.add(dr);
  }
  batchInsertWithKey(list,"TEST_BATCH","ID");
 }
}