Java代理实现方法前后切入时间记录

IRecordAspect:

/**
 * 切入方法具体内容自己实现
 */
public interface IRecordAspect {
    long begin(Method method);

    void after(Method method,long begin);
}

IFunction:

/**
 * 要被代理的方法接口
 */
public interface IFunction {
    /**简单方法*/
	void execute();
    /**带传入返回参数方法*/
	int addition(int a, int b);
}

FunctionImpl:

/**
 * 被代理执行的具体方法
 */
public class FunctionImpl implements IFunction {

	@Override
	public void execute() {
		try {
			Thread.sleep(200);
			System.out.println("执行方法");
		} catch (InterruptedException e) {
			e.printStackTrace();
		}
	}
	
	@Override
	public int addition(int a, int b) {
		return a + b;
	}

}

FunctionProxyFactory:

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;

public class FunctionProxyFactory implements InvocationHandler {
	private IFunction delegate;
	private IRecordAspect aspect;

	public static FunctionProxyFactory of(IRecordAspect aspect) {
		return new FunctionProxyFactory(aspect);
	}

	private FunctionProxyFactory(IRecordAspect aspect) {
		this.aspect = aspect;
	}

	/**
	 * 绑定委托对象并返回一个代理类
	 * 
	 * @param delegate
	 * @return
	 */
	public Object aspect(IFunction delegate) {
		this.delegate = delegate;
		/**
		 * vl:类加载器<br>
		 * v2:代理类需要实现的接口<br>
		 * v3:InvocationHandler处理类<br>
		 */
		return Proxy.newProxyInstance(delegate.getClass().getClassLoader(), delegate.getClass().getInterfaces(), this);
	}

	@Override
	public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
		/** 切入具体方法前 */
		long begin = aspect.begin(method);
		/**
		 * 执行代理方法 <br>
		 * invoke:执行方法后的返回值,当方法定义为void时为null<br>
		 * v1:接口的实现类<br>
		 * v2:定义方法的传入参数<br>
		 */
		Object invoke = method.invoke(delegate, args);
		/** 切入具体方法后 */
		aspect.after(method, begin);
		return invoke;
	}

}

Main:

public class Main implements IRecordAspect {

	@Override
	public long begin(Method method) {
		long currentTimeMillis = System.currentTimeMillis();
		System.out.println("开始执行 :" + method.getName() + "->" + currentTimeMillis + "毫秒");
		return currentTimeMillis;
	}

	@Override
	public void after(Method method, long begin) {
		long currentTimeMillis = System.currentTimeMillis();
		System.out.println("执行" + method.getName() + "结束" + "->" + (currentTimeMillis) + "毫秒");
		System.out.println("用时 : " + (currentTimeMillis - begin) + "毫秒");
	}

	@Test
	public void test() {
		IFunction iRecord = (IFunction) FunctionProxyFactory.of(this).aspect(new FunctionImpl());
		iRecord.execute();
		System.out.println(iRecord.addition(3, 2));
	}
}

执行结果:

开始执行 :execute->1508726615900毫秒
执行方法
执行execute结束->1508726616103毫秒
用时 : 203毫秒
开始执行 :addition->1508726616103毫秒
执行addition结束->1508726616103毫秒
用时 : 0毫秒
5

 

转载于:https://my.oschina.net/spoon2014/blog/1554759