java

关注公众号 jb51net

关闭
首页 > 软件编程 > java > 反射生成 MyBatisPlus中QueryWrapper动态条件

如何利用反射生成 MyBatisPlus中QueryWrapper动态条件

作者:梁云亮

这篇文章主要介绍了如何利用反射生成 MyBatisPlus中QueryWrapper动态条件,分享在MyBatisPlus中经常会用到代码来构造查询条件等内容,需要的小伙伴可以参考一下

1.问题

在MyBatisPlus中经常会用到如下所示的代码来构造查询条件:

QueryWrapper<User> queryWrapper = new QueryWrapper<>();

if (userQuery.getId() != null) {
    queryWrapper.eq("id", userQuery.getId());
}
if (userQuery.getName() != null) {
    queryWrapper.like("name", userQuery.getName());
}
if (userQuery.getNickname() != null) {
    queryWrapper.like("nickname", userQuery.getNickname());
}
if (userQuery.getAccount() != null) {
    queryWrapper.eq("account", userQuery.getAccount());
}
if (userQuery.getGender() != null) {
    queryWrapper.eq("gender", userQuery.getGender());
}
if (userQuery.getBirth1() != null) {
    queryWrapper.ge("birth", userQuery.getBirth1());
}
if (userQuery.getBirth2() != null) {
    queryWrapper.le("birth", userQuery.getBirth2());
}
if (userQuery.getTel() != null) {
    queryWrapper.eq("tel", userQuery.getTel());
}
if (userQuery.getEmail() != null) {
    queryWrapper.eq("email", userQuery.getEmail());
}
if (userQuery.getQq() != null) {
    queryWrapper.eq("qq", userQuery.getQq());
}
if (userQuery.getWechat() != null) {
    queryWrapper.eq("wechat", userQuery.getWechat());
}
if (userQuery.getHomePage() != null) {
    queryWrapper.eq("home_page", userQuery.getHomePage());
}
//.........
baseMapper.selectPage(page, queryWrapper);

大量这样的代码,可以通过如下所示的反射技术生成,从而一劳永安逸,

2.优化方案

public class APP{

    public static void main(String[] args) throws Exception {
        QueryWrapper<Dept> queryWrapper = new QueryWrapper<>();
        Dept dept = Dept.builder()
                .deptno(1234)
//                .dname("sales")
                .loc("NEWYORK")
                .build();

        Method[] methods = Dept.class.getDeclaredMethods();
        for (Method method : methods) {
            String methodName = method.getName();
            if (methodName.contains("get")) {
                Object res = method.invoke(dept, null);
                if(res != null){
                    String columnName = getColumnName(methodName);
                    queryWrapper.eq(columnName,res);
                }
            }
        }

        System.out.println(queryWrapper);//在此处设置断点,可以查看保存到QueryWrapper中的值
    }

    // 根据getter方法名得到对应表的字段名
    public static String getColumnName(String name) {
        StringBuffer sb = new StringBuffer();
        name = name.substring(3, name.length()); //去掉get前缀
        char[] chars = name.toCharArray();
        for (int i = 0; i < chars.length; i++) {
            char item = chars[i];
            if (item > 65 && item < 90) {//大写字母
                sb.append("_");
                sb.append((char)(item+32));
            }else {
                sb.append(item);
            }
        }
        return sb.deleteCharAt(0).toString();
    }
}

到此这篇关于如何利用反射生成 MyBatisPlus中QueryWrapper动态条件的文章就介绍到这了,更多相关反射生成 MyBatisPlus中QueryWrapper动态条件内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文