Spring validator+AOP自動パラメータチェックを実現

2175 ワード

背景:
Spring validatorを使用してControllerのパラメータチェックを行う場合、毎回手動でControllerメソッドにBlindingResult blindingResultパラメータを加えてチェック結果を取得し、チェック結果を処理します.AOPを使用している場合は自動的にチェックできます.
 
ソリューション:
package com.example.aspect;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.stereotype.Component;
import org.springframework.validation.beanvalidation.LocalValidatorFactoryBean;
import javax.annotation.Resource;
import javax.validation.ConstraintViolation;
import java.util.Iterator;
import java.util.Set;

/**
 * FileName: ValidatorAspect
 * DESCRIPTION: Controller       
 *
 * @author: SLY
 * @create: 2019/3/25
 */
@Component
@Aspect
@Slf4j
public class ValidatorAspect {

    @Resource
    private LocalValidatorFactoryBean localValidatorFactoryBean;

    @Pointcut("@annotation(org.springframework.web.bind.annotation.RequestMapping)" +
            "||@annotation(org.springframework.web.bind.annotation.GetMapping)" +
            "||@annotation(org.springframework.web.bind.annotation.PostMapping)" +
            "||@annotation(org.springframework.web.bind.annotation.PutMapping)")
    public void paramValidatePointCut(){};

    @Before("paramValidatePointCut() && args(request,..)")
    public void validateParam(JoinPoint joinPoint, Request request) {
        Set> validErrs = localValidatorFactoryBean.validate(request);
        Iterator> iterator = validErrs.iterator();
        StringBuilder sb = new StringBuilder();
        while (iterator.hasNext()) {
            ConstraintViolation constraintViolation = iterator.next();
            String error = constraintViolation.getMessage();
            sb.append(iterator.hasNext() ? error + ";" : error);
        }
        if (!validErrs.isEmpty()) {
            throw new BussinessException(ResultEnum.ERROR.getCode(), sb.toString());
        }
    }

}