需求：
列表页面实现按合同状态自定义排序
private Specification<TblContractPre> whereClause(final TblContractPre entity) {
return new Specification<TblContractPre>() {
@Override
public Predicate toPredicate(Root<TblContractPre> root, CriteriaQuery<?> query, CriteriaBuilder cb) {
List<Predicate> list = new ArrayList<Predicate>();
if (entity.getId() != null) {
list.add(cb.equal(root.get("id").as(Long.class), entity.getId()));
}
if ("1".equals(entity.getFrom())) {
UserObject user = UserObjectHelper.currentUserObject();
List<String> ids = DataUtil.findAuditByCurUser(user, WorkflowType.HB_CONTRACT_ONLINE);
if (ids.isEmpty()) {
list.add(root.get("id").as(Long.class).in(0));
} else {
list.add(root.get("id").as(Long.class).in(UtilPublic.strArrToListLong(ids.toArray(new String[] {}))));
}
}
if (StringUtils.isNotBlank(entity.getHouseName())) {
list.add(cb.like(root.get("houseName").as(String.class), "%" + entity.getHouseName() + "%"));
}
if (StringUtils.isNotBlank(entity.getDeedsn())) {
list.add(cb.equal(root.get("deedsn").as(String.class), entity.getDeedsn()));
}
if (entity.getDeptId() != null) {
list.add(root.get("deptId").as(Integer.class).in(DataUtil.findSubDeptIds(entity.getDeptId())));
}
if (StringUtils.isNotBlank(entity.getSalesmanName())) {
list.add(cb.equal(root.get("salesmanName").as(String.class), entity.getSalesmanName()));
}
UserObject user = UserObjectHelper.currentUserObject();
if (user != null) {
if (user.isSales()) {
list.add(cb.equal(root.get("salesmanName").as(String.class), user.getUserCode()));
} else if (user.isBus()) {
list.add(root.get("deptId").as(Integer.class).in(user.getDeptIdList()));
}
}
if (entity.getContractStatus() != null) {
list.add(cb.equal(root.get("contractStatus").as(Byte.class), entity.getContractStatus()));
}
if (StringUtils.isNotBlank(entity.getDateType())) {
if (entity.getDeedsnStart() != null) {
list.add(cb.greaterThanOrEqualTo(root.get(entity.getDateType()).as(Date.class), entity.getDeedsnStart()));
}
if (entity.getDeedsnEnd() != null) {
list.add(cb.lessThanOrEqualTo(root.get(entity.getDateType()).as(Date.class), entity.getDeedsnEnd()));
}
}
if (entity.getCompanyType() != null) {
list.add(cb.equal(root.get("companyType").as(CompanyType.class), entity.getCompanyType()));
}
List<Order> orders = new ArrayList<>();
orders.add(cb.asc(cb.selectCase().when(cb.equal(root.get("contractStatus").as(Byte.class), 2), 1)
.when(cb.equal(root.get("contractStatus").as(Byte.class), 4), 2)
.when(cb.equal(root.get("contractStatus").as(Byte.class), 1), 3)
.when(cb.equal(root.get("contractStatus").as(Byte.class), 3), 4).otherwise(10)));
orders.add(cb.desc(root.get("createDate")));
query.orderBy(orders);
Predicate[] p = new Predicate[list.size()];
return query.where(cb.and(list.toArray(p))).getGroupRestriction();
}
};
}