cn.featherfly.common.lang.matcher.MethodAnnotationMatcher Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of featherfly-common Show documentation
common tool
The newest version!

package cn.featherfly.common.lang.matcher;

import java.lang.reflect.Method;

import cn.featherfly.common.enums.Logic;
import cn.featherfly.common.lang.AssertIllegalArgument;
import cn.featherfly.common.lang.LangUtils;

/**
 * 
 * 匹配Method的注解的实现
 * 
 * 
 * @author zhongj
 */
public class MethodAnnotationMatcher implements MethodMatcher {

    private Class[] annotationClasses;

    /**
     * 使用并集判断逻辑来匹配.
     * 
     * @param annotationClasses 注解类型
     */
    public MethodAnnotationMatcher(Class... annotationClasses) {
        for (@SuppressWarnings("rawtypes")
        Class annotationClass : annotationClasses) {
            if (!annotationClass.isAnnotation()) {
                throw new IllegalArgumentException(String.format("类%s不是注解", annotationClass.getName()));
            }
        }
        this.annotationClasses = annotationClasses;
    }

    /**
     * 使用指定的判断逻辑来匹配. 并集，所有注解都要标注才算匹配；交集，只要有一个标注就算匹配。参见{@link Logic}
     * 
     * @param logic             判断逻辑
     * @param annotationClasses 注解类型
     */
    public MethodAnnotationMatcher(Logic logic, Class... annotationClasses) {
        this(annotationClasses);
        AssertIllegalArgument.isNotNull(logic, "Logic logic");
        this.logic = logic;
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public boolean match(Method field) {
        if (LangUtils.isEmpty(annotationClasses) || field == null) {
            return false;
        }
        if (logic == Logic.AND) {
            //并集，所有注解都要标注
            return matchAnd(field);
        } else {
            //交集，只要有一个标注就行
            return matchOr(field);
        }
    }

    @SuppressWarnings("unchecked")
    private boolean matchAnd(Method field) {
        for (@SuppressWarnings("rawtypes")
        Class annotationClass : annotationClasses) {
            if (field.getAnnotation(annotationClass) == null) {
                return false;
            }
        }
        return true;
    }

    @SuppressWarnings("unchecked")
    private boolean matchOr(Method field) {
        for (@SuppressWarnings("rawtypes")
        Class annotationClass : annotationClasses) {
            if (field.getAnnotation(annotationClass) != null) {
                return true;
            }
        }
        return false;
    }

    private Logic logic = Logic.AND;

}