com.itxiaoer.commons.jpa.LogicalCriterion Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of commons-jpa Show documentation

There is a newer version: 2.3.4

package com.itxiaoer.commons.jpa;

import com.itxiaoer.commons.core.Operator;
import com.itxiaoer.commons.core.ParameterException;

import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
import java.util.List;
import java.util.stream.Collectors;

/**
 * @author : liuyk
 */
@SuppressWarnings({"unused", "WeakerAccess"})
public class LogicalCriterion implements Criterion {

    /**
     * 逻辑表达式中包含的表达式
     */
    private List criterion;
    /**
     * 计算符
     */
    private Operator operator;

    public LogicalCriterion(List criterion, Operator operator) {
        this.criterion = criterion;
        this.operator = operator;
    }

    @Override
    public Predicate toPredicate(Root root, CriteriaQuery query,
                                 CriteriaBuilder builder) {
        List predicates = criterion.stream().map(c -> c.toPredicate(root, query, builder)).collect(Collectors.toList());
        switch (operator) {
            case OR:
                return builder.or(predicates.toArray(new Predicate[0]));
            case AND:
                return builder.and(predicates.toArray(new Predicate[0]));
            default:
                throw new ParameterException("operator not support.");
        }
    }
}