com.yammer.metrics.spring.MeteredMethodInterceptor Maven / Gradle / Ivy

Go to download
package com.yammer.metrics.spring;

import com.yammer.metrics.annotation.Metered;
import com.yammer.metrics.core.Meter;
import com.yammer.metrics.core.MetricsRegistry;
import org.aopalliance.intercept.MethodInterceptor;
import org.aopalliance.intercept.MethodInvocation;
import org.springframework.util.ReflectionUtils;
import org.springframework.util.ReflectionUtils.MethodCallback;
import org.springframework.util.ReflectionUtils.MethodFilter;

import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.Map;

public class MeteredMethodInterceptor implements MethodInterceptor, MethodCallback {

    private static final MethodFilter filter = new AnnotationMethodFilter(Metered.class);

    protected final MetricsRegistry metrics;
    protected final Class targetClass;
    protected final Map meters;

    public MeteredMethodInterceptor(final MetricsRegistry metrics, final Class targetClass) {
        this.metrics = metrics;
        this.targetClass = targetClass;
        this.meters = new HashMap();

        ReflectionUtils.doWithMethods(targetClass, this, filter);
    }

    @Override
    public Object invoke(MethodInvocation invocation) throws Throwable {
        meters.get(invocation.getMethod().getName()).mark();
        return invocation.proceed();
    }

    @Override
    public void doWith(Method method) throws IllegalArgumentException, IllegalAccessException {
        final Metered metered = method.getAnnotation(Metered.class);
        final String methodName = method.getName();
        final String meterName = metered.name().isEmpty() ? methodName : metered.name();
        final Meter meter = metrics.newMeter(targetClass,
                                             meterName,
                                             metered.eventType(),
                                             metered.rateUnit());
        meters.put(methodName, meter);
    }

}