Java动态代理实现接口方法调用统计

在软件开发过程中，对接口方法的调用情况进行统计是一项重要的监控手段。通过统计接口方法的调用次数和耗时，我们可以更好地优化系统性能和资源分配。本文将介绍如何利用Java动态代理技术实现接口方法调用的统计功能。

Java动态代理实现接口方法调用统计

动态代理简介

Java动态代理是一种强大的反射机制，允许我们在运行时创建一个代理对象，该对象可以动态地拦截对目标接口方法的调用。通过实现InvocationHandler接口，我们可以在调用目标方法前后添加自定义的逻辑。

实现步骤

1. 定义统计接口

首先，我们定义一个统计接口，用于记录方法的调用次数和耗时。

public interface MethodStats {
    void recordMethodCall(String methodName, long duration);
    Map<String, Long> getMethodCallCounts();
    Map<String, Long> getMethodDurations();
}

2. 实现统计类

接下来，实现统计接口，记录方法的调用次数和耗时。

public class MethodStatsImpl implements MethodStats {
    private final Map<String, Long> callCounts = new ConcurrentHashMap<>();
    private final Map<String, Long> durations = new ConcurrentHashMap<>();

    @Override
    public void recordMethodCall(String methodName, long duration) {
        callCounts.put(methodName, callCounts.getOrDefault(methodName, 0L) + 1);
        durations.put(methodName, durations.getOrDefault(methodName, 0L) + duration);
    }

    @Override
    public Map<String, Long> getMethodCallCounts() {
        return callCounts;
    }

    @Override
    public Map<String, Long> getMethodDurations() {
        return durations;
    }
}

3. 创建动态代理类

实现InvocationHandler接口，在调用目标方法前后添加统计逻辑。

public class StatsInvocationHandler implements InvocationHandler {
    private final Object target;
    private final MethodStats methodStats;

    public StatsInvocationHandler(Object target, MethodStats methodStats) {
        this.target = target;
        this.methodStats = methodStats;
    }

    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        long startTime = System.currentTimeMillis();
        Object result = method.invoke(target, args);
        long duration = System.currentTimeMillis() - startTime;
        methodStats.recordMethodCall(method.getName(), duration);
        return result;
    }
}

4. 使用动态代理

创建代理对象并使用它来替代原始对象。

public class ProxyDemo {
    public static void main(String[] args) {
        // 原始对象
        SomeService originalService = new SomeServiceImpl();
        // 统计对象
        MethodStats methodStats = new MethodStatsImpl();
        // 创建代理对象
        SomeService proxyService = (SomeService) Proxy.newProxyInstance(
                SomeService.class.getClassLoader(),
                new Class<?>[]{SomeService.class},
                new StatsInvocationHandler(originalService, methodStats)
        );

        // 使用代理对象
        proxyService.doSomething();
        proxyService.doSomethingElse();

        // 输出统计结果
        System.out.println("Method Call Counts: " + methodStats.getMethodCallCounts());
        System.out.println("Method Durations: " + methodStats.getMethodDurations());
    }
}