过滤器（十一）之ValidationFilter

本文基于 Dubbo 2.6.1 版本，望知悉。

1. 概述

本文分享 dubbo-filter-validation 项目的 ValidationFilter 过滤器，用于服务消费者和提供者中，提供 参数验证 的功能。在 《Dubbo 用户指南 —— 参数验证》 定义如下：

参数验证功能，是基于 JSR303 Bean Validation 实现的，用户只需标识 JSR303 标准的验证 annotation，并通过声明 filter 来实现验证。

• 配置示例和，官方文档已经写的很齐全，笔者就不多哔哔了。
• 如下是新版本的 Maven 依赖的例子：

<!-- JSR 303 - Bean Validation begin -->
<dependency>
    <groupId>javax.validation</groupId>
    <artifactId>validation-api</artifactId>
    <version>1.1.0.Final</version>
</dependency>
<dependency>
    <groupId>org.hibernate</groupId>
    <artifactId>hibernate-validator</artifactId>
    <version>5.3.4.Final</version>
</dependency>
<dependency>
    <groupId>javax.el</groupId>
    <artifactId>javax.el-api</artifactId>
    <version>2.2.4</version>
</dependency>
<dependency>
    <groupId>org.glassfish.web</groupId>
    <artifactId>javax.el</artifactId>
    <version>2.2.4</version>
</dependency>
<!-- JSR 303 - Bean Validation end -->

本文涉及的类，如下图所示：

2. ValidationFilter

com.alibaba.dubbo.validation.filter.ValidationFilter ，实现 Filter 接口，参数验证过滤器实现类。代码如下：

@Activate(group = {Constants.CONSUMER, Constants.PROVIDER}, value = Constants.VALIDATION_KEY, order = 10000)
public class ValidationFilter implements Filter {

    /**
     * Validation$Adaptive 对象
     *
     * 通过 Dubbo SPI 机制，调用 {@link #setValidation(Validation)} 方法，进行注入
     */
    private Validation validation;

    public void setValidation(Validation validation) {
        this.validation = validation;
    }

    @Override
    public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
        if (validation != null && !invocation.getMethodName().startsWith("$") // 非泛化调用和回音调用等方法
                && ConfigUtils.isNotEmpty(invoker.getUrl().getMethodParameter(invocation.getMethodName(), Constants.VALIDATION_KEY))) { // 方法开启 Validation 功能
            try {
                // 获得 Validator 对象
                Validator validator = validation.getValidator(invoker.getUrl());
                // 使用 Validator ，验证方法参数。若不合法，抛出异常。
                if (validator != null) {
                    validator.validate(invocation.getMethodName(), invocation.getParameterTypes(), invocation.getArguments());
                }
            } catch (RpcException e) {
                throw e;
            } catch (Throwable t) {
                return new RpcResult(t);
            }
        }
        // 服务调用
        return invoker.invoke(invocation);
    }

}

• 第 17 行：非泛化调用和回音调用等方法。
• 第 18 行：判断方法开启部分 Validation 功能。因为，一个服务里，可能只有方法开启了 Validation 功能。
• 第 21 行：调用 URL Validation$Adaptive#getValidator(url) 方法，基于为维度，获得 Validator 对象。
• 第 22 至 25 行：调用 Validator#validate(String methodName, Class<?>[] parameterTypes, Object[] arguments) 方法，方法参数验证。若不合法，抛出异常。
• 第 33 行：调用 Invoker#invoke(invocation) 方法，服务调用。

3. API 定义

3.1 Validator

com.alibaba.dubbo.validation.Validator ，验证器接口。代码如下：

public interface Validator {

    /**
     * 方法参数验证
     *
     * @param methodName 方法名
     * @param parameterTypes 参数类型数组
     * @param arguments 参数值数组
     * @throws Exception 当发生异常时
     */
    void validate(String methodName, Class<?>[] parameterTypes, Object[] arguments) throws Exception;

}

3.2 Validation

