为了说明后续的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属性
有什么不足呢?很明显,以下几种情形都无法处理:
- 两个重载的方法(方法名相同,参数不同),需要执行不同的sqlId
- 两个不同名的方法,需要引用相同的sqlId配置(参数不同,比如查询列表和查找元素,就常常可以共用一个<select>配置)
- 当前Dao中的方法需要引用命名空间和全限定类名不一致的sqlId(比如有一个公共<select>配置,各个业务模块都有可能引用这段配置,当然,这种情况下并不建议dao直接引用<select>配置,而建议编写一个公共服务)
- 此外,对于我们前面博客中提到的批量执行,也是无法处理的
其实,这里比较好一点的方法是添加一个类似下面的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 }
需要注意的是,上面的那些方法和步骤也并没有完全确定调用的方法,还需要根据调用时传入的实际参数类型进一步筛选。
简单总结一下,确定调用方法的依据:
- SqlMapper中的元素名称,如select | insert | update | delete
- 接口方法签名中特定类型或特定类型的子类型,如分页参数类型RowBounds、结果回调处理器类型ResultHandler
- 接口方法的返回值,如是否集合Collection类型或数组类型
- 接口方法的注解,如@Flush、@MapKey等
- 调用接口时的运行时参数
说到运行时参数,也引出第三个问题的处理:
第三个问题:确定执行SQL时的参数
上面例子中有的使用@Param注解,有的只是表单对象,有的包含类型的参数类型(如IPage等),Mybatis是怎么包装这些参数的呢?
首先有个规则:特殊类型参数不参与组装运行SQL时的参数,那什么是特殊类型参数?实际上就是有特殊意义的参数类型,在Mybatis原生用法中,只有两个,那就是分页对象类型RowBounds和结果回调处理类型ResultHandler,后面我们还会扩展这个特殊类型,添加IPage等。
除了特殊类型的参数之外,Mybatis按照如下的逻辑包装运行SQL时的参数:
- 如果实参为null或者除了特殊类型参数外没有形参,那就直接返回null
- 如果没有标注了@Param注解的形参,并且形参的个数只有1个(特殊类型参数之外),那么返回实参中的这个对应的参数对象,不做任何包装
- 其它情形,先创建一个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的扩展使用。