src/main/java/com/springaopopenschool1/firsttask/aspect/MethodExecutionTrackingAspect.java - master - allalex/Spring-AOP

src/main/java/com/springaopopenschool1/firsttask/aspect/MethodExecutionTrackingAspect.java

51 строка2 KB

Alexey Layko

фикс бага + рефактор

10 май 2024, 02:12

10 май 2024, 02:12e17481a

package com.springaopopenschool1.firsttask.aspect;

import com.springaopopenschool1.firsttask.service.ExecutionLogService;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

@Component
@Aspect
@Slf4j
public class MethodExecutionTrackingAspect {
    private String methodName;
    private Object[] methodArgs;

    private final ExecutionLogService executionLogServiceImpl;

    @Autowired
    public MethodExecutionTrackingAspect(ExecutionLogService executionLogServiceImpl) {
        this.executionLogServiceImpl = executionLogServiceImpl;
    }


    @Around("@annotation(com.springaopopenschool1.firsttask.annotation.TrackTime)")
    public Object trackTime(ProceedingJoinPoint joinPoint) throws Throwable {
        methodName = joinPoint.getSignature().getName();
        methodArgs = joinPoint.getArgs();
        log.info("выполнение метода {} с аргументами {}", methodName, methodArgs);
        long startTime = System.currentTimeMillis();
        Object result = joinPoint.proceed();
        long executionTime = System.currentTimeMillis() - startTime;
        log.info("Метод {} выполнился за {} мс с результатом {}", methodName, executionTime, result);
        executionLogServiceImpl.logExecutionTime(joinPoint, executionTime, false);
        return result;
    }

    @Around("@annotation(com.springaopopenschool1.firsttask.annotation.TrackAsyncTime)")
    public Object trackAsyncTime(ProceedingJoinPoint joinPoint) throws Throwable {
        methodName = joinPoint.getSignature().getName();
        methodArgs = joinPoint.getArgs();
        log.info("выполнение метода {} с аргументами {}", methodName, methodArgs);
        long startTime = System.currentTimeMillis();
        Object result = joinPoint.proceed();
        long executionTime = System.currentTimeMillis() - startTime;
        log.info("Метод {} выполнился за {} мс с результатом {}", methodName, executionTime, result);
        executionLogServiceImpl.logExecutionTime(joinPoint, executionTime, true);
        return result;
    }
}