com.alibaba.dubbo.validation.Validation ，Validator 工厂接口。代码如下：

@SPI("jvalidation")
public interface Validation {

    /**
     * 获得 Validator 对象
     *
     * @param url URL
     * @return Validator
     */
    @Adaptive(Constants.VALIDATION_KEY)
    Validator getValidator(URL url);

}

• @SPI("jvalidation") 拓展点注解，Dubbo SPI，默认为 "jvalidation"。
• @Adaptive("validation") 注解，基于 Dubbo SPI Adaptive 机制，加载对应的 Validator 实现，使用 URL.validation 属性。

3.2.1 AbstractValidation

com.alibaba.dubbo.validation.support.AbstractValidation ，实现 Validation 接口，Validator 工厂抽象类。代码如下：

public abstract class AbstractValidation implements Validation {

    /**
     * Validator 集合
     *
     * key：URL
     */
    private final ConcurrentMap<String, Validator> validators = new ConcurrentHashMap<String, Validator>();

    @Override
    public Validator getValidator(URL url) {
        // 获得 Validator 对象
        String key = url.toFullString();
        Validator validator = validators.get(key);
        // 不存在，创建 Validator 对象，并缓存
        if (validator == null) {
            validators.put(key, createValidator(url));
            validator = validators.get(key);
        }
        return validator;
    }

    /**
     * 创建 Validator 对象
     *
     * @param url URL
     * @return Validator 对象
     */
    protected abstract Validator createValidator(URL url);

}

3.3 @MethodValidated

com.alibaba.dubbo.validation.@MethodValidated ，方法分组验证注解。代码如下：

@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface MethodValidated {

    /**
     * @return 分组集合
     */
    Class<?>[] value() default {};

}

• 使用场景：当调用某个方法时，需要检查多个分组，可以在接口方法上加上该注解。
• 用法：

@MethodValidated({Save.class, Update.class})
void relatedQuery(ValidationParameter parameter);

• 在接口方法上增加注解，表示 #relatedQuery(ValidationParameter) 这个方法，需要同时检查 Save 和 Update 这两个分组。
• 如果胖友对 Java Bean Validation 分组不熟悉，可以理解起来比较绕。可以跑下官方提供的 com.alibaba.dubbo.config.validation 的立足。

4. JSR303 实现

基于 JSR303 Bean Validation 实现的，用户只需标识 JSR303 标准的验证 annotation 。

4.1 JValidator

com.alibaba.dubbo.validation.support.jvalidation.JValidator ，实现 Validator 接口，基于 JSR303 的验证器实现类。

4.1.1 构造方法

/**
 * 服务接口类
 */
private final Class<?> clazz;
/**
 * Validator 对象
 */
private final javax.validation.Validator validator;

@SuppressWarnings({"unchecked", "rawtypes"})
public JValidator(URL url) {
    // 获得服务接口类
    this.clazz = ReflectUtils.forName(url.getServiceInterface());
    // 获得 `"jvalidation"` 配置项
    String jvalidation = url.getParameter("jvalidation");
    // 获得 ValidatorFactory 对象
    ValidatorFactory factory;
    if (jvalidation != null && jvalidation.length() > 0) { // 指定实现
        factory = Validation.byProvider((Class) ReflectUtils.forName(jvalidation)).configure().buildValidatorFactory();
    } else { // 默认
        factory = Validation.buildDefaultValidatorFactory();
    }
    // 获得 javax Validator 对象
    this.validator = factory.getValidator();
}

• "jvalidation" 指定配置项，可具体的 JSR303 的实现类。
• 如果我们未配置，并且引入 Hibernate Validator ，则使用的是 HibernateValidatorFactory 。

4.1.2 validate

