mybatis多条件in查询拓展
创始人
2024-11-13 10:07:39

背景

最近碰上有个业务，查询的sql如下：

select * from table where (sku_id,batch_no) in ((#{skuId},#{batchNo}),...);

本来也没什么，很简单常见的一种sql。问题是我们使用的是mybatis-plus，然后写的时候又没有考虑到后面的查询条件，这里用的是mybatis-plus lambda的方式。

LambdaQueryChainWrapper query = tableService.lambdaQuery();   query.eq(Table::getId, param.getId());  
但是mysql-plus并没有支持这种sql的形式，要么用apply方法自定义拼接sql，要么不采用lambda方式，将语句写成 xml 形式。
不过，第一种方式感觉很 low，一大段 java 代码里插入一段sql字符串，看上去就很别扭，因为有点代码洁癖，只能果断放弃。
第二种改动又太大，前面那么多查询条件，又要全部移入xml里面，同时，本来已经通过测试的筛选条件，又要重新来一遍，太懒，实在干不动了。
想想这么常见的场景，网上理应有现成的解决方案。但不知道是搜索关键字不对还是确实没有，搜了半天没搜出来。
最后无奈，只能尝试自己扩展一下。
代码
import com.baomidou.mybatisplus.core.conditions.AbstractLambdaWrapper; import com.baomidou.mybatisplus.core.conditions.ISqlSegment; import com.baomidou.mybatisplus.core.conditions.segments.MergeSegments; import com.baomidou.mybatisplus.core.metadata.TableFieldInfo; import com.baomidou.mybatisplus.core.toolkit.ArrayUtils; import com.baomidou.mybatisplus.core.toolkit.StringPool; import com.baomidou.mybatisplus.core.toolkit.TableInfoHelper; import com.baomidou.mybatisplus.core.toolkit.support.SFunction; import io.vavr.Tuple; import org.springframework.util.Assert;  import java.util.List; import java.util.Map; import java.util.concurrent.atomic.AtomicInteger; import java.util.function.Predicate; import java.util.stream.Collectors;  import static com.baomidou.mybatisplus.core.enums.SqlKeyword.IN; import static java.util.stream.Collectors.joining;  public class CombinationLambdaQueryChainWrapper extends AbstractLambdaWrapper> {      /**      * 查询字段      */     private String sqlSelect;      public CombinationLambdaQueryChainWrapper() {         this(null);     }      public CombinationLambdaQueryChainWrapper(T entity) {         this.entity = entity;         this.initEntityClass();         this.initNeed();     }      CombinationLambdaQueryChainWrapper(T entity, Class entityClass, String sqlSelect, AtomicInteger paramNameSeq, Map paramNameValuePairs,                                        MergeSegments mergeSegments) {         this.entity = entity;         this.paramNameSeq = paramNameSeq;         this.paramNameValuePairs = paramNameValuePairs;         this.expression = mergeSegments;         this.sqlSelect = sqlSelect;         this.entityClass = entityClass;     }      /**      * 
      * SELECT 部分 SQL 设置      * 
      *      * @param columns 查询字段      */     @SafeVarargs     public final CombinationLambdaQueryChainWrapper select(SFunction... columns) {         if (ArrayUtils.isNotEmpty(columns)) {             this.sqlSelect = this.columnsToString(columns);         }         return typedThis;     }      public CombinationLambdaQueryChainWrapper select(Predicate predicate) {         return select(entityClass, predicate);     }      /**      * 
      * 过滤查询的字段信息(主键除外!)      * 
      * 
      * 例1: 只要 java 字段名以 "test" 开头的              -> select(i -> i.getProperty().startsWith("test"))      * 例2: 只要 java 字段属性是 CharSequence 类型的       -> select(TableFieldInfo::isCharSequence)      * 例3: 只要 java 字段没有填充策略的                   -> select(i -> i.getFieldFill == FieldFill.DEFAULT)      * 例4: 要全部字段                                   -> select(i -> true)      * 例5: 只要主键字段                                 -> select(i -> false)      * 
      *      * @param predicate 过滤方式      * @return this      */     public CombinationLambdaQueryChainWrapper select(Class entityClass, Predicate predicate) {         this.entityClass = entityClass;         this.sqlSelect = TableInfoHelper.getTableInfo(getCheckEntityClass()).chooseSelect(predicate);         return typedThis;     }      @Override     public String getSqlSelect() {         return sqlSelect;     }      /**      * 
      * 用于生成嵌套 sql      * 故 sqlSelect 不向下传递      * 
      */     @Override     protected CombinationLambdaQueryChainWrapper instance(AtomicInteger paramNameSeq, Map paramNameValuePairs) {         return new CombinationLambdaQueryChainWrapper<>(entity, entityClass, null, paramNameSeq, paramNameValuePairs, new MergeSegments());     }      /**      * 组合IN查询      * 此处引入元祖，需使用io.vavr包      **/     public CombinationLambdaQueryChainWrapper combinationIn(boolean condition, List list, SFunction... columns) {         Assert.isTrue(columns.length == list.get(0).arity(), "请检查组合IN查询参数长度");         return doIt(condition,                 () -> StringPool.LEFT_BRACKET + columnsToString(columns) + StringPool.RIGHT_BRACKET,                 IN,                 inExpressionOfParam(list));     }       private ISqlSegment inExpressionOfParam(List list) {         if (list.size() > 1000) {             throw new RuntimeException("组合查询禁止超过1000条");         }         List result = list.stream().map(m -> StringPool.LEFT_BRACKET + m.toSeq().asJava().stream()                 .map(n -> n instanceof String ? StringPool.SINGLE_QUOTE + n + StringPool.SINGLE_QUOTE : String.valueOf(n)).collect(Collectors.joining(StringPool.COMMA)) + StringPool.RIGHT_BRACKET).collect(Collectors.toList());         return () -> result.stream()                 .collect(joining(StringPool.COMMA, StringPool.LEFT_BRACKET, StringPool.RIGHT_BRACKET));     }  } 
使用示例
        productRelService.list(new CombinationLambdaQueryChainWrapper().select(ProductRel::getLevelId).eq(ProductRel::getLevelId, "GBZ0555A0002").combinationIn(true, Arrays.asList(Tuple.of("91cf1c0f3b0bc46e0238dc33717dc88e", "GBZ0555A0002")), ProductRel::getProductUuid, ProductRel::getLevelId));         productRelService.list(new CombinationLambdaQueryChainWrapper().combinationIn(true, Arrays.asList(Tuple.of(1L, "GBZ0555A0002")), ProductRel::getId, ProductRel::getLevelId)); 

                
               
            

                

                    上一篇：算法通关：014_1：用栈实现队列

                

                

                    下一篇：Java零基础之多线程篇：如何保证线程安全？                

            

                
                
相关内容

                
            

                 

            

        
        

            
            

                
                

                    
                    

                        
                        热门资讯
                        
                    

                    
                

                
            

            
            
            


                                
                

                    
                    
                    
                    
                    

                        裸辞做“一人公司”，我后悔了
                        去年这个时候，一位以色列程序员正在东南亚旅行。他顺手把一个在脑子里转了很久的想法做成了产品，一个让任...
                    

                    
                

                                
                

                    
                    
                    
                    
                    

                        南京建成国内首个Pre-6G试...
                        4月21日，2026全球6G技术与产业生态大会在南京开幕。全息互动技术展台前，一名远在北京的工作人员...
                    

                    
                

                                
                

                    
                    
                    
                    
                    

                        超梵求职受邀参加“2025抖音...
                        超梵求职受邀参加“2025抖音巨量引擎成人教育行业生态大会”，探讨分享优质内容传播，服务万千学员。 ...
                    

                    
                

                                
                

                    
                    
                    
                    
                    

                        摩托罗拉Razr 2026（R...
                        IT之家 4 月 22 日消息，摩托罗拉宣布新一代 Razr 折叠手机将于 4 月 29 日在美国发...
                    

                    
                

                                
                

                    
                    
                    
                    
                    

                        库克卸任，特纳斯领航：苹果新纪...
                        苹果首席执行官蒂姆·库克将卸任，硬件工程主管约翰·特纳斯将接任，苹果公司今天宣布此事。 库克将在夏季...