validation - spring-boot-aop : before, after, around
This commit is contained in:
69
aop/src/main/java/com/example/aop/config/BindingAdvice.java
Normal file
69
aop/src/main/java/com/example/aop/config/BindingAdvice.java
Normal 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());
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -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());
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user