60.实现一个微型的Mybatis-注解版

有了之前的基础，使用注解的方式来读取配置就很简单了，本文就简单说说使用注解的情况下如何获取。 　　‍

修改主配置文件

SqlMapConfig.xml，修改mapper的属性为class：

    <mappers>
<!--        <mapper resource="com/peterjxl/dao/IUserDao.xml"/>-->
         <mapper class="com.peterjxl.dao.IUserDao"/>
    </mappers>

‍

‍

新建Select注解

‍

package com.peterjxl.mybatis.annotations;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

/**
 * 查询的注解
 */

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface Select {
    /**
     * 配置SQL语句的
     * @return
     */
    String value();
}

‍

在IUserDao接口上配置注解

package com.peterjxl.dao;

import com.peterjxl.domain.User;
import com.peterjxl.mybatis.annotations.Select;

import java.util.List;

public interface IUserDao {

    /**
     * 查询所有用户
     * @return
     */

    @Select("select * from user")
    List<User> findAll();
}

‍

‍

‍

修改工具类XMLConfigBuilder​

我们之前注释掉了XMLConfigBuilder​类中，关于解析注解的方法loadMapperAnnotation​，现在我们取消注释，并引入缺失的类（例如注解）

该方法会获取到主配置文件的mapper标签，然后读取配置的全限定类名，再读取注解属性，存储到mapper对象里。

private static Map<String,Mapper> loadMapperAnnotation(String daoClassPath)throws Exception{
    //定义返回值对象
    Map<String,Mapper> mappers = new HashMap<String, Mapper>();

    //1.得到dao接口的字节码对象
    Class daoClass = Class.forName(daoClassPath);
    //2.得到dao接口中的方法数组
    Method[] methods = daoClass.getMethods();
    //3.遍历Method数组
    for(Method method : methods){
        //取出每一个方法，判断是否有select注解
        boolean isAnnotated = method.isAnnotationPresent(Select.class);
        if(isAnnotated){
            //创建Mapper对象
            Mapper mapper = new Mapper();
            //取出注解的value属性值
            Select selectAnno = method.getAnnotation(Select.class);
            String queryString = selectAnno.value();
            mapper.setQueryString(queryString);
            //获取当前方法的返回值，还要求必须带有泛型信息
            Type type = method.getGenericReturnType();//List<User>
            //判断type是不是参数化的类型
            if(type instanceof ParameterizedType){
                //强转
                ParameterizedType ptype = (ParameterizedType)type;
                //得到参数化类型中的实际类型参数
                Type[] types = ptype.getActualTypeArguments();
                //取出第一个
                Class domainClass = (Class)types[0];
                //获取domainClass的类名
                String resultType = domainClass.getName();
                //给Mapper赋值
                mapper.setResultType(resultType);
            }
            //组装key的信息
            //获取方法的名称
            String methodName = method.getName();
            String className = method.getDeclaringClass().getName();
            String key = className+"."+methodName;
            //给map赋值
            mappers.put(key,mapper);
        }
    }
    return mappers;
}

‍

‍

测试

我们再次运行测试方法，可以看到能正常运行

使用的是注解
User{id=41, username='张三', birthday=2018-02-27T17:47:08, sex='男', address='北京'}
User{id=42, username='李四', birthday=2018-03-02T15:09:37, sex='女', address='北京'}
User{id=43, username='王五', birthday=2018-03-04T11:34:34, sex='女', address='北京'}
User{id=45, username='赵六', birthday=2018-03-04T12:04:06, sex='男', address='北京'}
User{id=46, username='小七', birthday=2018-03-07T17:37:26, sex='男', address='北京'}
User{id=48, username='老八', birthday=2018-03-08T11:44, sex='男', address='北京'}

‍

‍

源码

所有代码已上传到了GitHub (opens new window)和Gitee (opens new window)上，并且创建了分支demo5，读者可以通过切换分支来查看本文的示例代码。

‍