使用 Spring\SpringBoot 中的组件进行代理(Aop)

296 阅读1分钟
package com.jay.prj.mybaits;

import org.aopalliance.intercept.MethodInterceptor;
import org.aopalliance.intercept.MethodInvocation;
import org.springframework.aop.support.DefaultBeanFactoryPointcutAdvisor;
import org.springframework.aop.support.StaticMethodMatcherPointcut;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.EnableAspectJAutoProxy;
import org.springframework.context.annotation.Role;

import javax.sql.DataSource;
import java.lang.reflect.Method;
import java.sql.SQLException;

@SpringBootApplication
@EnableAspectJAutoProxy(proxyTargetClass = true)
public class Application {

    public static void main(String[] args) {
        ConfigurableApplicationContext run = SpringApplication.run(Application.class, args);
        DataSource bean = (DataSource) run.getBean("dataSource2");
        try {
            System.out.println(bean.getConnection());
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }

    @Bean(name = "dataSource2")
    @Role(BeanDefinition.ROLE_SUPPORT)
    public DataSource dataSource() {
        return new PooledDataSource();
    }

    
    /**
     * 代理条件
     */
    @Bean
    @Role(BeanDefinition.ROLE_INFRASTRUCTURE)
    public DefaultBeanFactoryPointcutAdvisor transactionAdvisor() {
        DefaultBeanFactoryPointcutAdvisor advisor = new DefaultBeanFactoryPointcutAdvisor();
        advisor.setPointcut(staticMethodMatcherPointcut());
        advisor.setAdvice(transactionInterceptor());
        return advisor;
    }

    /**
     * 代理条件判断
     * @return
     */
    @Bean
    @Role(BeanDefinition.ROLE_INFRASTRUCTURE)
    public StaticMethodMatcherPointcut staticMethodMatcherPointcut() {
        return new StaticMethodMatcherPointcut() {
            public boolean matches(Method method, Class<?> aClass) {
                return DataSource.class.isAssignableFrom(aClass) && method.getName().equals("getConnection");
            }
        };
    }

    /**
     * 函数代理逻辑
     */
    public MethodInterceptor transactionInterceptor() {
        return new MethodInterceptor() {
            public Object invoke(MethodInvocation methodInvocation) throws Throwable {
                System.out.println("方法运行前");
                return null;
            }
        };
    }


}

执行结果:

2021-06-11 11:11:14.820  INFO 10776 --- [           main] s.b.c.e.t.TomcatEmbeddedServletContainer : Tomcat started on port(s): 2998 (http)
2021-06-11 11:11:14.826  INFO 10776 --- [           main] com.jay.prj.mybaits.Application          : Started Application in 4.8 seconds (JVM running for 5.775)
org.apache.ibatis.datasource.pooled.PooledDataSource@1690e14
方法运行前
null