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

书欣 MyBatis 2022-08-15 16:39:04 17889 1
下文笔者讲述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);
}
版权声明

本文仅代表作者观点,不代表本站立场。
本文系作者授权发表,未经许可,不得转载。

本文链接: http://www.Java265.com/JavaFramework/MyBatis/202208/4240.html

最近发表

热门文章

好文推荐

Java265.com

//www.java265.com

站长统计|USA-001

Powered By Java265.com信息维护小组

使用手机扫描二维码

关注我们看更多资讯

java爱好者