springboot JPA

package com.ketayao.ketacustom.util.persistence; import java.text.ParseException; import java.text.SimpleDateFormat; import java.util.ArrayList; import java.util.Collection; import java.util.Date; import java.util.HashSet; import java.util.List; import java.util.Map; import java.util.Set; import javax.persistence.criteria.CriteriaBuilder; import javax.persistence.criteria.CriteriaQuery; import javax.persistence.criteria.Path; import javax.persistence.criteria.Predicate; import javax.persistence.criteria.Root; import javax.servlet.ServletRequest; import javax.servlet.http.HttpServletRequest; import org.apache.commons.lang3.EnumUtils; import org.apache.commons.lang3.StringUtils; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.data.jpa.domain.Specification; import com.ketayao.ketacustom.SecurityConstants; import com.ketayao.utils.Exceptions; import com.ketayao.utils.ServletUtils; public class DynamicSpecifications { private static final Logger logger = LoggerFactory.getLogger(DynamicSpecifications.class); // 用于存储每个线程的request请求 private static final ThreadLocal<HttpServletRequest> LOCAL_REQUEST = new ThreadLocal<>(); private static final String SHORT_DATE = "yyyy-MM-dd"; private static final String LONG_DATE = "yyyy-MM-dd mm:HH:ss"; private static final String TIME = "mm:HH:ss"; public static void putRequest(HttpServletRequest request) { LOCAL_REQUEST.set(request); } public static HttpServletRequest getRequest() { return LOCAL_REQUEST.get(); } public static void removeRequest() { LOCAL_REQUEST.remove(); } public static Collection<SearchFilter> genSearchFilter(ServletRequest request) { Map<String, Object> searchParams = ServletUtils.getParametersStartingWith(request, SecurityConstants.SEARCH_PREFIX); Map<String, SearchFilter> filters = SearchFilter.parse(searchParams); return filters.values(); } public static <T> Specification<T> bySearchFilter(ServletRequest request, final Class<T> entityClazz, final Collection<SearchFilter> searchFilters) { return bySearchFilter(request, entityClazz, searchFilters.toArray(new SearchFilter[] {})); } public static <T> Specification<T> bySearchFilter(ServletRequest request, final Class<T> entityClazz, final SearchFilter... searchFilters) { Collection<SearchFilter> filters = genSearchFilter(request); Set<SearchFilter> set = new HashSet<>(filters); for (SearchFilter searchFilter : searchFilters) { set.add(searchFilter); } return bySearchFilter(entityClazz, set); } @SuppressWarnings("unchecked") public static <T> Specification<T> bySearchFilter(Class<T> entityClazz, final Collection<SearchFilter> searchFilters) { final Set<SearchFilter> filterSet = new HashSet<>(); ServletRequest request = getRequest(); if (request != null) { // 数据权限中的filter Collection<SearchFilter> nestFilters = (Collection<SearchFilter>)request.getAttribute(SecurityConstants.NEST_DYNAMIC_SEARCH); if (nestFilters != null && !nestFilters.isEmpty()) { for (SearchFilter searchFilter : nestFilters) { filterSet.add(searchFilter); } } } // 自定义 for (SearchFilter searchFilter : searchFilters) { filterSet.add(searchFilter); } return new Specification<T>() { @SuppressWarnings({"rawtypes"}) @Override public Predicate toPredicate(Root<T> root, CriteriaQuery<?> query, CriteriaBuilder builder) { if (filterSet != null && !filterSet.isEmpty()) { List<Predicate> predicates = new ArrayList<>(); for (SearchFilter filter : filterSet) { // nested path translate, 如Task的名为"user.name"的filedName, 转换为Task.user.name属性 String[] names = StringUtils.split(filter.getFieldName(), "."); Path expression = root.get(names[0]); for (int i = 1; i < names.length; i++) { expression = expression.get(names[i]); } // 自动进行enum和date的转换。 Class clazz = expression.getJavaType(); if (Date.class.isAssignableFrom(clazz) && !filter.getValue().getClass().equals(clazz)) { filter.setValue(convert2Date((String)filter.getValue())); } else if (Enum.class.isAssignableFrom(clazz) && !filter.getValue().getClass().equals(clazz)) { filter.setValue(convert2Enum(clazz, (String)filter.getValue())); } // logic operator switch (filter.getOperator()) { case EQ: predicates.add(builder.equal(expression, filter.getValue())); break; case LIKE: predicates.add(builder.like(expression, "%" + filter.getValue() + "%")); break; case GT: predicates.add(builder.greaterThan(expression, (Comparable)filter.getValue())); break; case LT: predicates.add(builder.lessThan(expression, (Comparable)filter.getValue())); break; case GTE: predicates.add(builder.greaterThanOrEqualTo(expression, (Comparable)filter.getValue())); break; case LTE: predicates.add(builder.lessThanOrEqualTo(expression, (Comparable)filter.getValue())); break; case IN: predicates.add(builder.and(expression.in((Object[])filter.getValue()))); break; case ISNULL: predicates.add(builder.isNull(expression)); break; case ISNOTNULL: predicates.add(builder.isNotNull(expression)); break; } } // 将所有条件用 and 联合起来 if (!predicates.isEmpty()) { return builder.and(predicates.toArray(new Predicate[predicates.size()])); } } return builder.conjunction(); } }; } private static Date convert2Date(String dateString) { SimpleDateFormat sFormat = new SimpleDateFormat(SHORT_DATE); try { return sFormat.parse(dateString); } catch (ParseException e) { try { return sFormat.parse(LONG_DATE); } catch (ParseException e1) { try { return sFormat.parse(TIME); } catch (ParseException e2) { logger.error("Convert time is error! The dateString is {}.{} ", dateString, Exceptions.getStackTr