mybatis多条件in查询的实现
作者：JAVA拾贝
使用MyBatis-Plus的Lambda方法进行查询时,遇到多条件IN查询的问题,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
最近碰上有个业务，查询的sql如下：
select * from table where (sku_id,batch_no) in ((#{skuId},#{batchNo}),...); 本来也没什么，很简单常见的一种sql。问题是我们使用的是mybatis-plus，然后写的时候又没有考虑到后面的查询条件，这里用的是mybatis-plus lambda的方式。
LambdaQueryChainWrapper<Table> 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<T> extends AbstractLambdaWrapper<T, CombinationLambdaQueryChainWrapper<T>> {
/**
* 查询字段
*/
private String sqlSelect;
public CombinationLambdaQueryChainWrapper() {
this(null);
}
public CombinationLambdaQueryChainWrapper(T entity) {
this.entity = entity;
this.initEntityClass();
this.initNeed();
}
CombinationLambdaQueryChainWrapper(T entity, Class<T> entityClass, String sqlSelect, AtomicInteger paramNameSeq, Map<String, Object> paramNameValuePairs,
MergeSegments mergeSegments) {
this.entity = entity;
this.paramNameSeq = paramNameSeq;
this.paramNameValuePairs = paramNameValuePairs;
this.expression = mergeSegments;
this.sqlSelect = sqlSelect;
this.entityClass = entityClass;
}
/**
* <p>
* SELECT 部分 SQL 设置
* </p>
*
* @param columns 查询字段
*/
@SafeVarargs
public final CombinationLambdaQueryChainWrapper<T> select(SFunction<T, ?>... columns) {
if (ArrayUtils.isNotEmpty(columns)) {
this.sqlSelect = this.columnsToString(columns);
}
return typedThis;
}
public CombinationLambdaQueryChainWrapper<T> select(Predicate<TableFieldInfo> predicate) {
return select(entityClass, predicate);
}
/**
* <p>
* 过滤查询的字段信息(主键除外!)
* </p>
* <p>
* 例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)
* </p>
*
* @param predicate 过滤方式
* @return this
*/
public CombinationLambdaQueryChainWrapper<T> select(Class<T> entityClass, Predicate<TableFieldInfo> predicate) {
this.entityClass = entityClass;
this.sqlSelect = TableInfoHelper.getTableInfo(getCheckEntityClass()).chooseSelect(predicate);
return typedThis;
}
@Override
public String getSqlSelect() {
return sqlSelect;
}
/**
* <p>
* 用于生成嵌套 sql
* 故 sqlSelect 不向下传递
* </p>
*/
@Override
protected CombinationLambdaQueryChainWrapper<T> instance(AtomicInteger paramNameSeq, Map<String, Object> paramNameValuePairs) {
return new CombinationLambdaQueryChainWrapper<>(entity, entityClass, null, paramNameSeq, paramNameValuePairs, new MergeSegments());
}
/**
* 组合IN查询
* 此处引入元组，需使用io.vavr包
**/
public CombinationLambdaQueryChainWrapper<T> combinationIn(boolean condition, List<Tuple> list, SFunction<T, ?>... 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<Tuple> list) {
if (list.size() > 1000) {
throw new RuntimeException("组合查询禁止超过1000条");
}
List<String> 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<ProductRel>().select(ProductRel::getLevelId).eq(ProductRel::getLevelId, "GBZ0555A0002").combinationIn(true, Arrays.asList(Tuple.of("91cf1c0f3b0bc46e0238dc33717dc88e", "GBZ0555A0002")), ProductRel::getProductUuid, ProductRel::getLevelId));
productRelService.list(new CombinationLambdaQueryChainWrapper<ProductRel>().combinationIn(true, Arrays.asList(Tuple.of(1L, "GBZ0555A0002")), ProductRel::getId, ProductRel::getLevelId));到此这篇关于mybatis多条件in查询的实现的文章就介绍到这了,更多相关mybatis多条件in查询内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！