Insight mybatis insert useGeneratedKeys

有这么一个场景:

使用mybatis insert操作,需要新增记录的主键Id,可以配置useGeneratedKeys=true 并且keyProperty=xxx。

这样,mybatis 就把主键Id回填至入参object。

useGeneratedKeys 是怎样实现的?

首先需要明确的是,insert 执行完成后,才发生的主键回填的事件。如下所示:

public int update(Statement statement) throws SQLException {
    // 1.insert sql 执行
    PreparedStatement ps = (PreparedStatement) statement;
    ps.execute();
    int rows = ps.getUpdateCount();
    // 2.获取当前MS 配置的KeyGenerator,并执行回填操作
    Object parameterObject = boundSql.getParameterObject();
    KeyGenerator keyGenerator = mappedStatement.getKeyGenerator();
    keyGenerator.processAfter(executor, mappedStatement, ps, parameterObject);
    return rows;
}

其次,主键生成分为auto 或者人工指定(包括sequence),useGeneratedKeys做了两种适配。

1.通过Jdbc接口查询自增记录并回填,参考

public void processBatch(MappedStatement ms, Statement stmt, List parameters) {
    ResultSet rs = null;
    try {
        // Retrieves any auto-generated keys created as a result of executing this Statement object.
        rs = stmt.getGeneratedKeys();
        // mapper配置的keyProperty
        final String[] keyProperties = ms.getKeyProperties();
        if (keyProperties != null && rsmd.getColumnCount() >= keyProperties.length) {
            for (Object parameter : parameters) {
              if (!rs.next()) break; // there should be one row for each statement (also one for each parameter)
              final MetaObject metaParam = configuration.newMetaObject(parameter);
              if (typeHandlers == null) typeHandlers = getTypeHandlers(typeHandlerRegistry, metaParam, keyProperties);
              populateKeys(rs, metaParam, keyProperties, typeHandlers);
            }
        }
    }
    //...
}

2.人工指定的主键回填,参考:SelectKeyGenerator。


总结:

主键回填原来是这样的,豁然开朗。JDBC 接口需要认真读一读。

你可能感兴趣的:(读书笔记,mybatis)