package de.gedoplan.seminar.sbt.di.exercise.aop;

import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.AfterReturning;
import org.aspectj.lang.annotation.Aspect;
import org.springframework.stereotype.Component;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import java.util.concurrent.atomic.LongAdder;
import java.util.stream.Collectors;

@Aspect
@Component
public class MethodCounter {

    private ConcurrentMap<String, LongAdder> methodCounts = new ConcurrentHashMap<>();


    protected void increase(String methodName) {
        methodCounts.computeIfAbsent(methodName,s -> new LongAdder())
                .increment();
    }

    public Map<String,Long> getMethodCounts() {
        return methodCounts.entrySet().stream()
                .collect(Collectors.toMap(Map.Entry::getKey,e -> e.getValue().longValue()));

    }

    @AfterReturning("@annotation(Counted)")
    protected void count(JoinPoint joinPoint) {
        increase(joinPoint.getSignature().getName());
    }
}