网站开发人员工工资,网站开发一个支付功能要好多钱,做网站设计的,事业单位网站建设注销情况说明*************************************优雅的分割线 **********************************
分享一波:程序员赚外快-必看的巅峰干货
如果以上内容对你觉得有用,并想获取更多的赚钱方式和免费的技术教程
请关注微信公众号:HB荷包 一个能让你学习技术和赚钱方法的公众号,持续更…*************************************优雅的分割线 **********************************
分享一波:程序员赚外快-必看的巅峰干货
如果以上内容对你觉得有用,并想获取更多的赚钱方式和免费的技术教程
请关注微信公众号:HB荷包 一个能让你学习技术和赚钱方法的公众号,持续更新 *************************************优雅的分割线 ********************************** 前言
在前面两篇博客中我们介绍了对于select语句的简单映射和嵌套映射。mybatis中使用ResultHandler等一系列的类将查询结果封装到实体类中可以说是mybatis中最复杂的过程而剩下的insert、update、delete语句的操作则显得较为简单没有复杂的映射逻辑。这里需要提的是在insert语句中关于主键自增的问题。 KeyGenerator
在我们实际的开发中自增主键还是使用比较多的。而mybatis在处理insert语句时并不会将自增主键给返回而是返回插入的条数。当我们有业务需求要获取插入时产生的自增主键或者其他类型不由程序生成的主键时则可以使用KeyGenerator。
KeyGenerator是个接口定义了processBefore和processAfter两个方法分别在insert之前和之后执行代码如下。
/** insert语句不会反回自动生成的主键 该接口用于在插入记录时获取自增主键 author Clinton Begin */ public interface KeyGenerator { /** 执行insert之前执行设置属性order“BEFORE”param executorparam msparam stmtparam parameter */ void processBefore(Executor executor, MappedStatement ms, Statement stmt, Object parameter); /** 执行insert之后执行设置属性order“AFTER”param executorparam msparam stmtparam parameter */ void processAfter(Executor executor, MappedStatement ms, Statement stmt, Object parameter);
}
KeyGenerator有三个实现类
其中NoKeyGenerator的方法都是空实现本文不再讲解。 Jdbc3KeyGenerator
Jdbc3KeyGenerator用于取回数据库生成的自增id它对应于mybatis-config.xml配置的useGeneratedKeys以及insert节点中配置的useGeneratedKeys属性。
Jdbc3KeyGenerator的processBefore是空实现只实现了processAfter方法该方法会调用processBatch方法将SQL语句执行后生成的主键记录到用户传递的实参中而不是直接返回。代码如下。
/*** 核心方法将sql语句生成的主键存放到参数中** param ms* param stmt* param parameter*/
public void processBatch(MappedStatement ms, Statement stmt, Object parameter) {// 获取sql节点配置的keyProperties指定主键的字段final String[] keyProperties ms.getKeyProperties();if (keyProperties null || keyProperties.length 0) {return;}// 获取数据库生成的主键。如果没有生成主键则结果集为空try (ResultSet rs stmt.getGeneratedKeys()) {// 获取resultSet元信息final ResultSetMetaData rsmd rs.getMetaData();// 获取Configurationfinal Configuration configuration ms.getConfiguration();// 查到的列数小于主键的列数说明查询有误。mybatis不进行处理if (rsmd.getColumnCount() keyProperties.length) {// Error?} else {assignKeys(configuration, rs, rsmd, keyProperties, parameter);}} catch (Exception e) {throw new ExecutorException(Error getting generated key or setting result to parameter object. Cause: e, e);}
}在assignKeys方法中处理对主键赋值的逻辑。该方法会判断参数的类型将Map、集合、对象三种情况分别进行处理。代码如下。
private void assignKeys(Configuration configuration, ResultSet rs, ResultSetMetaData rsmd, String[] keyProperties,Object parameter) throws SQLException {if (parameter instanceof ParamMap || parameter instanceof StrictMap) {// 参数是mapassignKeysToParamMap(configuration, rs, rsmd, keyProperties, (MapString, ?) parameter);} else if (parameter instanceof ArrayList !((ArrayList?) parameter).isEmpty() ((ArrayList?) parameter).get(0) instanceof ParamMap) {// 参数是集合assignKeysToParamMapList(configuration, rs, rsmd, keyProperties, ((ArrayListParamMap?) parameter));} else {// 参数是对象assignKeysToParam(configuration, rs, rsmd, keyProperties, parameter);}
}三个方法逻辑类似这里只对assignKeysToParam方法进行分析。
该方法会在进入方法时将参数转为集合并遍历集合去给主键进行赋值代码如下。
/*** 当参数是对象时的处理逻辑** param configuration* param rs* param rsmd* param keyProperties* param parameter* throws SQLException*/
private void assignKeysToParam(Configuration configuration, ResultSet rs, ResultSetMetaData rsmd,String[] keyProperties, Object parameter) throws SQLException {// 将参数转为集合此时这个集合最多只有一个对象Collection? params collectionize(parameter);if (params.isEmpty()) {return;}// 存放主键属性的信息ListKeyAssigner assignerList new ArrayList();for (int i 0; i keyProperties.length; i) {assignerList.add(new KeyAssigner(configuration, rsmd, i 1, null, keyProperties[i]));}Iterator? iterator params.iterator();while (rs.next()) {if (!iterator.hasNext()) {throw new ExecutorException(String.format(MSG_TOO_MANY_KEYS, params.size()));}Object param iterator.next();assignerList.forEach(x - x.assign(rs, param));}
}SelectKeyGenerator
MySql、PostgreSql等数据库支持自增主键在执行insert时可以不指定主键插入的过程中由数据库去生成自增主键。而像Oracle、DB2等数据库产品则需要使用sequence实现自增id因此在insert之前必须明确指定主键的值。SelectKeyGenerator就用来处理不支持自增主键的数据库。
SelectKeyGenerator主要用于生成主键它会执行映射配置文件中定义的selectKey节点的sql该语句会获取insert语句所需要的主键。
SelectKeyGenerator中有两个核心字段keyStatement用于存放解析后的selectKey节点executeBefore用于标识该节点是在insert之前还是之后执行。
/*** 标识selectKey是在insert之前还是之后执行* true为之前false为之后*/
private final boolean executeBefore;
/*** 存放selectKey节点用于获取insert语句使用的主键*/
private final MappedStatement keyStatement;SelectKeyGenerator的processBefore和processAfter都是执行processGeneratorKeys方法根据executeBefore字段的值决定是在insert之前还是之后执行。
Override
public void processBefore(Executor executor, MappedStatement ms, Statement stmt, Object parameter) {if (executeBefore) {processGeneratedKeys(executor, ms, parameter);}
}Override
public void processAfter(Executor executor, MappedStatement ms, Statement stmt, Object parameter) {if (!executeBefore) {processGeneratedKeys(executor, ms, parameter);}
}processGeneratorKeys方法是处理主键的核心代码。该方法会先获取selectKey节点的keyProperties配置的属性名称该属性对应着主键。接着创建Executor执行器执行selectKey节点中的sql查询主键。Executor是mybatis的核心执行器后面的博客会对其进行介绍。
selectKey执行完毕后会判断查询结果是否合法。主键的查询结果肯定只会返回一条因此查询结果条数不为1的全部视为非法查询。该方法代码如下。
private void processGeneratedKeys(Executor executor, MappedStatement ms, Object parameter) {try {if (parameter ! null keyStatement ! null keyStatement.getKeyProperties() ! null) {// 获取selectKey节点的keyProperties配置的属性名称表示主键对应的属性String[] keyProperties keyStatement.getKeyProperties();final Configuration configuration ms.getConfiguration();// 创建用户传入的实参对应的MetaObject对象final MetaObject metaParam configuration.newMetaObject(parameter);// 创建Executor对象执行keyStatement记录的sql。Executor是mybatis中的执行器后面会讲Executor keyExecutor configuration.newExecutor(executor.getTransaction(), ExecutorType.SIMPLE);// 执行selectKey节点的sql查询主键ListObject values keyExecutor.query(keyStatement, parameter, RowBounds.DEFAULT, Executor.NO_RESULT_HANDLER);if (values.size() 0) {throw new ExecutorException(SelectKey returned no data.);} else if (values.size() 1) {throw new ExecutorException(SelectKey returned more than one value.);} else {// 主键查询出来肯定只有一条因此values的size不等于1的情况都是错误的// 创建主键对象对应的MetaObjectMetaObject metaResult configuration.newMetaObject(values.get(0));if (keyProperties.length 1) {if (metaResult.hasGetter(keyProperties[0])) {// 取出主键的值设置到用户参数对应的属性中setValue(metaParam, keyProperties[0], metaResult.getValue(keyProperties[0]));} else {// 如果主键对象不包含get方法就可能是基本类型或者String直接复制setValue(metaParam, keyProperties[0], values.get(0));}} else {// 处理主键有多列的情况handleMultipleProperties(keyProperties, metaParam, metaResult);}}}} catch (ExecutorException e) {throw e;} catch (Exception e) {throw new ExecutorException(Error selecting key or setting result to parameter object. Cause: e, e);}
}该方法中有一个handleMultipleProperties方法用于处理一张表多个主键的情况。由于复合主键耦合性高、影响性能并且操作起来较为繁琐因此不推荐数据库中给一张表设置多个主键这里也就不对该方法进行介绍。 结语
KeyGenerator的代码较为简单阅读起来也不会吃力。本周我公司已经开始了上班因此博客也需要开始写了mybatis的源码已经阅读了一大半争取在7月份之前把剩下代码的博客全部编写完毕 *************************************优雅的分割线 **********************************
分享一波:程序员赚外快-必看的巅峰干货
如果以上内容对你觉得有用,并想获取更多的赚钱方式和免费的技术教程
请关注微信公众号:HB荷包 一个能让你学习技术和赚钱方法的公众号,持续更新 *************************************优雅的分割线 **********************************
