package cn.besteam.modules.system.util;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
import com.baomidou.mybatisplus.core.toolkit.support.SFunction;
|
import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
|
import com.baomidou.mybatisplus.extension.conditions.update.LambdaUpdateChainWrapper;
|
import org.apache.commons.lang3.ObjectUtils;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
|
/**
|
* @author xhb 2022-01-07
|
**/
|
|
public class MybatisParameterUtils {
|
|
|
public static <T, F> void cutInParameter(LambdaQueryWrapper<T> wrapper, SFunction<T, ?> column, List<F> coll) throws Exception {
|
List<List<F>> newList = splitList(coll, 900);
|
if (CollectionUtils.isEmpty(newList)) {
|
throw new Exception("参数错误");
|
} else if (newList.size() == 1) {
|
wrapper.in(column, newList.get(0));
|
return;
|
}
|
|
wrapper.and(i -> {
|
i.in(column, newList.get(0));
|
newList.remove(0);
|
for (List<F> objects : newList) {
|
i.or().in(column, objects);
|
}
|
});
|
}
|
|
public static <T, F> void cutNotInParameter(LambdaQueryWrapper<T> wrapper, SFunction<T, ?> column, List<F> coll) throws Exception {
|
List<List<F>> newList = splitList(coll, 900);
|
if (CollectionUtils.isEmpty(newList)) {
|
throw new Exception("参数错误");
|
} else if (newList.size() == 1) {
|
wrapper.notIn(column, newList.get(0));
|
return;
|
}
|
|
wrapper.and(i -> {
|
i.in(column, newList.get(0));
|
newList.remove(0);
|
for (List<F> objects : newList) {
|
i.or().notIn(column, objects);
|
}
|
});
|
}
|
|
|
public static <T, F> void cutInParameter(LambdaQueryChainWrapper<T> wrapper, SFunction<T, ?> column, List<F> coll) throws Exception {
|
List<List<F>> newList = splitList(coll, 900);
|
if (CollectionUtils.isEmpty(newList)) {
|
throw new Exception("参数错误");
|
} else if (newList.size() == 1) {
|
wrapper.in(column, newList.get(0));
|
return;
|
}
|
|
wrapper.and(i -> {
|
i.in(column, newList.get(0));
|
newList.remove(0);
|
for (List<F> objects : newList) {
|
i.or().in(column, objects);
|
}
|
});
|
}
|
|
public static <T, F> void cutNotInParameter(LambdaQueryChainWrapper<T> wrapper, SFunction<T, ?> column, List<F> coll) throws Exception {
|
List<List<F>> newList = splitList(coll, 900);
|
if (CollectionUtils.isEmpty(newList)) {
|
throw new Exception("参数错误");
|
} else if (newList.size() == 1) {
|
wrapper.notIn(column, newList.get(0));
|
return;
|
}
|
|
wrapper.and(i -> {
|
i.in(column, newList.get(0));
|
newList.remove(0);
|
for (List<F> objects : newList) {
|
i.or().notIn(column, objects);
|
}
|
});
|
}
|
|
public static <T, F> void cutInParameter(LambdaUpdateWrapper<T> wrapper, SFunction<T, ?> column, List<F> coll) throws Exception {
|
List<List<F>> newList = splitList(coll, 900);
|
if (CollectionUtils.isEmpty(newList)) {
|
throw new Exception("参数错误");
|
} else if (newList.size() == 1) {
|
wrapper.in(column, newList.get(0));
|
return;
|
}
|
|
wrapper.and(i -> {
|
i.in(column, newList.get(0));
|
newList.remove(0);
|
for (List<F> objects : newList) {
|
i.or().in(column, objects);
|
}
|
});
|
}
|
|
public static <T, F> void cutNotInParameter(LambdaUpdateWrapper<T> wrapper, SFunction<T, ?> column, List<F> coll) throws Exception {
|
List<List<F>> newList = splitList(coll, 900);
|
if (CollectionUtils.isEmpty(newList)) {
|
throw new Exception("参数错误");
|
} else if (newList.size() == 1) {
|
wrapper.notIn(column, newList.get(0));
|
return;
|
}
|
|
wrapper.and(i -> {
|
i.in(column, newList.get(0));
|
newList.remove(0);
|
for (List<F> objects : newList) {
|
i.or().notIn(column, objects);
|
}
|
});
|
}
|
|
|
public static <T, F> void cutInParameter(LambdaUpdateChainWrapper<T> wrapper, SFunction<T, ?> column, List<F> coll) throws Exception {
|
List<List<F>> newList = splitList(coll, 900);
|
if (CollectionUtils.isEmpty(newList)) {
|
throw new Exception("参数错误");
|
} else if (newList.size() == 1) {
|
wrapper.in(column, newList.get(0));
|
return;
|
}
|
|
wrapper.and(i -> {
|
i.in(column, newList.get(0));
|
newList.remove(0);
|
for (List<F> objects : newList) {
|
i.or().in(column, objects);
|
}
|
});
|
}
|
|
public static <T, F> void cutNotInParameter(LambdaUpdateChainWrapper<T> wrapper, SFunction<T, ?> column, List<F> coll) throws Exception {
|
List<List<F>> newList = splitList(coll, 900);
|
if (CollectionUtils.isEmpty(newList)) {
|
throw new Exception("参数错误");
|
} else if (newList.size() == 1) {
|
wrapper.notIn(column, newList.get(0));
|
return;
|
}
|
|
wrapper.and(i -> {
|
i.in(column, newList.get(0));
|
newList.remove(0);
|
for (List<F> objects : newList) {
|
i.or().notIn(column, objects);
|
}
|
});
|
}
|
|
|
public static <F> List<List<F>> splitList(List<F> list, int groupSize) {
|
int length = list.size();
|
// 计算可以分成多少组
|
int num = (length + groupSize - 1) / groupSize;
|
List<List<F>> newList = new ArrayList<>(num);
|
for (int i = 0; i < num; i++) {
|
// 开始位置
|
int fromIndex = i * groupSize;
|
// 结束位置
|
int toIndex = Math.min((i + 1) * groupSize, length);
|
newList.add(list.subList(fromIndex, toIndex));
|
}
|
return newList;
|
}
|
}
|
