为了说明后续的Mybatis扩展,插播一篇广告,先来简要说明一下Mybatis的一种原生用法,不过先声明:下面说的只是Mybatis的其中一种用法,如需要更深入了解Mybatis,请参考官方文档,或者研读源码。  

  我们知道,使用Mybatis的方式有很多种,从是否集成上分,可以单独使用,也可以和Spring集成使用;从使用方式上分,可以编写静态工具类,在静态工具中调用SqlSession,也可以直接注入SqlSession/ SqlSessionTemplate,还可以编写Dao接口,让mybatis自动生成代理子类(对于只编写接口,不写实现类就可以运行有疑问的朋友,可以先了解一下JDK中的动态代理技术,这里就不展开了);在sql脚本的编写上,可以使用xml——也就是通常所说的SqlMapper,可以使用注解,还可以使用Mybatis提供的API生成等等。

集成Spring + Dao接口/自动动态代理子类 + SqlMapper配置。

  代理接口方式,我们姑且不管是怎么样代理的,但能够运行起来,肯定是有一段真正的可以运行的代码——实际上就是调用SqlSession接口中的方法。因此首先需要解决的就是下面三个基本问题:

问题1:确定需要执行的sqlId

  从SqlSession接口的方法签名可以知道,所有的数据访问方法,都必须有一个参数“statement”,也就是我们通常所说的sqlId。那么,sqlId是怎么确定的?SqlMapper中那么多配置,Mybatis怎么知道调用哪一个?

问题2:确定需要执行的方法

  不管是用哪种方式,归根结底还是调用SqlSession接口中的方法,那么问题来了,SqlSession接口中那么多不同方法,怎么知道调用哪一个呢?比如Dao接口中的一个查询方法,是调用SqlSession的selectOne?还是selectList?还是有回调的select?还是分页查询呢?

问题3:确定执行SQL时的参数

  除了sqlId,还有执行SQL时的参数,也是需要解决的一个基本问题,如果Dao接口中有多个参数,传入了多个参数,又怎么组装成一个统一的参数呢?SQL运行时的参数也会影响到执行的方法,因为很多重载的方法就只能通过运行时参数来确定。

  下面,我们带着这三个问题,看一个简单的用户维护的例子,分析一下代理接口方式原生用法是怎么处理这三个问题的,然后,也思考一下Mybatis的处理有没有什么不足和缺陷,我们又怎么处理这些不足和缺陷?

  业务场景做了简化,用户模型只包括用户ID、用户名称和用户所属的机构,需要完成的操作有:查询用户列表、查询用户列表(分页查询)、查找用户对象、新增用户、修改用户以及删除用户。

(1)Dao接口



1 @Repository("userDao")
 2 public interface IUserDao {
 3     
 4     /**
 5      * 查询用户列表
 6      * @param user    用户查询参数
 7      * @return
 8      */
 9     public List<UserBean> dList(UserForm user);
10     
11     /**
12      * 查询用户列表(分页查询)
13      * @param user    用户查询参数
14      * @param page  分页对象
15      * @return
16      */
17     public List<UserBean> dPageList(UserForm user, IPage page);
18     
19     /**
20      * 查找单个用户
21      * @param userId 用户ID
22      * @return
23      */
24     public UserBean dFind(@Param("userId")String userId);
25     
26     /**
27      * 新增单个用户
28      * @param user
29      * @return
30      */
31     public int dInsert(UserForm user);
32     
33     /**
34      * 更新单个用户
35      * @param user
36      * @return
37      */
38     public int dUpdate(UserForm user);
39     
40     /**
41      * 删除单个用户
42      * @param userId
43      * @return
44      */
45     public int dDelete(@Param("userId")String userId);
46 }



(2)SqlMapper配置



1 <?xml version="1.0" encoding="UTF-8" ?>
 2 <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
 3 <mapper namespace="com.forms.beneform4j.webapp.systemmanage.user.dao.IUserDao">
 4     
 5     <select id="dList" resultType="com.forms.beneform4j.webapp.systemmanage.user.bean.UserBean">
 6         SELECT U.USER_ID, U.USER_NAME, U.ORG_ID
 7           FROM BF_USER U
 8          <where>
 9              <if test="null != userId and '' != userId">
