Spring aop Aspect获取注解在类的自定义注解里面的参数

在Spring Boot + Mybatis 动态数据源基础上,很多教程(文末附链接了)都是实现方法级别的切换,按照普遍教程都是这样的

1.声明一个自定义注解,并且带参数的

@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DataSource {
    
    /**
     * 数据源key值
     * @return
     */
    String value();
    
}

2.然后按照设置切点

@Before("@annotation(dataSource))")

3.再到实现上就是

@RestController
@RequestMapping("user")
public class SysUserController {

    @Autowired
    private SysUserService sysUserService;
    
    
    @DataSource(value="master")
    @PostMapping(value="/findAll")
    public Object findAll() {
        return sysUserService.findAll();
    }
    
    @DataSource(value="slave")
    @PostMapping(value="/findAll2")
    public Object findAll2() {
        return sysUserService.findAll();
    }

}

通过方法上的注解,来切换不同的数据源

但是我懒啊

需求就来了:::::

  • 我不想一个一个方法写注解,大同小异参数又没有什么大变化,都是DataSource的key
  • 我想只能在类上面注解,实现整个类的数据源变化
  • 我还想有的方法可能不适用,我得有优先级,如果在方法上有注解话,优先使用方法上的

开搞:如何把数据源自定义注解放在类上,然后切点去扫描类,获取切点类上面的自定义参数,来实现切换数据源

主要方向就是切点设置这边啦直接贴出来

package com.wsh.demo.asp;

import com.wsh.demo.config.dynDataSource.DynamicDataSourceContextHolder;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.core.annotation.AnnotationUtils;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;

import java.lang.reflect.Method;

/**
 * @Author: wsh
 */
@Aspect
@Order(-1)  // 该切面应当先于 @Transactional 执行
@Component
public class DynamicDataSourceAspect {
    private Logger logger = LoggerFactory.getLogger(this.getClass());

    /**
     * 切换数据源
     */
    @Around("execution(@com.wsh.demo.asp.DataSource * *(..))")
    public Object  aroundController(ProceedingJoinPoint pjp) throws Throwable {
        // 获取切入的 Method
        MethodSignature joinPointObject = (MethodSignature) pjp.getSignature();
        Method method = joinPointObject.getMethod();

        boolean flag = method.isAnnotationPresent(DataSource.class);
        if (flag) {
            DataSource dataSource = method.getAnnotation(DataSource.class);
            setDataSource(dataSource,method);
        } else {
            // 如果方法上没有注解,则搜索类上是否有注解
            DataSource classDataSource = AnnotationUtils.findAnnotation(joinPointObject.getMethod().getDeclaringClass(), DataSource.class);
            if (classDataSource != null) {
                setDataSource(classDataSource,method);
            } else {
                DynamicDataSourceContextHolder.clearDataSourceKey();
                logger.info("没有注解使用默认的");
            }
        }
        Object proceed = pjp.proceed();
        // 将数据源置为默认数据源
        DynamicDataSourceContextHolder.clearDataSourceKey();
        return proceed;
    }

    private void setDataSource(DataSource dataSource,Method method ) {
        if (!DynamicDataSourceContextHolder.containDataSourceKey(dataSource.value())) {
            logger.info("DataSource [{}] doesn't exist, use default DataSource [{}] " + dataSource.value());
        } else {
            // 切换数据源
            DynamicDataSourceContextHolder.setDataSourceKey(dataSource.value());
            logger.info("Switch DataSource to [" + DynamicDataSourceContextHolder.getDataSourceKey() +
                    "] in Method [" + method.getName() + "]");
        }
    }

}

感谢下面的文章,一个是教程,一个是优化时候,从切点上获取到灵光点 

Spring Boot + Mybatis 实现动态数据源

SpringAOP如何获得执行方法的class上的注解信息

  • 1
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
### 回答1: 可以使用Spring AOP来通过自定义注解来实现切入点。首先,定义一个自定义注解,并在需要切入的方法上使用该注解,然后定义一个切面,并在切面中使用@Before 或 @Around 注解来拦截被标记的方法,最后,在Spring的配置文件中声明所需的切面,此时,所有被标记的方法都会被拦截。 ### 回答2: 使用Spring AOP切入自定义注解有以下几个步骤: 1. 在项目中引入Spring AOP的依赖,一般为spring-aopspring-aspects。 2. 在配置文件中开启Spring AOP的自动代理功能。可以通过在XML配置文件中添加<aop:aspectj-autoproxy />或在Java配置文件中添加@EnableAspectJAutoProxy注解实现。 3. 创建一个切面,用于定义切入的逻辑。这个需要使用@Component或者其他Spring注解进行标识,以便Spring能够扫描到。 4. 在切面的方法中,使用@Before、@After等注解定义切入点和具体的切入操作。例如,使用@Before注解定义在某个注解标记的方法执行之前切入的逻辑。 5. 在注解中定义自定义的切点。可以使用@Retention和@Target等元注解来配置注解的生命周期和使用范围。 6. 在目标或方法上添加自定义的注解。例如,在一个Service的某个方法上添加自定义注解。 7. 运行项目,Spring会根据配置自动代理目标,当目标或方法被调用时,切面中定义的切入逻辑就会自动被执行。 8. 可以通过配置切入的顺序、通知的型等来进一步细化切入的逻辑。 通过以上步骤,我们可以使用Spring AOP方便地切入自定义注解,实现对目标或方法的增强、日志记录、权限控制等功能。 ### 回答3: 使用Spring AOP切入自定义注解需要以下几个步骤: 1. 定义一个自定义的注解。可以使用Java提供的`@interface`关键字创建一个注解,例如: ```java @Target(ElementType.METHOD) // 定义注解的作用范围为方法 @Retention(RetentionPolicy.RUNTIME) // 注解在运行时可见 public @interface CustomAnnotation { // 自定义注解的属性 } ``` 2. 创建一个切面来处理注解。可以使用Spring提供的`@Aspect`注解来标记切面,并在方法上使用`@Before`、`@After`等注解来定义切入点和增强逻辑。例如: ```java @Aspect @Component public class CustomAspect { @Before("@annotation(customAnnotation)") // 拦截带有CustomAnnotation注解的方法 public void beforeMethod(CustomAnnotation customAnnotation) { // 在方法执行前执行的逻辑 } } ``` 3. 配置Spring AOP。在Spring配置文件中添加AOP的配置,例如使用`<aop:aspectj-autoproxy>`标签开启自动代理,并指定切面的包名,让Spring能够自动扫描并应用切面逻辑。 4. 在目标方法上使用自定义注解。在需要切入的方法上标记使用自定义注解,例如: ```java @CustomAnnotation public void doSomething() { // 方法的实际逻辑 } ``` 这样,在调用`doSomething()`方法时,Spring AOP会拦截到带有`@CustomAnnotation`注解的方法,并执行切面逻辑。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值