top.openyuan.jpa.specification.handler.impl.InPredicateHandler Maven / Gradle / Ivy

Go to download
package top.openyuan.jpa.specification.handler.impl;

import top.openyuan.jpa.specification.annotation.In;
import top.openyuan.jpa.specification.handler.AbstractPredicateHandler;

import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.From;
import javax.persistence.criteria.Predicate;
import java.lang.annotation.Annotation;

/**
 * 构建“范围匹配条件”({@code field IN ('xxx', 'yyy')})场景的 {@link Predicate} 处理器.
 *
 * @author lzy
 * @since v1.0.0
 */
public class InPredicateHandler extends AbstractPredicateHandler {

    @Override
    public Class getAnnotation() {
        return In.class;
    }

    @Override
    public  Predicate buildPredicate(
            CriteriaBuilder criteriaBuilder, From from, String fieldName, Object value, Annotation annotation) {
        return criteriaBuilder.and(
                super.buildInPredicate(criteriaBuilder, from, fieldName, value, super.isAllowNull(annotation)));
    }

    @Override
    public Predicate buildPredicate(
            CriteriaBuilder criteriaBuilder, From from, String fieldName, Object value) {
        return criteriaBuilder.and(super.buildInPredicate(criteriaBuilder, from, fieldName, value, false));
    }

}