前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >byte-buddy实现mybatis-plus动态mapper

byte-buddy实现mybatis-plus动态mapper

作者头像
阿超
发布2023-01-12 15:01:36
6400
发布2023-01-12 15:01:36
举报
文章被收录于专栏:快乐阿超快乐阿超

掌握了教育,就掌握了国家——法国谚语

构造动态mapper部分代码如下:

代码语言:javascript
复制
public static void buildMapper(Configuration configuration, Class<?> entityClass) {
    if (!(configuration instanceof MybatisConfiguration)) {
        throw new IllegalArgumentException("configuration must be MybatisConfiguration");
    }
    ENTITY_MAPPER_CLASS_CACHE.computeIfAbsent(entityClass, k -> {
        Class<?> dynamicMapper = new ByteBuddy()
                .makeInterface(TypeDescription.Generic.Builder.parameterizedType(IMapper.class, entityClass).build())
                .name(String.format("io.github.vampireachao.mapper.%sMapper", entityClass.getSimpleName()))
                .make()
                .load(ClassUtils.class.getClassLoader())
                .getLoaded();
        configuration.addMapper(dynamicMapper);
        return dynamicMapper;
    });
}

这里是放到一个map里,获取的地方如下:

代码语言:javascript
复制
public static <T> Class<?> getMapperClass(Class<T> clazz) {
    if (clazz == null || clazz.isPrimitive() || SimpleTypeRegistry.isSimpleType(clazz) || clazz.isInterface()) {
        throw ExceptionUtils.mpe("找不到指定的class!请仅在明确确定会有 class 的时候,调用该方法");
    }
    Class<?> targetClass = ClassUtils.getUserClass(clazz);
    Class<?> mapperClass = ENTITY_MAPPER_CLASS_CACHE.get(targetClass);
    if (null != mapperClass) {
        return mapperClass;
    }
    Class<?> currentClass = clazz;
    while (null == mapperClass && Object.class != currentClass) {
        currentClass = currentClass.getSuperclass();
        mapperClass = ENTITY_MAPPER_CLASS_CACHE.get(ClassUtils.getUserClass(currentClass));
    }
    if (mapperClass == null) {
        mapperClass = ClassUtils.toClassConfident(getTableInfo(clazz).getCurrentNamespace());
    }
    if (mapperClass != null) {
        ENTITY_MAPPER_CLASS_CACHE.put(targetClass, mapperClass);
    }
    return mapperClass;
}

使用:

代码语言:javascript
复制
@Test
void testBuildMapper() {
    UserInfo userInfo = new UserInfo() {{
        setId(1L);
        setName("Jone");
    }};
    Configuration configuration = TableInfoHelper.getTableInfo(UserInfo.class).getConfiguration();
    Database.buildMapper(configuration, UserInfo.class);
    TableInfo tableInfo = TableInfoHelper.getTableInfo(userInfo.getClass());
    Assertions.assertNotNull(tableInfo);
    Assertions.assertFalse(Database.list(userInfo.getClass()).isEmpty());
}

完整源码:https://gitee.com/VampireAchao/stream-query

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2023-01-11,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档