MyBatis注解之模拟sql语句注解用法示例说明

下文笔者讲述Mybatis中模拟使用注解编写sql语句的方法分享，如下所示

注解与解析

新建一个annotation包,加一个SQL类

@Retention(RetentionPolicy.Runtime)
@Target(ElementType.METHOD)
public @interface SQL {
    String value();
}

@interface表示这是一个注解
@Target里可以指定它能被注在哪个地方，选项有METHOD\TYPE\FIELD等
此处选择注在METHOD(方法)上
@Retention中可以选择存活的时期
选择RUNTIME(运行中)

注解使用方法

public interface UserMapper {
    // 正规写法是@SQL(value="xxxx")，若只有value属性可略去
    @SQL("select * from users where id = {0}")
    void findById(int id);
}

直接从method中可以取到此注解的值
@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable 
{
    SQL sql = method.getAnnotation(SQL.class);
    System.out.println(sql.value());
    return null;
}

参数代入与回调Session

以上代码会打印出相应的sql信息

public class MapperProxy<T> implements InvocationHandler {

    private Session session;

    @SuppressWarnings("unchecked")
    public static <T> T newInstance(Class<T> clazz, Session session) 
    {
        MapperProxy<T> proxy = new MapperProxy<>();
        proxy.session = session;
        // 动态代理
        return (T) Proxy.newProxyInstance(clazz.getClassLoader(), new Class[] { clazz }, proxy);
    }

    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable 
    {
        SQL anno = method.getAnnotation(SQL.class);
        if (anno == null) {
            return null;
        }
        String sql = anno.value();
        Pattern p = Pattern.compile("\\{[^\\}]+\\}");
        Matcher m = p.matcher(sql);
        if (m.find()) {
            StringBuffer sb = new StringBuffer();
            do {
                //获取{param}，用BeanWrapper处理后的返回值来替换
                String s = m.group();
                int index = Integer.valueOf(s.substring(1, s.length() - 1));
                m.appendreplacement(sb, args[index].toString());
            } while (m.find());
            sql = m.appendTail(sb).toString();
        }
        System.out.println(sql);
        session.exec(sql);
        return null;
    }
}
 
注意改动了构造函数，需要代入session，因为在执行时我们希望回调session了。而session中也新增一个接口：

@Override
public <T> T getMapper(Class<T> clazz) {
    return MapperProxy.newInstance(clazz, this);
}

测试代码

public static void main(String args[]) throws Exception {
    SessionFactory factory = new VSessionFactory("config.xml");
    Session session = factory.openSession();
    UserMapper mapper = session.getMapper(UserMapper.class);
    mapper.findById(1);
}