@Override
public void validate(String methodName, Class<?>[] parameterTypes, Object[] arguments) throws Exception {
    // 验证分组集合
    List<Class<?>> groups = new ArrayList<Class<?>>();
    // 【第一种】添加以方法命名的内部接口，作为验证分组。例如 `ValidationService#save(...)` 方法，对应 `ValidationService.Save` 接口。
    String methodClassName = clazz.getName() + "$" + toUpperMethodName(methodName);
    Class<?> methodClass;
    try {
        methodClass = Class.forName(methodClassName, false, Thread.currentThread().getContextClassLoader());
        groups.add(methodClass);
    } catch (ClassNotFoundException e) {
    }
    // 【第二种】添加方法的 @MethodValidated 注解的值对应的类，作为验证分组。
    Method method = clazz.getMethod(methodName, parameterTypes);
    Class<?>[] methodClasses;
    if (method.isAnnotationPresent(MethodValidated.class)){
        methodClasses = method.getAnnotation(MethodValidated.class).value();
        groups.addAll(Arrays.asList(methodClasses));
    }
    // 【第三种】添加 Default.class 类，作为验证分组。在 JSR 303 中，未设置分组的验证注解，使用 Default.class 。
    // add into default group
    groups.add(0, Default.class);
    // 【第四种】添加服务接口类，作为验证分组。
    groups.add(1, clazz);
    // convert list to array
    Class<?>[] classGroups = groups.toArray(new Class[0]);

    // 验证错误集合
    Set<ConstraintViolation<?>> violations = new HashSet<ConstraintViolation<?>>();
    // 【第一步】获得方法参数的 Bean 对象。因为，JSR 303 是 Java Bean Validation ，以 Bean 为维度。
    Object parameterBean = getMethodParameterBean(clazz, method, arguments);
    // 【第一步】验证 Bean 对象。
    if (parameterBean != null) {
        violations.addAll(validator.validate(parameterBean, classGroups));
    }
    // 【第二步】验证集合参数
    for (Object arg : arguments) {
        validate(violations, arg, classGroups);
    }
    // 若有错误，抛出 ConstraintViolationException 异常。
    if (!violations.isEmpty()) {
        logger.error("Failed to validate service: " + clazz.getName() + ", method: " + methodName + ", cause: " + violations);
        throw new ConstraintViolationException("Failed to validate service: " + clazz.getName() + ", method: " + methodName + ", cause: " + violations, violations);
    }
}

• ============ 【第一步】获得验证分组集合 ============
• 第 4 行：验证分组集合 group ，目前有四种来源。
• 【第一种】第 5 至 12 行：添加以方法命名的内部接口，作为验证分组。例如 ValidationService#save(…) 方法，对应 ValidationService.Save 接口。
• 【第二种】第 13 至 19 行：添加方法的 @MethodValidated 注解的值对应的类，作为验证分组。
• 【第三种】第 22 行：添加 Default.class 类，作为验证分组。在 JSR 303 中，未设置分组的验证注解，使用 Default.class 。
• 【第四种】第 24 行：添加服务接口类 clazz ，作为验证分组。
• 最终生成的验证分组集合的顺序为：【第三种】》【第四种】》【第一种】》【第二种】。
• ============ 【第二步】验证方法参数 ============
• 第 29 行：验证错误集合 violations 。
• 【第一步】调用 「4.1.3 getMethodParameterBean」 #getMethodParameterBean(Class<?> clazz, Method method, Object[] args) 方法，获得方法参数的 Bean 对象。因为，JSR 303 是 Java Bean Validation ，以 Bean 为维度。具体的实现，我们在中，详细解析。
• 【第一步】调用 Bean Validator#validate(T object, Class<?>… groups) 方法，验证对象。
• 【第二步】循环方法参数，调用 集合为什么会有这一步不会校验集合中的每个元素 #validate(violations, arg, classGroups) 方法，验证参数。？因为，在【第一步】中，校验的是 Constraint 注解的参数( 例如 @NotNull ) ，但是呢，若是集合参数，。我们来举个例子：

void saves(@NotNull(message = "至少需要保存一个用户") User[] users);

• 【第一步】校验 users 参数的 @NotNull 约束。

