版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/qq_27088383/article/details/84566558
package com.topsec.tsm.util;
import com.google.gson.reflect.TypeToken;
import com.topsec.tsm.common.WhereEqual;
import org.apache.commons.lang.StringUtils;
import org.springframework.data.jpa.domain.Specification;
import javax.persistence.criteria.*;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;
/**
* Created by lx1 on 2017/5/25.
*/
public class ConvertUtil<T> {
public Specification<T> getWhereClause(final Map<String, String> params) {
final Map<String, String> paramsa = params;
return new Specification<T>() {
/**
*
* @param root 定义查询的From子句中能出现的类型
* @param criteriaQuery 安全查询主语句
* @param criteriaBuilder 安全查询创建工厂
* @return
*/
@Override
public Predicate toPredicate(Root<T> root, CriteriaQuery<?> criteriaQuery, CriteriaBuilder criteriaBuilder) {
// or 条件
List<Predicate> predicates = new ArrayList<Predicate>();
// 创建具体查询条件
Predicate predicate = criteriaBuilder.conjunction();
for (Map.Entry<String, String> p : paramsa.entrySet()) {
if (StringUtils.isNotBlank(p.getValue()) && (p.getKey().equals("t_responsible") || p.getKey().equals("task_founder"))) {
System.out.println(" ===>>> 分页条件查询进入 or");
predicates.add(criteriaBuilder.equal(root.<String>get(p.getKey()), StringUtils.trim(p.getValue())));
} else if (StringUtils.isNotBlank(p.getValue()) && (p.getKey().equals("ipProto"))) {
System.out.println(" ===>>> ipProto分页条件查询进入 or");
String ipProto = p.getValue();
List<String> unjson = JsonUtil.unjson(ipProto, new TypeToken<List<String>>() {
}.getType());
for (String s : unjson) {
predicates.add(criteriaBuilder.like(root.<String>get(p.getKey()), "%" + s + "%"));
}
} else if (StringUtils.isNotBlank(p.getValue()) && !p.getKey().contains("Time") && !p.getKey().contains("starIpLong") && !p.getKey().contains("endIpLong")) {
System.out.println(" ===>>> 分页条件查询进入 and");
if (WhereEqual.equalSql.containsKey(p.getKey())) {
// 精确搜索
predicate.getExpressions().add(criteriaBuilder.equal(root.<String>get(p.getKey()), StringUtils.trim(p.getValue())));
} else {
// 模糊搜索
predicate.getExpressions().add(criteriaBuilder.like(root.<String>get(p.getKey()), "%" + StringUtils.trim(p.getValue()) + "%"));
}
}
}
if (predicates.size() > 0) {
predicate.getExpressions().add(criteriaBuilder.or(predicates.toArray(new Predicate[predicates.size()])));
}
//此处时间格式一定要和页面时间设置的格式一样，如不同，时间转换会出问题
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
Date startTime = null; //页面传过来的参数值
if (StringUtils.isNotBlank(paramsa.get("starIpLong"))) {
// 前端页面时间选择项选择第一个开始时间、第二个开始时间和第一个结束时间、第二个结束时间
try {
Long starIpLong = (JskgUtil.ipToLong(paramsa.get("starIpLong")));
predicate.getExpressions().add(criteriaBuilder.greaterThanOrEqualTo(root.<Long>get("ipLong"), starIpLong));
} catch (Exception e) {
e.printStackTrace();
}
}
if (StringUtils.isNotBlank(paramsa.get("endIpLong"))) {
// 前端页面时间选择项选择第一个开始时间、第二个开始时间和第一个结束时间、第二个结束时间
try {
Long endIpLong = JskgUtil.ipToLong(paramsa.get("endIpLong"));
predicate.getExpressions().add(criteriaBuilder.lessThanOrEqualTo(root.<Long>get("ipLong"), endIpLong));
} catch (Exception e) {
e.printStackTrace();
}
}
if (StringUtils.isNotBlank(paramsa.get("startTime"))) {
// 前端页面时间选择项选择第一个开始时间、第二个开始时间和第一个结束时间、第二个结束时间
try {
startTime = sdf.parse((String) paramsa.get("startTime"));
predicate.getExpressions().add(criteriaBuilder.greaterThanOrEqualTo(root.<Date>get("startTime"), startTime));
} catch (Exception e) {
e.printStackTrace();
}
}
if (StringUtils.isNotBlank(paramsa.get("startTime1"))) {
// 前端页面时间选择项选择第一个开始时间、第二个开始时间和第一个结束时间、第二个结束时间
try {
startTime = sdf.parse((String) paramsa.get("startTime1"));
predicate.getExpressions().add(criteriaBuilder.lessThanOrEqualTo(root.<Date>get("startTime"), startTime));
} catch (Exception e) {
e.printStackTrace();
}
}
if (StringUtils.isNotBlank(paramsa.get("endTime"))) {
// 前端页面时间选择项选择第一个开始时间、第二个开始时间和第一个结束时间、第二个结束时间
try {
startTime = sdf.parse((String) paramsa.get("endTime"));
predicate.getExpressions().add(criteriaBuilder.greaterThanOrEqualTo(root.<Date>get("endTime"), startTime));
} catch (Exception e) {
e.printStackTrace();
}
}
if (StringUtils.isNotBlank(paramsa.get("endTime1"))) {
// 前端页面时间选择项选择第一个开始时间、第二个开始时间和第一个结束时间、第二个结束时间
try {
startTime = sdf.parse((String) paramsa.get("endTime1"));
predicate.getExpressions().add(criteriaBuilder.lessThanOrEqualTo(root.<Date>get("endTime"), startTime));
} catch (Exception e) {
e.printStackTrace();
}
}
return predicate;
}
};
}
}