/**
* ==============================================
* Copy right 2015-2018 by ja
* ----------------------------------------------
* This is not a free software, without any authorization is not allowed to use and spread.
* ==============================================
*
* @author : Jalan
* @version : v1.0.0
* @desc : 把对象属性抽象出来,可做为参数传递
* @since : 2018/2/7 21:07
*/
@FunctionalInterface
public interface ObjectPropertyPredicate<T> {
/**
* 把属性当参数传递到方法中,由方法去处理这个属性的值做什么。
* 传参使用: o -> o.propertyName
* 接收参数方法内使用:
* 参数:ObjectPropertyPredicate<FreeReportDataVO> express
* Predicate<Object> press = (obj) -> express.getProperty(obj).toString().startsWith("测试");
* express.getProperty(obj)
* @param o
* @return
*/
Object getProperty(T o);
}
/**
*
* ==============================================
* Copy right 2015-2018 by Jalan
* ----------------------------------------------
* This is not a free software, without any authorization is not allowed to use and spread.
* ==============================================
*
* @author : Jalan
* @version : v1.0.0
* @desc : 查询选项
* @since : 2018/2/7 19:28
*/
public enum FilterModeEnum {
startsWith ,
equals,
contains
}
/**
*
* ==============================================
* Copy right 2015-2018 by Jalan
* ----------------------------------------------
* This is not a free software, without any authorization is not allowed to use and spread.
* ==============================================
*
* @author : Jalan
* @version : v1.0.0
* @desc : 测试实体类
* @since : 2018/2/7 11:58
*/
@ApiModel(description = "测试实体类")
@Data
@EqualsAndHashCode(callSuper = true)
public class TestDataVO{
private String code;
private String name;
private String remark;
}
/**
* 通过批量过滤条件(codes) 过滤指定对象属性的值
* @param dataSource 数据源集合
* @param codes 过滤条件集合
* @param express 过滤集合对象目标属性表达式
* @param filterMode 过滤方式
* @return 返回符合过滤条件codes的对象集合
*/
public static List<TestDataVO> queryArray(List<TestDataVO> dataSource, List<String> codes, ObjectPropertyPredicate<TestDataVO> express, FilterModeEnum filterMode) {
List<TestDataVO> result = new ArrayList<>();
if (dataSource == null) {
return result;
}
if (codes == null || (long) codes.size() == 0) {
return dataSource;
}
if (express == null) {
return dataSource;
}
if (filterMode == FilterModeEnum.startsWith) {
Predicate<TestDataVO> expressOr = f -> false;
for (String code : codes) {
Predicate<TestDataVO> press = (w) -> express.getProperty(w).toString().startsWith(code);
expressOr = expressOr.or(press);
}
result = dataSource.stream().filter(expressOr).collect(Collectors.toList());
} else if (filterMode == FilterModeEnum.contains) {
result = dataSource.stream().filter(f -> codes.contains(express.getProperty(f).toString())).collect(Collectors.toList());
} else if (filterMode == FilterModeEnum.equals) {
Predicate<TestDataVO> expressOr = f -> false;
for (String code : codes) {
Predicate<TestDataVO> press = (w) ->StringUtils.equals(express.getProperty(w).toString(),code);
expressOr = expressOr.or(press);
}
result = dataSource.stream().filter(expressOr).collect(Collectors.toList());
}
return result;
}