• 【第二步】校验 users 参数中的每个 User 的约束。

• 第 40 至 44 行：若有验证错误，抛出 ConstraintViolationException 异常。

#validate(Set<ConstraintViolation<?» violations, Object arg, Class<?>… groups) 方法，代码如下：

/**
 * 验证集合参数
 *
 * @param violations 验证错误集合
 * @param arg 参数
 * @param groups 验证分组集合
 */
private void validate(Set<ConstraintViolation<?>> violations, Object arg, Class<?>... groups) {
    if (arg != null && !isPrimitives(arg.getClass())) {
        // [] 数组
        if (Object[].class.isInstance(arg)) {
            for (Object item : (Object[]) arg) {
                validate(violations, item, groups); // 单个元素
            }
        // Collection
        } else if (Collection.class.isInstance(arg)) {
            for (Object item : (Collection<?>) arg) {
                validate(violations, item, groups); // 单个元素
            }
        // Map
        } else if (Map.class.isInstance(arg)) {
            for (Map.Entry<?, ?> entry : ((Map<?, ?>) arg).entrySet()) {
                validate(violations, entry.getKey(), groups); // 单个元素
                validate(violations, entry.getValue(), groups); // 单个元素
            }
        // 单个元素
        } else {
            violations.addAll(validator.validate(arg, groups));
        }
    }
}

• 第 9 行：调用 基本类型第一步 #isPrimitives(Class<?> cls) 方法，判断是否为。若是基本类型，已经被【】给验证了，避免重复验证。代码如下：

private static boolean isPrimitives(Class<?> cls) {
    // [] 数组，使用内部的类来判断
    if (cls.isArray()) {
        return isPrimitive(cls.getComponentType());
    }
    // 直接判断
    return isPrimitive(cls);
}

private static boolean isPrimitive(Class<?> cls) {
    return cls.isPrimitive() || cls == String.class || cls == Boolean.class || cls == Character.class
            || Number.class.isAssignableFrom(cls) || Date.class.isAssignableFrom(cls);
}

• 第 10 至 14 行：验证数组参数，循环调用【第 26 至 29】的验证。
• 第 15 至 19 行：验证 Collection 参数，循环调用【第 26 至 29】的验证。
• 第 20 至 25 行：验证 Map 参数，循环调用【第 26 至 29】的验证。
• 第 26 至 29 行单个：验证参数。

4.1.3 getMethodParameterBean

在看 #getMethodParameterBean(Class<?> clazz, Method method, Object[] args) 的具体实现代码之前，我们来看下它，根据方法，自动生成 Bean 类的例子。

• 接口方法，代码如下：

void demo(@NotNull(message = "名字不能为空") @Min(value = 6, message = "昵称不能太短") String name,
          String password, // 不校验
          @NotNull(message = "至少需要保存一个用户") User user);

• 生成 Bean 类，代码如下：

package com.alibaba.dubbo.demo.DemoService_DemoParameter_java.lang.String_java.lang.String_com.alibaba.dubbo.demo.entity;

public class User {

    @NotNull(message = "名字不能为空")
    @Min(value = 6, message = "昵称不能太短")
    public java.lang.String name;

    public java.lang.String password;

    @NotNull(message = "至少需要保存一个用户")
    public com.alibaba.dubbo.demo.entity.User user;

    public User() {}

}

• Javassist 生成的类，使用 JD-GUI 反编译一直报错。所以，该类是笔者，手工生成的。哈哈哈，意思能达到就好。

#getMethodParameterBean(Class<?> clazz, Method method, Object[] args) 方法，代码如下：

/**
 * 使用方法参数，创建 Bean 对象。
 *
 * 因为该 Bean 对象，实际不存在对应类，使用 Javassist 动态编译生成。
 *
 * @param clazz 服务接口类
 * @param method 方法
 * @param args 参数数组
 * @return Bean 对象
 */
