인프런 커뮤니티 질문&답변

안녕하세요 강사님!

강사님께서 말씀해주신 AOP로 구현하면 좋은 기능을 실제 코드로 작성해봤습니다.

로그 트레이스기능에 매서드 종료가 1초 5초 10초가 될때 로그 레벨을 높여서 로그를 찍는 기능입니다.

    @Aspect
    @Slf4j
    static class LogTraceSelf {

        ThreadLocal<TraceStatus> trace = new ThreadLocal<>();

        private TraceStatus SynchronizedTrace(){
            TraceStatus traceStatus = trace.get();
            if (traceStatus == null) {
                trace.set(new TraceStatus());
                traceStatus = trace.get();
            }
            return traceStatus;
        }

        @Pointcut("@annotation(hello.aop.exam.annotation.Trace)")
        public void logTrace() {}

        @Before("logTrace()")
        public void doTrace(JoinPoint joinPoint) {
            TraceStatus traceStatus = SynchronizedTrace();
            traceStatus.begin(joinPoint.getSignature().getName());
        }

        @AfterReturning("logTrace()")
        public void doTraceSuccess(JoinPoint joinPoint) {
            TraceStatus traceStatus = SynchronizedTrace();
            long gap = traceStatus.end(joinPoint.getSignature().getName());
        }

        @AfterThrowing(value = "logTrace()", throwing = "ex")
        public void doTraceException(JoinPoint joinPoint, Exception ex) {
            TraceStatus traceStatus = SynchronizedTrace();
            traceStatus.end(joinPoint.getSignature().getName(),ex);
        }

    }

조인포인트 실행 권한이 필요없는 부가 기능이다보니

@Around 사용을 하지 않고 나누어서 처리를 하다보니 불필요한 코드가 더 증가하게되었습니다.

그리고 @Around와 다르게 어드바이스내 지역변수에 TraceStatus를 저장하고 사용할수 없기때문에

ThreadLoacal에 공유할 데이터와 본인이 가지고 있어야하는 timestamp도 저장했습니다.

@Slf4j
public class TraceStatus {

    private final String id;
    private int depth = 0;
    private Deque<Long> timeStampHolder = new ArrayDeque<>();
    private final int offsetInfoMs;
    private final int offsetWarningMs;
    private final int offsetErrorMs;

}

기능은 stack 자료구조를 활용해서 begin을 할때마다 timeStampHolder에 시작시간을 저장합니다.

end를 호출하면 pop으로 자신이 넣은 시작 시간을 꺼내서 시간을 계산하고 로그를 출력합니다.

기타 기능 메서드들은 강사님이 작성하신거와 유사해서 제외했습니다.

이제 소요된 시간이 1초, 5초, 10초가 걸릴 경우 info,warning,error로 출력하는 로그를 남기려고할때

1. 어드바이스가 조건을 가지고 추가 기능을 만든다.

2. 위 코드처럼 로그 트레이스 내에 저장하고 로그 트레이스 내부에서 출력하도록 한다.

3. 경고 알림용 트레이스를 따로 만들고 상태를 전달한다.

제 생각은 트레이스에게 넘길경우 트레이스를 인터페이스로 만들어서 필요에 따라 다른 트레이스를 사용한다고 한다면 이 기능을 계속 구현해야 된다는 단점이 생길거 같습니다.
저라면 어드바이스에서 소요시간을 트레이스에게 받아서 추가 로직을 수행하려고 할거같습니다.

강사님께서 코드를 작성하신다면

어떤 포인트컷을 사용하셔서 구현하실건지,

그리고 애노테이션에서 넘어온 값을 가지고 어드바이스에서 처리하게 하시는지

아니면 로그 트레이스 내에서 처리하게 하실건지 그 이유가 궁금합니다.