没有xml配置的非Spring轻量级AOP,用于编织方法的注释

时间:2019-01-30 00:36:51

标签: java annotations aop

需要在某些注释的方法之前和之后运行。

不使用spring,没有xml。是否可以使用main()设置的某种AOP引擎,以便可以在需要时调用它?对于我来说,也可以放置一个方法来手动调用评估方法。

示例:

public void doThis(@RequiredSecurityRole("admin") user){
    doAOPStuff();
}

before()从数据库获取并检查用户是否为admin,如果不是admin,则抛出异常。

after()登录到操作数据库。

如何实现?

1 个答案:

答案 0 :(得分:1)

您可以使用java.lang.reflex.Proxy类自己完成此操作。它确实需要在接口中定义要代理的代码。

import static java.lang.annotation.RetentionPolicy.RUNTIME;

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

public class DoItYourAop {

  public static void main(String[] args) {
    SaysHello saysHello = new SaysHelloImpl();
    InvocationHandler logger = new LoggingProxy(saysHello);
    SaysHello proxy = (SaysHello) Proxy.newProxyInstance(SaysHello.class.getClassLoader(),
        new Class[]{SaysHello.class}, logger);
    proxy.sayHello();
  }

  public interface SaysHello {

    void sayHello();

    void sayGoodbye();
  }

  public static class SaysHelloImpl implements SaysHello {
    @Log
    @Override
    public void sayHello() {
      System.out.println("Says Hello");
    }

    @Override
    public void sayGoodbye() {
      System.out.println("Says Goodbye");
    }
  }

  @Retention(RUNTIME)
  @interface Log {
  }

  public static class LoggingProxy implements InvocationHandler {

    private final Object proxied;

    public LoggingProxy(Object proxied) {
      this.proxied = proxied;
    }

    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
      Method proxiedMethod = proxied.getClass().getMethod(method.getName(), method.getParameterTypes());
      boolean log = proxiedMethod.isAnnotationPresent(Log.class);
      if (log) {
        System.out.println("Before");
      }

      Object result = method.invoke(proxied, args);

      if (log) {
        System.out.println("After");
      }

      return result;
    }
  }
}