private static Object getMethodParameterBean(Class<?> clazz, Method method, Object[] args) {
    // 无 Constraint 注解的方法参数，无需创建 Bean 对象。
    if (!hasConstraintParameter(method)) {
        return null;
    }
    try {
        // 获得 Bean 类名
        String parameterClassName = generateMethodParameterClassName(clazz, method);
        Class<?> parameterClass;
        try {
            // 获得 Bean 类
            parameterClass = Class.forName(parameterClassName, true, clazz.getClassLoader());
        } catch (ClassNotFoundException e) { // 类不存在，使用 Javassist 动态编译生成
            // 创建 ClassPool 对象
            ClassPool pool = ClassGenerator.getClassPool(clazz.getClassLoader());
            // 创建 CtClass 对象
            CtClass ctClass = pool.makeClass(parameterClassName);
            // 设置 Java 版本为 5
            ClassFile classFile = ctClass.getClassFile();
            classFile.setVersionToJava5();
            // 添加默认构造方法
            ctClass.addConstructor(CtNewConstructor.defaultConstructor(pool.getCtClass(parameterClassName)));
            // 循环每个方法参数，生成对应的类的属性
            // parameter fields
            Class<?>[] parameterTypes = method.getParameterTypes();
            Annotation[][] parameterAnnotations = method.getParameterAnnotations();
            for (int i = 0; i < parameterTypes.length; i++) {
                Class<?> type = parameterTypes[i];
                Annotation[] annotations = parameterAnnotations[i];
                // 创建注解属性
                AnnotationsAttribute attribute = new AnnotationsAttribute(classFile.getConstPool(), AnnotationsAttribute.visibleTag);
                // 循环每个方法参数的每个注解
                for (Annotation annotation : annotations) {
                    if (annotation.annotationType().isAnnotationPresent(Constraint.class)) { // 约束条件的注解，例如 @NotNull
                        javassist.bytecode.annotation.Annotation ja = new javassist.bytecode.annotation.Annotation(
                                classFile.getConstPool(), pool.getCtClass(annotation.annotationType().getName()));
                        // 循环注解的每个方法
                        Method[] members = annotation.annotationType().getMethods();
                        for (Method member : members) {
                            if (Modifier.isPublic(member.getModifiers())
                                    && member.getParameterTypes().length == 0
                                    && member.getDeclaringClass() == annotation.annotationType()) {
                                // 将注解，添加到类的属性上
                                Object value = member.invoke(annotation);
                                if (null != value) {
                                    MemberValue memberValue = createMemberValue(
                                            classFile.getConstPool(), pool.get(member.getReturnType().getName()), value);
                                    ja.addMemberValue(member.getName(), memberValue);
                                }
                            }
                        }
                        attribute.addAnnotation(ja);
                    }
                }
                // 创建属性
                String fieldName = method.getName() + "Argument" + i;
                CtField ctField = CtField.make("public " + type.getCanonicalName() + " " + fieldName + ";", pool.getCtClass(parameterClassName));
                ctField.getFieldInfo().addAttribute(attribute);
                // 添加属性
                ctClass.addField(ctField);
            }
            // 生成类
            parameterClass = ctClass.toClass(clazz.getClassLoader(), null);
        }
        // 创建 Bean 对象
        Object parameterBean = parameterClass.newInstance();
        // 设置 Bean 对象的每个属性的值
        for (int i = 0; i < args.length; i++) {
            Field field = parameterClass.getField(method.getName() + "Argument" + i);
            field.set(parameterBean, args[i]);
        }
        return parameterBean;
    } catch (Throwable e) {
        logger.warn(e.getMessage(), e);
        return null;
    }
}

• 第 13 至 15 行：调用 Constraint #hasConstraintParameter(method) 方法，判断是否有注解( 例如，@NotNull )的方法参数。若没有，则无需创建 Bean 对象。代码如下：

