validation - spring-boot-aop : before, after, around

This commit is contained in:
kim
2021-02-03 00:20:17 +09:00
parent 1c2c5c2b2a
commit 94a5456c34
2 changed files with 70 additions and 14 deletions

View File

@@ -0,0 +1,69 @@
package com.example.aop.config;
import com.example.aop.domain.CommonDto;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.springframework.http.HttpStatus;
import org.springframework.stereotype.Component;
import org.springframework.validation.BindingResult;
import org.springframework.validation.FieldError;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import javax.servlet.http.HttpServletRequest;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
// @Configuration -> @Controller , @Service, @Repository -> @Component
@Component
@Aspect
public class BindingAdvice {
@Before("execution(* com.example.aop.web..*Controller.*(..))")
public void before() {
HttpServletRequest request =
((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes()).getRequest();
System.out.println("전처리 로그");
}
@After("execution(* com.example.aop.web..*Controller.*(..))")
public void after() {
System.out.println("후처리 로그");
}
@Around("execution(* com.example.aop.web..*Controller.*(..))")
public Object validCheck(ProceedingJoinPoint proceedingJoinPoint) throws Throwable {
String type = proceedingJoinPoint.getSignature().getDeclaringTypeName();
String method = proceedingJoinPoint.getSignature().getName();
System.out.println(type);
System.out.println(method);
Object[] args = proceedingJoinPoint.getArgs();
Map<String, String> errorMap = new HashMap<>();
// for (Object arg : args) {
// if (arg instanceof BindingResult) {
// BindingResult bindingResult = (BindingResult) arg;
// if (bindingResult.hasErrors()) {
// for (FieldError error : bindingResult.getFieldErrors()) {
// errorMap.put(error.getField(), error.getDefaultMessage());
// }
// }
// }
// }
Arrays.stream(args)
.filter(v -> v instanceof BindingResult)
.filter(v -> ((BindingResult) v).hasErrors())
.forEach(v -> ((BindingResult) v).getFieldErrors()
.forEach(error -> errorMap.put(error.getField(), error.getDefaultMessage())));
return errorMap.isEmpty() ? proceedingJoinPoint.proceed() : new CommonDto<>(errorMap, HttpStatus.BAD_REQUEST.value());
}
}

View File

@@ -40,13 +40,6 @@ public class UserController {
@PostMapping("/user")
public CommonDto<?> save(@RequestBody @Valid JoinReqDto dto, BindingResult bindingResult) {
System.out.println("save()");
if (bindingResult.hasErrors()) {
Map<String, String> errorMap = new HashMap<>();
bindingResult.getFieldErrors().forEach(error -> {
errorMap.put(error.getField(), error.getDefaultMessage());
});
return new CommonDto<>(errorMap, HttpStatus.BAD_REQUEST.value());
}
userRepository.save(dto);
return new CommonDto<>("ok", HttpStatus.CREATED.value());
}
@@ -63,14 +56,8 @@ public class UserController {
@PutMapping("/user/{id}")
public CommonDto<?> update(@PathVariable int id, @Valid @RequestBody UpdateReqDto dto, BindingResult bindingResult){
System.out.println("update()");
if (bindingResult.hasErrors()) {
Map<String, String> errorMap = new HashMap<>();
bindingResult.getFieldErrors().forEach(error -> {
errorMap.put(error.getField(), error.getDefaultMessage());
});
return new CommonDto<>(errorMap, HttpStatus.BAD_REQUEST.value());
}
userRepository.update(id, dto);
return new CommonDto<>(HttpStatus.OK.value());
}
}