Aspect to hookup execution metrics to Spring Data Repositores

SpringDataMetrics.java

@Aspect
@Component
public class SpringDataMetrics {
    private static final Logger logger = LoggerFactory.getLogger(SpringDataMetrics.class);
    private final MeterRegistry meterRegistry;
​
    public SpringDataMetrics(MeterRegistry meterRegistry) {
        this.meterRegistry = meterRegistry;
    }
​
    @Around("this(org.springframework.data.repository.Repository)")
    public Object aroundRepositoryMethod(ProceedingJoinPoint proceedingJoinPoint) throws Throwable {
        StopWatch stopWatch = new StopWatch();
​
        try {
            stopWatch.start();
            return proceedingJoinPoint.proceed();
        } finally {
            stopWatch.stop();
​
            try {
                Class<?> targetClass = AopUtils.getTargetClass(proceedingJoinPoint.getThis());
​
                Class<?> repositoryClass = Arrays.stream(targetClass.getInterfaces())
                        .filter(Repository.class::isAssignableFrom)
                        .findFirst()
                        .orElse(proceedingJoinPoint.getSignature().getDeclaringType());
​
                String type = Arrays.stream(repositoryClass.getInterfaces())
                        .filter(Repository.class::isAssignableFrom)
                        .findFirst()
                        .map(Class::getSimpleName)
                        .orElse("Unknown");
​
                String repository = repositoryClass.getSimpleName();
                String method = proceedingJoinPoint.getSignature().getName();
​
                logger.debug(append("type", type)
                                .and(append("repository", repository))
                                .and(append("method", method)),
                        "Repository method executed");
​
                Timer.builder("spring.data.methods")
                        .tag("type", type)
                        .tag("repository", repository)
                        .tag("method", method)
                        .register(meterRegistry)
                        .record(Duration.ofMillis(stopWatch.getTotalTimeMillis()));
            } catch (Exception e) {
                logger.error("Error recording spring data repository metrics", e);
            }
        }
    }
}