private static boolean hasConstraintParameter(Method method) {
    Annotation[][] parameterAnnotations = method.getParameterAnnotations();
    // 循环所有方法参数的注解
    if (parameterAnnotations != null && parameterAnnotations.length > 0) {
        // 循环每个方法参数的注解数组
        for (Annotation[] annotations : parameterAnnotations) {
            // 是否有 Constraint 注解
            for (Annotation annotation : annotations) {
                if (annotation.annotationType().isAnnotationPresent(Constraint.class)) {
                    return true;
                }
            }
        }
    }
    return false;
}

• 第 17 至 22 行：获得 Bean 类。
• 第 23 至 73 行：若 Bean 类不存在每个属性的值，使用 Javassist 动态编译生成。 代码比较简单，已经添加详细注释，胖友耐心看看哈。其中，#createMemberValue(ConstPool cp, CtClass type, Object value) 方法，获得注解，代码如下：

// Copy from javassist.bytecode.annotation.Annotation.createMemberValue(ConstPool, CtClass);
private static MemberValue createMemberValue(ConstPool cp, CtClass type, Object value) throws NotFoundException {
    MemberValue memberValue = javassist.bytecode.annotation.Annotation.createMemberValue(cp, type);
    if (memberValue instanceof BooleanMemberValue) // Boolean
        ((BooleanMemberValue) memberValue).setValue((Boolean) value);
    else if (memberValue instanceof ByteMemberValue) // Byte
        ((ByteMemberValue) memberValue).setValue((Byte) value);
    else if (memberValue instanceof CharMemberValue) // Char
        ((CharMemberValue) memberValue).setValue((Character) value);
    else if (memberValue instanceof ShortMemberValue) // Short
        ((ShortMemberValue) memberValue).setValue((Short) value);
    else if (memberValue instanceof IntegerMemberValue) // Integer
        ((IntegerMemberValue) memberValue).setValue((Integer) value);
    else if (memberValue instanceof LongMemberValue) // Long
        ((LongMemberValue) memberValue).setValue((Long) value);
    else if (memberValue instanceof FloatMemberValue) // Float
        ((FloatMemberValue) memberValue).setValue((Float) value);
    else if (memberValue instanceof DoubleMemberValue)
        ((DoubleMemberValue) memberValue).setValue((Double) value);
    else if (memberValue instanceof ClassMemberValue) // Class
        ((ClassMemberValue) memberValue).setValue(((Class<?>) value).getName());
    else if (memberValue instanceof StringMemberValue) // String
        ((StringMemberValue) memberValue).setValue((String) value);
    else if (memberValue instanceof EnumMemberValue) // Enum
        ((EnumMemberValue) memberValue).setValue(((Enum<?>) value).name());
    /* else if (memberValue instanceof AnnotationMemberValue) */
    else if (memberValue instanceof ArrayMemberValue) { // 数组
        CtClass arrayType = type.getComponentType();
        int len = Array.getLength(value);
        // 循环，递归
        MemberValue[] members = new MemberValue[len];
        for (int i = 0; i < len; i++) {
            members[i] = createMemberValue(cp, arrayType, Array.get(value, i));
        }
        ((ArrayMemberValue) memberValue).setValue(members);
    }
    return memberValue;
}

• 第 75 至 81 行：创建 Bean 对象，设置 Bean 对象的每个属性的值。

又是一处，使用 Javassist 动态编译类的代码。好用！！！

4.2 JValidation

com.alibaba.dubbo.validation.support.jvalidation.JValidation ，实现 AbstractValidation 抽象类，代码如下：

public class JValidation extends AbstractValidation {

    @Override
    protected Validator createValidator(URL url) {
        return new JValidator(url);
    }

}

666. 彩蛋

美滋滋，终于弄懂，为什么 JSR303 是 Java Bean Validation ，结果接口方法上，每个参数都添加 Constraint 的注解，结果也可以做校验。

推荐两篇文章：

• 《JSR 303 - Bean Validation 介绍及最佳实践》
• 《Spring4新特性——集成Bean Validation 1.1(JSR-349)到SpringMVC》