| New file | 
|  |  |  | 
|---|
|  |  |  | package com.dy.common.multiDataSource; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | import com.dy.common.contant.Constant; | 
|---|
|  |  |  | import com.mysql.cj.util.StringUtils; | 
|---|
|  |  |  | import lombok.extern.slf4j.Slf4j; | 
|---|
|  |  |  | 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.springframework.core.annotation.Order; | 
|---|
|  |  |  | import org.springframework.stereotype.Component; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | import java.lang.reflect.Method; | 
|---|
|  |  |  | import java.util.Objects; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * 多数据源应用中,通过注解确定某个数据源 | 
|---|
|  |  |  | * 例如在Services层方法上注解: | 
|---|
|  |  |  | * @DataSource("test") | 
|---|
|  |  |  | * 表示应用数据源test | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | @Slf4j | 
|---|
|  |  |  | @Aspect | 
|---|
|  |  |  | @Order(Constant.AspectOrderDataSource) | 
|---|
|  |  |  | @Component | 
|---|
|  |  |  | public class DataSourceAspect { | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Pointcut("@annotation(com.dy.common.multiDataSource.DataSource)") | 
|---|
|  |  |  | public void dataSourcePointCut() { | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Around("dataSourcePointCut()") | 
|---|
|  |  |  | public Object around(ProceedingJoinPoint point) throws Throwable { | 
|---|
|  |  |  | MethodSignature signature = (MethodSignature) point.getSignature(); | 
|---|
|  |  |  | Method method = signature.getMethod(); | 
|---|
|  |  |  | DataSource dataSource = method.getAnnotation(DataSource.class); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (Objects.nonNull(dataSource) && !StringUtils.isNullOrEmpty(dataSource.value())) { | 
|---|
|  |  |  | //log.info("切换数据源为" + dataSource.value()); | 
|---|
|  |  |  | //强制转成方法上配置的数据源,替换掉DataSourceContext中保存的数据源 | 
|---|
|  |  |  | DataSourceContext.set(dataSource.value()); | 
|---|
|  |  |  | }else{ | 
|---|
|  |  |  | String dataSourceName = DataSourceContext.get() ; | 
|---|
|  |  |  | if(!StringUtils.isNullOrEmpty(dataSourceName)){ | 
|---|
|  |  |  | log.info("当前数据源为" + dataSourceName); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | try { | 
|---|
|  |  |  | log.info("数据库操作开始" + dataSource.value()); | 
|---|
|  |  |  | return point.proceed(); | 
|---|
|  |  |  | } finally { | 
|---|
|  |  |  | // 销毁数据源 在执行方法之后 | 
|---|
|  |  |  | log.info("数据源操作完毕" + dataSource.value()); | 
|---|
|  |  |  | DataSourceContext.remove(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | } | 
|---|