10                  AND U.USER_ID = #{userId, jdbcType=VARCHAR}
11              </if>
12              <if test="null != orgId and '' != orgId">
13                  AND U.ORG_ID = #{orgId, jdbcType=VARCHAR}
14              </if>
15              <if test="null != userName and '' != userName">
16                  AND U.USER_NAME LIKE '%'||#{userName, jdbcType=VARCHAR}||'%'
17              </if>
18          </where>
19          ORDER BY U.ORG_ID, U.USER_ID
20     </select>
21     
22     <select id="dPageList" resultType="com.forms.beneform4j.webapp.systemmanage.user.bean.UserBean">
23         SELECT U.USER_ID, U.USER_NAME, U.ORG_ID
24           FROM BF_USER U
25          <where>
26              <if test="null != userId and '' != userId">
27                  AND U.USER_ID = #{userId, jdbcType=VARCHAR}
28              </if>
29              <if test="null != orgId and '' != orgId">
30                  AND U.ORG_ID = #{orgId, jdbcType=VARCHAR}
31              </if>
32              <if test="null != userName and '' != userName">
33                  AND U.USER_NAME LIKE '%'||#{userName, jdbcType=VARCHAR}||'%'
34              </if>
35          </where>
36          ORDER BY U.ORG_ID, U.USER_ID
37     </select>
38     
39     <select id="dFind" resultType="com.forms.beneform4j.webapp.systemmanage.user.bean.UserBean">
40         SELECT U.USER_ID, U.USER_NAME, U.ORG_ID
41           FROM BF_USER U
42          WHERE U.USER_ID = #{userId, jdbcType=VARCHAR}
43     </select>
44     
45     <insert id="dInsert" >
46         INSERT INTO BF_USER (USER_ID, USER_NAME, ORG_ID)
47         VALUES (#{userId,jdbcType=VARCHAR}, #{userName,jdbcType=VARCHAR}, #{orgId,jdbcType=VARCHAR})
48     </insert>
49     <update id="dUpdate" >
50         UPDATE BF_USER
51            SET USER_NAME = #{userName,jdbcType=VARCHAR},
52                ORG_ID = #{orgId,jdbcType=VARCHAR}
53          WHERE USER_ID = #{userId,jdbcType=VARCHAR}
54     </update>
55     <delete id="dDelete">
56         DELETE FROM BF_USER 
57          WHERE USER_ID = #{userId, jdbcType=VARCHAR}
58     </delete>
59 </mapper>



  现在来看Mybatis是怎么解决上面的问题的。

  第一个问题:确定需要执行的sqlId

  Mybatis的解决方法很简单,就是将接口类加上方法作为sqlId,可以用下面一个公式来描述:



需要执行的SqlId
 = Dao接口所在的类名(全限定名)+ 点号(.)+ 方法名
 = SqlMapper配置文件中的命名空间 + 点号(.) + [select|insert|update|delete]元素的id属性



  有什么不足呢?很明显,以下几种情形都无法处理:

  1. 两个重载的方法(方法名相同,参数不同),需要执行不同的sqlId
  2. 两个不同名的方法,需要引用相同的sqlId配置(参数不同,比如查询列表和查找元素,就常常可以共用一个<select>配置)
  3. 当前Dao中的方法需要引用命名空间和全限定类名不一致的sqlId(比如有一个公共<select>配置,各个业务模块都有可能引用这段配置,当然,这种情况下并不建议dao直接引用<select>配置,而建议编写一个公共服务)
  4. 此外,对于我们前面博客中提到的批量执行,也是无法处理的

  其实,这里比较好一点的方法是添加一个类似下面的sqlId查找策略接口:



1 public interface ISqlIdLookupStrategy {
2 
3     public String lookup(Method method);
4 }



  Mybatis可以实现一个默认查找策略,然后用户也可以自己实现这个查找策略,比如就可以利用这个接口实现sqlId的置换——是的,就是传说中的张大教主的乾坤大挪移。但是遗憾的是,Mybatis开发人员估计认为没有必要不按默认方法查找,所以直接在代码中写死了。更加遗憾的是,在我们自己的平台中,也因为种种评审,将这个接口给砍掉,理由是既然Mybatis没有提供,我们也就可以不提供(好有道理,我竟然无法反驳,所以做开发平台也有很多不爽的地方——这是后话,暂且不说了)。

  第二个问题:确定需要执行的方法

  确定了sqlId,Mybatis怎么确定需要执行的方法呢?

  首先想到的,可以根据sqlId读取到SqlMapper中的配置(不用担心每次都去解析xml,Mybatis会有缓存的),然后就可以知道是select|insert|update|delete中的哪一种,因此,可以先确定sql语句的类型。对于insert、update、delete,基本上也就确定了调用方法,但是对于select,因为SqlSession中的方法还是很多,还需要进一步区分。

  进而想到的是,可以根据dao接口方法的签名来区分,签名包括返回值和形参,比如返回值是void,并且形参中包含处理查询结果的回调接口ResultHandler,那么,就可以大致确定需要调用void select(String statement, ResultHandler handler)这族重载方法中的一种,进一步可以分析形参中是否包含分页参数RowBounds、是否包含执行参数从而选择重载方法中的其中一种;又比如可以根据返回值是否为集合(Collection)类型来区分是查询单笔(selectOne)还是查询列表(selectList)。

  如果上面两种方法还确定不了,Mybatis还有第三种方式:在dao接口方法上添加特定的注解,比如使用@Flush表示批量执行,使用@MapKey表示selectMap等。

  说了这么多,可能还不够清晰,那既然是程序员,那就用程序员的语言来沟通,把Mybatis的源码贴在这里:



1 public Object execute(SqlSession sqlSession, Object[] args) {
 2     Object result;
 3     if (SqlCommandType.INSERT == command.getType()) {
 4       Object param = method.convertArgsToSqlCommandParam(args);
 5       result = rowCountResult(sqlSession.insert(command.getName(), param));
 6     } else if (SqlCommandType.UPDATE == command.getType()) {
 7       Object param = method.convertArgsToSqlCommandParam(args);
 8       result = rowCountResult(sqlSession.update(command.getName(), param));
 9     } else if (SqlCommandType.DELETE == command.getType()) {
10       Object param = method.convertArgsToSqlCommandParam(args);
11       result = rowCountResult(sqlSession.delete(command.getName(), param));
12     } else if (SqlCommandType.SELECT == command.getType()) {
13       if (method.returnsVoid() && method.hasResultHandler()) {
14         executeWithResultHandler(sqlSession, args);
15         result = null;
16       } else if (method.returnsMany()) {
17         result = executeForMany(sqlSession, args);
18       } else if (method.returnsMap()) {
19         result = executeForMap(sqlSession, args);
20       } else {
21         Object param = method.convertArgsToSqlCommandParam(args);
22         result = sqlSession.selectOne(command.getName(), param);
23       }
24     } else if (SqlCommandType.FLUSH == command.getType()) {
25         result = sqlSession.flushStatements();
26     } else {
27       throw new BindingException("Unknown execution method for: " + command.getName());
28     }
29     if (result == null && method.getReturnType().isPrimitive() && !method.returnsVoid()) {
30       throw new BindingException("Mapper method '" + command.getName() 
31           + " attempted to return null from a method with a primitive return type (" + method.getReturnType() + ").");
32     }
33     return result;
34   }



需要注意的是,上面的那些方法和步骤也并没有完全确定调用的方法,还需要根据调用时传入的实际参数类型进一步筛选。

  简单总结一下,确定调用方法的依据:

  1. SqlMapper中的元素名称,如select | insert | update | delete
  2. 接口方法签名中特定类型或特定类型的子类型,如分页参数类型RowBounds、结果回调处理器类型ResultHandler
  3. 接口方法的返回值,如是否集合Collection类型或数组类型
  4. 接口方法的注解,如@Flush、@MapKey等
  5. 调用接口时的运行时参数

说到运行时参数,也引出第三个问题的处理:

  第三个问题:确定执行SQL时的参数

  上面例子中有的使用@Param注解,有的只是表单对象,有的包含类型的参数类型(如IPage等),Mybatis是怎么包装这些参数的呢?

    首先有个规则:特殊类型参数不参与组装运行SQL时的参数,那什么是特殊类型参数?实际上就是有特殊意义的参数类型,在Mybatis原生用法中,只有两个,那就是分页对象类型RowBounds和结果回调处理类型ResultHandler,后面我们还会扩展这个特殊类型,添加IPage等。

     除了特殊类型的参数之外,Mybatis按照如下的逻辑包装运行SQL时的参数:

  1. 如果实参为null或者除了特殊类型参数外没有形参,那就直接返回null
  2. 如果没有标注了@Param注解的形参,并且形参的个数只有1个(特殊类型参数之外),那么返回实参中的这个对应的参数对象,不做任何包装
  3. 其它情形,先创建一个Map结构,将含@Param注解的参数以该注解的值为键值,对应索引处的实际参数为值存入Map结构,其它没有@Param注解的参数,就使用"param"加上索引序号为键值,对应实际参数为值存入Map结构,最后再返回这个Map结构作为SqlSession接口方法的参数对象

  同样,把相关源码贴一下:



1 public Object convertArgsToSqlCommandParam(Object[] args) {
 2     /**
 3      *  1.params 是一个类属性,属于Map类型,其中存储的是参数位置索引和@Param注解中value值的映射关系,但会忽略特殊类型的参数
 4      *  2.hasNamedParameters也是一个类属性,boolean类型,表示参数是否有@Param注解(只要任意一个参数有该注解就算有)
 5      */
 6     final int paramCount = params.size();
 7     if (args == null || paramCount == 0) {
 8         return null;
 9     } else if (!hasNamedParameters && paramCount == 1) {
10         return args[params.keySet().iterator().next().intValue()];
11     } else {
12         final Map<String, Object> param = new ParamMap<Object>();
13         int i = 0;
14         for (Map.Entry<Integer, String> entry : params.entrySet()) {
15             param.put(entry.getValue(), args[entry.getKey().intValue()]);
16             // issue #71, add param names as param1, param2...but ensure
17             // backward compatibility
18             final String genericParamName = "param" + String.valueOf(i + 1);
19             if (!param.containsKey(genericParamName)) {
20                 param.put(genericParamName, args[entry.getKey()]);
21             }
22             i++;
23         }
24         return param;
25     }
26 }



  对于执行参数的组装还不明确的朋友,可以自己跟踪源码调试分析一下。

  上面三个基本问题很重要,这是我们赖以使用和扩展的基础。广告插播先到这里,后面继续记录对Mybatis的扩展使用。