Mybatis深入剖析 - 各种KeyGenerator的说明

KeyGenerator接口有三个实现类,SelectkeyGenerator,Jdbc3KeyGenerator,NoKeyGenerator

其中SelectkeyGenerator对应于生成主键操作,即针对insert类型的sql语句,配置selectKey的策略,自动为insert语句的参数补全id

Jdbc3KeyGenerator和NoKeyGenerator对应于insert写库,是否取回表的自增Id,它对应的Configuration的useGeneratedKeys全局配置,或者Mapper中insert操作的useGeneratedKeys属性,可以看到如果配置了,则生成Jdbc3KeyGenerator对象,没有配置则生成NoKeyGenerator对象,其中NoKeyGenerator的方法都为空,不产生任何效果

  keyGenerator = context.getBooleanAttribute("useGeneratedKeys",
          configuration.isUseGeneratedKeys() && SqlCommandType.INSERT.equals(sqlCommandType))
          ? new Jdbc3KeyGenerator(context.getStringAttribute("keyColumn", null)) : new NoKeyGenerator();

 

public class NoNoKeyGenerator implements KeyGenerator {

  public void processBefore(Executor executor, MappedStatement ms, Statement stmt, Object parameter) {
  }

  public void processAfter(Executor executor, MappedStatement ms, Statement stmt, Object parameter) {
  }

}

 

 

SelectKeyGenerator

SelectKey在Mybatis中是为了解决Insert数据时不支持主键自动生成的问题。在实际应用中最好代码设置主键,或者简单使用数据库自增主键。

属性 描述
keyProperty selectKey 语句结果应该被设置的目标属性。
resultType 结果的类型。MyBatis 通常可以算出来,但是写上也没有问题。MyBatis 允许任何简单类型用作主键的类型,包括字符串。
order 这可以被设置为 BEFORE 或 AFTER。如果设置为 BEFORE,那么它会首先选择主键,设置 keyProperty 然后执行插入语句。如果设置为 AFTER,那么先执行插入语句,然后是 selectKey 元素-这和如 Oracle 数据库相似,可以在插入语句中嵌入序列调用。
statementType 和前面的相 同,MyBatis 支持 STATEMENT ,PREPARED 和CALLABLE 语句的映射类型,分别代表 PreparedStatement 和CallableStatement 类型。

 

SelectKey需要注意order属性,像Mysql一类支持自动增长类型的数据库中,order需要设置为after才会取到正确的值。

像Oracle这样取序列的情况,需要设置为before,否则会报错。

 

另外在用Spring管理事务时,SelectKey和插入在同一事务当中,因而Mysql这样的情况由于数据未插入到数据库中,所以是得不到自动增长的Key。取消事务管理就不会有问题。

 

下面是一个xml和注解的例子,SelectKey很简单,两个例子就够了:

 

[html]  view plain copy 在CODE上查看代码片
 
  1. <insert id="insert" parameterType="map">  
  2.     insert into table1 (name) values (#{name})  
  3.     <selectKey resultType="java.lang.Integer" keyProperty="id">  
  4.       CALL IDENTITY()  
  5.     </selectKey>  
  6.   </insert>  


上面xml的传入参数是map,selectKey会将结果放到入参数map中。用POJO的情况一样,但是有一点需要注意的是,keyProperty对应的字段在POJO中必须有相应的setter方法,setter的参数类型还要一致,否则会报错。

 

在statementHandler的数据库操作前后,例如准备parameter时执行processBefore,update之后执行processAfter,SelectKeyGenerator根据mapper里配置的order判断是否执行获取id操作

 

  public void processBefore(Executor executor, MappedStatement ms, Statement stmt, Object parameter) {
    if (executeBefore) {
      processGeneratedKeys(executor, ms, stmt, parameter);
    }
  }

  public void processAfter(Executor executor, MappedStatement ms, Statement stmt, Object parameter) {
    if (!executeBefore) {
      processGeneratedKeys(executor, ms, stmt, parameter);
    }
  }
 看下面的核心代码,可以看到其核心的逻辑是,执行mapper对应statement的<selectKey>中的sql语句,并且根据keyProperty回写id到sql的param中

 

 

private void processGeneratedKeys(Executor executor, MappedStatement ms, Statement stmt, Object parameter) {
    try {
      final Configuration configuration = ms.getConfiguration();
      if (parameter != null) {
        String keyStatementName = ms.getId() + SELECT_KEY_SUFFIX;
        if (configuration.hasStatement(keyStatementName)) {

          if (keyStatement != null) {
            String keyProperty = keyStatement.getKeyProperty();
            final MetaObject metaParam = configuration.newMetaObject(parameter);
            if (keyProperty != null && metaParam.hasSetter(keyProperty)) {
              // Do not close keyExecutor.
              // The transaction will be closed by parent executor.
              Executor keyExecutor = configuration.newExecutor(executor.getTransaction(), ExecutorType.SIMPLE);
              List values = keyExecutor.query(keyStatement, parameter, RowBounds.DEFAULT, Executor.NO_RESULT_HANDLER);
              if (values.size() > 1) {
                throw new ExecutorException("Select statement for SelectKeyGenerator returned more than one value.");
              }
              metaParam.setValue(keyProperty, values.get(0));
            }
          }
        }
      }
    } catch (Exception e) {
      throw new ExecutorException("Error selecting key or setting result to parameter object. Cause: " + e, e);
    }
  }
 

 

 

Jdbc3KeyGenerator

 statment中useGeneratedKeys属性对应的类,该配置的作用是调用java.sql.Statement.getGeneratedKeys()获取自动生成的主键创建此Statement对象执行的结果。如果此Statement对象没有产生任何键,则返回空的ResultSet对象。

可以看到该类作用于插入操作结束后,同样是借助keyProperty配置,将取回的id回写到传入的param对象中

public void processBefore(Executor executor, MappedStatement ms, Statement stmt, Object parameter) {
  }

  public void processAfter(Executor executor, MappedStatement ms, Statement stmt, Object parameter) {
    try {
      final Configuration configuration = ms.getConfiguration();
      final TypeHandlerRegistry typeHandlerRegistry = configuration.getTypeHandlerRegistry();
      if (parameter != null) {
        String keyProperty = ms.getKeyProperty();
        final MetaObject metaParam = configuration.newMetaObject(parameter);
        if (keyProperty != null && metaParam.hasSetter(keyProperty)) {
          Class<?> keyPropertyType = metaParam.getSetterType(keyProperty);
          TypeHandler th = typeHandlerRegistry.getTypeHandler(keyPropertyType);
          if (th != null) {
            ResultSet rs = stmt.getGeneratedKeys();
            try {
              ResultSetMetaData rsmd = rs.getMetaData();
              int colCount = rsmd.getColumnCount();
              if (colCount > 0) {
                String colName;
                if (keyColumnName != null && keyColumnName.length() > 0) {
                  colName = keyColumnName;
                } else {
                  colName = rsmd.getColumnName(1);
                }
              
                while (rs.next()) {
                  Object value = th.getResult(rs, colName);
                  metaParam.setValue(keyProperty, value);
                }
              }
            } finally {
              try {
                if (rs != null) rs.close();
              } catch (Exception e) {
                //ignore
              }
            }
          }
        }
      }
    } catch (Exception e) {
      throw new ExecutorException("Error getting generated key or setting result to parameter object. Cause: " + e, e);
    }
  }

 

 

 

你可能感兴趣的:(mybatis)