zhaoqiaobo
11 months ago
27 changed files with 916 additions and 808 deletions
@ -0,0 +1,82 @@ |
|||||||
|
package com.logpm.basicdata.aspect; |
||||||
|
|
||||||
|
import cn.hutool.http.HttpRequest; |
||||||
|
import cn.hutool.http.HttpResponse; |
||||||
|
import com.alibaba.fastjson.JSONObject; |
||||||
|
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder; |
||||||
|
import lombok.AllArgsConstructor; |
||||||
|
import org.aspectj.lang.ProceedingJoinPoint; |
||||||
|
import org.aspectj.lang.annotation.Around; |
||||||
|
import org.aspectj.lang.annotation.Aspect; |
||||||
|
import org.springblade.common.cache.CacheNames; |
||||||
|
import org.springblade.common.constant.LauncherConstant; |
||||||
|
import org.springblade.core.redis.cache.BladeRedis; |
||||||
|
import org.springblade.core.tool.utils.ThreadLocalUtil; |
||||||
|
import org.springframework.core.env.Environment; |
||||||
|
import org.springframework.http.HttpHeaders; |
||||||
|
import org.springframework.mock.web.MockHttpServletRequest; |
||||||
|
import org.springframework.stereotype.Component; |
||||||
|
import org.springframework.web.context.request.RequestContextHolder; |
||||||
|
import org.springframework.web.context.request.ServletRequestAttributes; |
||||||
|
|
||||||
|
import java.util.HashMap; |
||||||
|
import java.util.Map; |
||||||
|
import java.util.Objects; |
||||||
|
|
||||||
|
@Aspect |
||||||
|
@Component |
||||||
|
@AllArgsConstructor |
||||||
|
public class JobAnnotationAspect { |
||||||
|
private final BladeRedis bladeRedis; |
||||||
|
private final Environment environment; |
||||||
|
|
||||||
|
@Around("@annotation(com.xxl.job.core.handler.annotation.XxlJob)") |
||||||
|
public Object xxlJobAnnotationMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
|
||||||
|
|
||||||
|
JSONObject jsonObject = mockLogin(); |
||||||
|
MockHttpServletRequest mockRequest = new MockHttpServletRequest(); |
||||||
|
mockRequest.addHeader("Blade-Auth", "bearer "+jsonObject.get("access_token")); |
||||||
|
RequestContextHolder.setRequestAttributes(new ServletRequestAttributes(mockRequest)); |
||||||
|
|
||||||
|
HttpHeaders httpHeaders = new HttpHeaders(); |
||||||
|
httpHeaders.add("Blade-Auth","bearer "+jsonObject.get("access_token") ); |
||||||
|
httpHeaders.add( "Authorization", "Basic c2FiZXI6c2FiZXJfc2VjcmV0"); |
||||||
|
ThreadLocalUtil.put("bladeContext", httpHeaders); |
||||||
|
|
||||||
|
DynamicDataSourceContextHolder.push(jsonObject.getString("tenant_id")); |
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
// 在方法执行后，从数据源上下文中移除租户ID
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
|
||||||
|
return result; |
||||||
|
|
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
private JSONObject mockLogin(){ |
||||||
|
|
||||||
|
String account ="shujutongbu"; |
||||||
|
JSONObject data =bladeRedis.get(CacheNames.LOCAL_SERVER_USER+account); |
||||||
|
if(Objects.isNull(data)){ |
||||||
|
String url = "http://"+ LauncherConstant.loginAddr(Objects.requireNonNull(environment.getActiveProfiles()[0]))+"/blade-auth/oauth/token"; |
||||||
|
HttpRequest urlRequest = HttpRequest.post(url); |
||||||
|
urlRequest.header("Authorization", "Basic c2FiZXI6c2FiZXJfc2VjcmV0"); |
||||||
|
urlRequest.header("Tenant-Id", "627683"); |
||||||
|
|
||||||
|
Map<String, Object> params = new HashMap<>(); |
||||||
|
params.put("grant_type", "local_server"); |
||||||
|
params.put("scope", "all"); |
||||||
|
params.put("username", account); |
||||||
|
params.put("tenantId", "627683"); |
||||||
|
HttpResponse execute = urlRequest.form(params).execute(); |
||||||
|
String body = execute.body(); |
||||||
|
data = JSONObject.parseObject(body); |
||||||
|
bladeRedis.setEx(CacheNames.LOCAL_SERVER_USER+account,data,3200L); |
||||||
|
} |
||||||
|
return data; |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
} |
||||||
@ -1,147 +0,0 @@ |
|||||||
package com.logpm.factorydata.jinpai.aspect; |
|
||||||
|
|
||||||
import cn.hutool.http.HttpRequest; |
|
||||||
import cn.hutool.http.HttpResponse; |
|
||||||
import com.alibaba.fastjson.JSONObject; |
|
||||||
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder; |
|
||||||
import lombok.AllArgsConstructor; |
|
||||||
import org.aspectj.lang.ProceedingJoinPoint; |
|
||||||
import org.aspectj.lang.annotation.Around; |
|
||||||
import org.aspectj.lang.annotation.Aspect; |
|
||||||
import org.aspectj.lang.reflect.MethodSignature; |
|
||||||
import org.springblade.common.annotations.LogpmAsync; |
|
||||||
import org.springblade.common.cache.CacheNames; |
|
||||||
import org.springblade.common.constant.LauncherConstant; |
|
||||||
import org.springblade.core.redis.cache.BladeRedis; |
|
||||||
import org.springblade.core.secure.utils.AuthUtil; |
|
||||||
import org.springblade.core.tool.utils.StringUtil; |
|
||||||
import org.springblade.core.tool.utils.ThreadLocalUtil; |
|
||||||
import org.springframework.core.env.Environment; |
|
||||||
import org.springframework.http.HttpHeaders; |
|
||||||
import org.springframework.mock.web.MockHttpServletRequest; |
|
||||||
import org.springframework.scheduling.annotation.Async; |
|
||||||
import org.springframework.stereotype.Component; |
|
||||||
import org.springframework.web.context.request.RequestContextHolder; |
|
||||||
import org.springframework.web.context.request.ServletRequestAttributes; |
|
||||||
|
|
||||||
import java.lang.reflect.Method; |
|
||||||
import java.util.HashMap; |
|
||||||
import java.util.Map; |
|
||||||
import java.util.Objects; |
|
||||||
|
|
||||||
@Aspect |
|
||||||
@Component |
|
||||||
@AllArgsConstructor |
|
||||||
public class AnnotationAspect { |
|
||||||
private final BladeRedis bladeRedis; |
|
||||||
private final Environment environment; |
|
||||||
|
|
||||||
|
|
||||||
/** |
|
||||||
* 定义一个切点，匹配所有带有@LogpmAsync("asyncExecutor")注解的方法。 |
|
||||||
* 注意：实际上Spring Framework自带对@LogpmAsync("asyncExecutor")的处理，直接这样配置可能会导致预期之外的行为。 |
|
||||||
*/ |
|
||||||
@Around("@annotation(org.springblade.common.annotations.LogpmAsync)") |
|
||||||
public Object asyncAroundAsyncMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
|
||||||
|
|
||||||
MethodSignature signature = (MethodSignature) joinPoint.getSignature(); |
|
||||||
Method method = signature.getMethod(); |
|
||||||
|
|
||||||
LogpmAsync myAsync = method.getAnnotation(LogpmAsync.class); |
|
||||||
String annotationValue = myAsync.value(); |
|
||||||
|
|
||||||
// 在方法执行前的操作
|
|
||||||
String tenantId = AuthUtil.getTenantId(); |
|
||||||
DynamicDataSourceContextHolder.push(tenantId); |
|
||||||
|
|
||||||
// 执行原方法
|
|
||||||
Object result = joinPoint.proceed(); |
|
||||||
|
|
||||||
// 在方法执行后的操作
|
|
||||||
DynamicDataSourceContextHolder.poll(); |
|
||||||
|
|
||||||
return result; |
|
||||||
|
|
||||||
|
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* 该方法为一个切面，用于对标记了@RabbitListener注解的方法进行增强，主要增加了日志记录和数据源上下文处理的功能。 |
|
||||||
* |
|
||||||
* @param joinPoint 切面的连接点，表示当前被拦截的方法。 |
|
||||||
* @return 返回被拦截方法的执行结果。 |
|
||||||
* @throws Throwable 如果执行过程中发生异常，则抛出。 |
|
||||||
*/ |
|
||||||
@Around("@annotation(org.springframework.amqp.rabbit.annotation.RabbitListener)") |
|
||||||
public Object logAroundAsyncMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
|
||||||
// 在方法执行前，获取当前租户ID并设置到数据源上下文中
|
|
||||||
String tenantId = AuthUtil.getTenantId(); |
|
||||||
DynamicDataSourceContextHolder.push(tenantId); |
|
||||||
// 执行原方法
|
|
||||||
Object result = joinPoint.proceed(); |
|
||||||
// 在方法执行后，从数据源上下文中移除租户ID
|
|
||||||
DynamicDataSourceContextHolder.poll(); |
|
||||||
return result; |
|
||||||
} |
|
||||||
|
|
||||||
@Around("@annotation(org.springframework.amqp.rabbit.annotation.RabbitHandler)") |
|
||||||
public Object rabbitHandlerAsyncMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
|
||||||
// 在方法执行前，获取当前租户ID并设置到数据源上下文中
|
|
||||||
String tenantId = AuthUtil.getTenantId(); |
|
||||||
DynamicDataSourceContextHolder.push(tenantId); |
|
||||||
// 执行原方法
|
|
||||||
Object result = joinPoint.proceed(); |
|
||||||
// 在方法执行后，从数据源上下文中移除租户ID
|
|
||||||
DynamicDataSourceContextHolder.poll(); |
|
||||||
return result; |
|
||||||
} |
|
||||||
|
|
||||||
@Around("@annotation(com.xxl.job.core.handler.annotation.XxlJob)") |
|
||||||
public Object xxlJobAnnotationMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
|
||||||
|
|
||||||
|
|
||||||
JSONObject jsonObject = mockLogin(); |
|
||||||
MockHttpServletRequest mockRequest = new MockHttpServletRequest(); |
|
||||||
mockRequest.addHeader("Blade-Auth", "bearer " + jsonObject.get("access_token")); |
|
||||||
RequestContextHolder.setRequestAttributes(new ServletRequestAttributes(mockRequest)); |
|
||||||
|
|
||||||
HttpHeaders httpHeaders = new HttpHeaders(); |
|
||||||
httpHeaders.add("Blade-Auth", "bearer " + jsonObject.get("access_token")); |
|
||||||
httpHeaders.add("Authorization", "Basic c2FiZXI6c2FiZXJfc2VjcmV0"); |
|
||||||
ThreadLocalUtil.put("bladeContext", httpHeaders); |
|
||||||
|
|
||||||
DynamicDataSourceContextHolder.push(jsonObject.getString("tenant_id")); |
|
||||||
// 执行原方法
|
|
||||||
Object result = joinPoint.proceed(); |
|
||||||
// 在方法执行后，从数据源上下文中移除租户ID
|
|
||||||
DynamicDataSourceContextHolder.poll(); |
|
||||||
|
|
||||||
return result; |
|
||||||
|
|
||||||
|
|
||||||
} |
|
||||||
|
|
||||||
private JSONObject mockLogin() { |
|
||||||
|
|
||||||
String account = "shujutongbu"; |
|
||||||
JSONObject data = bladeRedis.get(CacheNames.LOCAL_SERVER_USER + account); |
|
||||||
if (Objects.isNull(data)) { |
|
||||||
String url = "http://" + LauncherConstant.loginAddr(Objects.requireNonNull(environment.getActiveProfiles()[0])) + "/blade-auth/oauth/token"; |
|
||||||
HttpRequest urlRequest = HttpRequest.post(url); |
|
||||||
urlRequest.header("Authorization", "Basic c2FiZXI6c2FiZXJfc2VjcmV0"); |
|
||||||
urlRequest.header("Tenant-Id", "627683"); |
|
||||||
|
|
||||||
Map<String, Object> params = new HashMap<>(); |
|
||||||
params.put("grant_type", "local_server"); |
|
||||||
params.put("scope", "all"); |
|
||||||
params.put("username", account); |
|
||||||
params.put("tenantId", "627683"); |
|
||||||
HttpResponse execute = urlRequest.form(params).execute(); |
|
||||||
String body = execute.body(); |
|
||||||
data = JSONObject.parseObject(body); |
|
||||||
bladeRedis.setEx(CacheNames.LOCAL_SERVER_USER + account, data, 3200L); |
|
||||||
} |
|
||||||
return data; |
|
||||||
} |
|
||||||
|
|
||||||
} |
|
||||||
@ -0,0 +1,51 @@ |
|||||||
|
package com.logpm.factorydata.jinpai.aspect; |
||||||
|
|
||||||
|
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder; |
||||||
|
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.reflect.MethodSignature; |
||||||
|
import org.springblade.common.annotations.LogpmAsync; |
||||||
|
import org.springblade.core.secure.utils.AuthUtil; |
||||||
|
import org.springframework.core.annotation.Order; |
||||||
|
import org.springframework.stereotype.Component; |
||||||
|
|
||||||
|
import java.lang.reflect.Method; |
||||||
|
|
||||||
|
@Aspect |
||||||
|
@Component |
||||||
|
@Slf4j |
||||||
|
@Order(-1) |
||||||
|
public class AsyncAnnotationAspect { |
||||||
|
|
||||||
|
/** |
||||||
|
* 定义一个切点，匹配所有带有@LogpmAsync("asyncExecutor")注解的方法。 |
||||||
|
* 注意：实际上Spring Framework自带对@LogpmAsync("asyncExecutor")的处理，直接这样配置可能会导致预期之外的行为。 |
||||||
|
*/ |
||||||
|
@Around("@annotation(org.springblade.common.annotations.LogpmAsync)") |
||||||
|
public Object logAroundAsyncMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
|
||||||
|
MethodSignature signature = (MethodSignature) joinPoint.getSignature(); |
||||||
|
Method method = signature.getMethod(); |
||||||
|
|
||||||
|
LogpmAsync myAsync = method.getAnnotation(LogpmAsync.class); |
||||||
|
String annotationValue = myAsync.value(); |
||||||
|
log.info(">>>>>>>>>>>>>>>>>> annotationValue={}",annotationValue); |
||||||
|
|
||||||
|
log.info("@LogpmAsync注解的值为asyncExecutor，进行数据源切换 "); |
||||||
|
|
||||||
|
// 在方法执行前的操作
|
||||||
|
String tenantId = AuthUtil.getTenantId(); |
||||||
|
log.info(">> tenandId {} ",tenantId); |
||||||
|
DynamicDataSourceContextHolder.push(tenantId); |
||||||
|
|
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
|
||||||
|
// 在方法执行后的操作
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,82 @@ |
|||||||
|
package com.logpm.factorydata.jinpai.aspect; |
||||||
|
|
||||||
|
import cn.hutool.http.HttpRequest; |
||||||
|
import cn.hutool.http.HttpResponse; |
||||||
|
import com.alibaba.fastjson.JSONObject; |
||||||
|
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder; |
||||||
|
import lombok.AllArgsConstructor; |
||||||
|
import org.aspectj.lang.ProceedingJoinPoint; |
||||||
|
import org.aspectj.lang.annotation.Around; |
||||||
|
import org.aspectj.lang.annotation.Aspect; |
||||||
|
import org.springblade.common.cache.CacheNames; |
||||||
|
import org.springblade.common.constant.LauncherConstant; |
||||||
|
import org.springblade.core.redis.cache.BladeRedis; |
||||||
|
import org.springblade.core.tool.utils.ThreadLocalUtil; |
||||||
|
import org.springframework.core.env.Environment; |
||||||
|
import org.springframework.http.HttpHeaders; |
||||||
|
import org.springframework.mock.web.MockHttpServletRequest; |
||||||
|
import org.springframework.stereotype.Component; |
||||||
|
import org.springframework.web.context.request.RequestContextHolder; |
||||||
|
import org.springframework.web.context.request.ServletRequestAttributes; |
||||||
|
|
||||||
|
import java.util.HashMap; |
||||||
|
import java.util.Map; |
||||||
|
import java.util.Objects; |
||||||
|
|
||||||
|
@Aspect |
||||||
|
@Component |
||||||
|
@AllArgsConstructor |
||||||
|
public class JobAnnotationAspect { |
||||||
|
private final BladeRedis bladeRedis; |
||||||
|
private final Environment environment; |
||||||
|
|
||||||
|
@Around("@annotation(com.xxl.job.core.handler.annotation.XxlJob)") |
||||||
|
public Object xxlJobAnnotationMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
|
||||||
|
|
||||||
|
JSONObject jsonObject = mockLogin(); |
||||||
|
MockHttpServletRequest mockRequest = new MockHttpServletRequest(); |
||||||
|
mockRequest.addHeader("Blade-Auth", "bearer "+jsonObject.get("access_token")); |
||||||
|
RequestContextHolder.setRequestAttributes(new ServletRequestAttributes(mockRequest)); |
||||||
|
|
||||||
|
HttpHeaders httpHeaders = new HttpHeaders(); |
||||||
|
httpHeaders.add("Blade-Auth","bearer "+jsonObject.get("access_token") ); |
||||||
|
httpHeaders.add( "Authorization", "Basic c2FiZXI6c2FiZXJfc2VjcmV0"); |
||||||
|
ThreadLocalUtil.put("bladeContext", httpHeaders); |
||||||
|
|
||||||
|
DynamicDataSourceContextHolder.push(jsonObject.getString("tenant_id")); |
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
// 在方法执行后，从数据源上下文中移除租户ID
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
|
||||||
|
return result; |
||||||
|
|
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
private JSONObject mockLogin(){ |
||||||
|
|
||||||
|
String account ="shujutongbu"; |
||||||
|
JSONObject data =bladeRedis.get(CacheNames.LOCAL_SERVER_USER+account); |
||||||
|
if(Objects.isNull(data)){ |
||||||
|
String url = "http://"+ LauncherConstant.loginAddr(Objects.requireNonNull(environment.getActiveProfiles()[0]))+"/blade-auth/oauth/token"; |
||||||
|
HttpRequest urlRequest = HttpRequest.post(url); |
||||||
|
urlRequest.header("Authorization", "Basic c2FiZXI6c2FiZXJfc2VjcmV0"); |
||||||
|
urlRequest.header("Tenant-Id", "627683"); |
||||||
|
|
||||||
|
Map<String, Object> params = new HashMap<>(); |
||||||
|
params.put("grant_type", "local_server"); |
||||||
|
params.put("scope", "all"); |
||||||
|
params.put("username", account); |
||||||
|
params.put("tenantId", "627683"); |
||||||
|
HttpResponse execute = urlRequest.form(params).execute(); |
||||||
|
String body = execute.body(); |
||||||
|
data = JSONObject.parseObject(body); |
||||||
|
bladeRedis.setEx(CacheNames.LOCAL_SERVER_USER+account,data,3200L); |
||||||
|
} |
||||||
|
return data; |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,48 @@ |
|||||||
|
package com.logpm.factorydata.jinpai.aspect; |
||||||
|
|
||||||
|
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder; |
||||||
|
import lombok.extern.slf4j.Slf4j; |
||||||
|
import org.aspectj.lang.ProceedingJoinPoint; |
||||||
|
import org.aspectj.lang.annotation.Around; |
||||||
|
import org.aspectj.lang.annotation.Aspect; |
||||||
|
import org.springblade.core.secure.utils.AuthUtil; |
||||||
|
import org.springframework.stereotype.Component; |
||||||
|
|
||||||
|
@Aspect |
||||||
|
@Component |
||||||
|
@Slf4j |
||||||
|
public class RabbitAnnotationAspect { |
||||||
|
|
||||||
|
|
||||||
|
/** |
||||||
|
* 该方法为一个切面，用于对标记了@RabbitListener注解的方法进行增强，主要增加了日志记录和数据源上下文处理的功能。 |
||||||
|
* |
||||||
|
* @param joinPoint 切面的连接点，表示当前被拦截的方法。 |
||||||
|
* @return 返回被拦截方法的执行结果。 |
||||||
|
* @throws Throwable 如果执行过程中发生异常，则抛出。 |
||||||
|
*/ |
||||||
|
@Around("@annotation(org.springframework.amqp.rabbit.annotation.RabbitListener)") |
||||||
|
public Object rabbitMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
// 在方法执行前，获取当前租户ID并设置到数据源上下文中
|
||||||
|
String tenantId = AuthUtil.getTenantId(); |
||||||
|
DynamicDataSourceContextHolder.push(tenantId); |
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
// 在方法执行后，从数据源上下文中移除租户ID
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
@Around("@annotation(org.springframework.amqp.rabbit.annotation.RabbitHandler)") |
||||||
|
public Object rabbitHandlerMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
// 在方法执行前，获取当前租户ID并设置到数据源上下文中
|
||||||
|
String tenantId = AuthUtil.getTenantId(); |
||||||
|
DynamicDataSourceContextHolder.push(tenantId); |
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
// 在方法执行后，从数据源上下文中移除租户ID
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,51 @@ |
|||||||
|
package com.logpm.factorydata.olo.aspect; |
||||||
|
|
||||||
|
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder; |
||||||
|
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.reflect.MethodSignature; |
||||||
|
import org.springblade.common.annotations.LogpmAsync; |
||||||
|
import org.springblade.core.secure.utils.AuthUtil; |
||||||
|
import org.springframework.core.annotation.Order; |
||||||
|
import org.springframework.stereotype.Component; |
||||||
|
|
||||||
|
import java.lang.reflect.Method; |
||||||
|
|
||||||
|
@Aspect |
||||||
|
@Component |
||||||
|
@Slf4j |
||||||
|
@Order(-1) |
||||||
|
public class AsyncAnnotationAspect { |
||||||
|
|
||||||
|
/** |
||||||
|
* 定义一个切点，匹配所有带有@LogpmAsync("asyncExecutor")注解的方法。 |
||||||
|
* 注意：实际上Spring Framework自带对@LogpmAsync("asyncExecutor")的处理，直接这样配置可能会导致预期之外的行为。 |
||||||
|
*/ |
||||||
|
@Around("@annotation(org.springblade.common.annotations.LogpmAsync)") |
||||||
|
public Object logAroundAsyncMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
|
||||||
|
MethodSignature signature = (MethodSignature) joinPoint.getSignature(); |
||||||
|
Method method = signature.getMethod(); |
||||||
|
|
||||||
|
LogpmAsync myAsync = method.getAnnotation(LogpmAsync.class); |
||||||
|
String annotationValue = myAsync.value(); |
||||||
|
log.info(">>>>>>>>>>>>>>>>>> annotationValue={}",annotationValue); |
||||||
|
|
||||||
|
log.info("@LogpmAsync注解的值为asyncExecutor，进行数据源切换 "); |
||||||
|
|
||||||
|
// 在方法执行前的操作
|
||||||
|
String tenantId = AuthUtil.getTenantId(); |
||||||
|
log.info(">> tenandId {} ",tenantId); |
||||||
|
DynamicDataSourceContextHolder.push(tenantId); |
||||||
|
|
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
|
||||||
|
// 在方法执行后的操作
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,48 @@ |
|||||||
|
package com.logpm.factorydata.olo.aspect; |
||||||
|
|
||||||
|
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder; |
||||||
|
import lombok.extern.slf4j.Slf4j; |
||||||
|
import org.aspectj.lang.ProceedingJoinPoint; |
||||||
|
import org.aspectj.lang.annotation.Around; |
||||||
|
import org.aspectj.lang.annotation.Aspect; |
||||||
|
import org.springblade.core.secure.utils.AuthUtil; |
||||||
|
import org.springframework.stereotype.Component; |
||||||
|
|
||||||
|
@Aspect |
||||||
|
@Component |
||||||
|
@Slf4j |
||||||
|
public class RabbitAnnotationAspect { |
||||||
|
|
||||||
|
|
||||||
|
/** |
||||||
|
* 该方法为一个切面，用于对标记了@RabbitListener注解的方法进行增强，主要增加了日志记录和数据源上下文处理的功能。 |
||||||
|
* |
||||||
|
* @param joinPoint 切面的连接点，表示当前被拦截的方法。 |
||||||
|
* @return 返回被拦截方法的执行结果。 |
||||||
|
* @throws Throwable 如果执行过程中发生异常，则抛出。 |
||||||
|
*/ |
||||||
|
@Around("@annotation(org.springframework.amqp.rabbit.annotation.RabbitListener)") |
||||||
|
public Object rabbitMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
// 在方法执行前，获取当前租户ID并设置到数据源上下文中
|
||||||
|
String tenantId = AuthUtil.getTenantId(); |
||||||
|
DynamicDataSourceContextHolder.push(tenantId); |
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
// 在方法执行后，从数据源上下文中移除租户ID
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
@Around("@annotation(org.springframework.amqp.rabbit.annotation.RabbitHandler)") |
||||||
|
public Object rabbitHandlerMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
// 在方法执行前，获取当前租户ID并设置到数据源上下文中
|
||||||
|
String tenantId = AuthUtil.getTenantId(); |
||||||
|
DynamicDataSourceContextHolder.push(tenantId); |
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
// 在方法执行后，从数据源上下文中移除租户ID
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,51 @@ |
|||||||
|
package com.logpm.factorydata.suofeiya.aspect; |
||||||
|
|
||||||
|
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder; |
||||||
|
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.reflect.MethodSignature; |
||||||
|
import org.springblade.common.annotations.LogpmAsync; |
||||||
|
import org.springblade.core.secure.utils.AuthUtil; |
||||||
|
import org.springframework.core.annotation.Order; |
||||||
|
import org.springframework.stereotype.Component; |
||||||
|
|
||||||
|
import java.lang.reflect.Method; |
||||||
|
|
||||||
|
@Aspect |
||||||
|
@Component |
||||||
|
@Slf4j |
||||||
|
@Order(-1) |
||||||
|
public class AsyncAnnotationAspect { |
||||||
|
|
||||||
|
/** |
||||||
|
* 定义一个切点，匹配所有带有@LogpmAsync("asyncExecutor")注解的方法。 |
||||||
|
* 注意：实际上Spring Framework自带对@LogpmAsync("asyncExecutor")的处理，直接这样配置可能会导致预期之外的行为。 |
||||||
|
*/ |
||||||
|
@Around("@annotation(org.springblade.common.annotations.LogpmAsync)") |
||||||
|
public Object logAroundAsyncMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
|
||||||
|
MethodSignature signature = (MethodSignature) joinPoint.getSignature(); |
||||||
|
Method method = signature.getMethod(); |
||||||
|
|
||||||
|
LogpmAsync myAsync = method.getAnnotation(LogpmAsync.class); |
||||||
|
String annotationValue = myAsync.value(); |
||||||
|
log.info(">>>>>>>>>>>>>>>>>> annotationValue={}",annotationValue); |
||||||
|
|
||||||
|
log.info("@LogpmAsync注解的值为asyncExecutor，进行数据源切换 "); |
||||||
|
|
||||||
|
// 在方法执行前的操作
|
||||||
|
String tenantId = AuthUtil.getTenantId(); |
||||||
|
log.info(">> tenandId {} ",tenantId); |
||||||
|
DynamicDataSourceContextHolder.push(tenantId); |
||||||
|
|
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
|
||||||
|
// 在方法执行后的操作
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,51 @@ |
|||||||
|
package com.logpm.factorydata.zbom.aspect; |
||||||
|
|
||||||
|
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder; |
||||||
|
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.reflect.MethodSignature; |
||||||
|
import org.springblade.common.annotations.LogpmAsync; |
||||||
|
import org.springblade.core.secure.utils.AuthUtil; |
||||||
|
import org.springframework.core.annotation.Order; |
||||||
|
import org.springframework.stereotype.Component; |
||||||
|
|
||||||
|
import java.lang.reflect.Method; |
||||||
|
|
||||||
|
@Aspect |
||||||
|
@Component |
||||||
|
@Slf4j |
||||||
|
@Order(-1) |
||||||
|
public class AsyncAnnotationAspect { |
||||||
|
|
||||||
|
/** |
||||||
|
* 定义一个切点，匹配所有带有@LogpmAsync("asyncExecutor")注解的方法。 |
||||||
|
* 注意：实际上Spring Framework自带对@LogpmAsync("asyncExecutor")的处理，直接这样配置可能会导致预期之外的行为。 |
||||||
|
*/ |
||||||
|
@Around("@annotation(org.springblade.common.annotations.LogpmAsync)") |
||||||
|
public Object logAroundAsyncMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
|
||||||
|
MethodSignature signature = (MethodSignature) joinPoint.getSignature(); |
||||||
|
Method method = signature.getMethod(); |
||||||
|
|
||||||
|
LogpmAsync myAsync = method.getAnnotation(LogpmAsync.class); |
||||||
|
String annotationValue = myAsync.value(); |
||||||
|
log.info(">>>>>>>>>>>>>>>>>> annotationValue={}",annotationValue); |
||||||
|
|
||||||
|
log.info("@LogpmAsync注解的值为asyncExecutor，进行数据源切换 "); |
||||||
|
|
||||||
|
// 在方法执行前的操作
|
||||||
|
String tenantId = AuthUtil.getTenantId(); |
||||||
|
log.info(">> tenandId {} ",tenantId); |
||||||
|
DynamicDataSourceContextHolder.push(tenantId); |
||||||
|
|
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
|
||||||
|
// 在方法执行后的操作
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,82 @@ |
|||||||
|
package com.logpm.factory.aspect; |
||||||
|
|
||||||
|
import cn.hutool.http.HttpRequest; |
||||||
|
import cn.hutool.http.HttpResponse; |
||||||
|
import com.alibaba.fastjson.JSONObject; |
||||||
|
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder; |
||||||
|
import lombok.AllArgsConstructor; |
||||||
|
import org.aspectj.lang.ProceedingJoinPoint; |
||||||
|
import org.aspectj.lang.annotation.Around; |
||||||
|
import org.aspectj.lang.annotation.Aspect; |
||||||
|
import org.springblade.common.cache.CacheNames; |
||||||
|
import org.springblade.common.constant.LauncherConstant; |
||||||
|
import org.springblade.core.redis.cache.BladeRedis; |
||||||
|
import org.springblade.core.tool.utils.ThreadLocalUtil; |
||||||
|
import org.springframework.core.env.Environment; |
||||||
|
import org.springframework.http.HttpHeaders; |
||||||
|
import org.springframework.mock.web.MockHttpServletRequest; |
||||||
|
import org.springframework.stereotype.Component; |
||||||
|
import org.springframework.web.context.request.RequestContextHolder; |
||||||
|
import org.springframework.web.context.request.ServletRequestAttributes; |
||||||
|
|
||||||
|
import java.util.HashMap; |
||||||
|
import java.util.Map; |
||||||
|
import java.util.Objects; |
||||||
|
|
||||||
|
@Aspect |
||||||
|
@Component |
||||||
|
@AllArgsConstructor |
||||||
|
public class JobAnnotationAspect { |
||||||
|
private final BladeRedis bladeRedis; |
||||||
|
private final Environment environment; |
||||||
|
|
||||||
|
@Around("@annotation(com.xxl.job.core.handler.annotation.XxlJob)") |
||||||
|
public Object xxlJobAnnotationMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
|
||||||
|
|
||||||
|
JSONObject jsonObject = mockLogin(); |
||||||
|
MockHttpServletRequest mockRequest = new MockHttpServletRequest(); |
||||||
|
mockRequest.addHeader("Blade-Auth", "bearer "+jsonObject.get("access_token")); |
||||||
|
RequestContextHolder.setRequestAttributes(new ServletRequestAttributes(mockRequest)); |
||||||
|
|
||||||
|
HttpHeaders httpHeaders = new HttpHeaders(); |
||||||
|
httpHeaders.add("Blade-Auth","bearer "+jsonObject.get("access_token") ); |
||||||
|
httpHeaders.add( "Authorization", "Basic c2FiZXI6c2FiZXJfc2VjcmV0"); |
||||||
|
ThreadLocalUtil.put("bladeContext", httpHeaders); |
||||||
|
|
||||||
|
DynamicDataSourceContextHolder.push(jsonObject.getString("tenant_id")); |
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
// 在方法执行后，从数据源上下文中移除租户ID
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
|
||||||
|
return result; |
||||||
|
|
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
private JSONObject mockLogin(){ |
||||||
|
|
||||||
|
String account ="shujutongbu"; |
||||||
|
JSONObject data =bladeRedis.get(CacheNames.LOCAL_SERVER_USER+account); |
||||||
|
if(Objects.isNull(data)){ |
||||||
|
String url = "http://"+ LauncherConstant.loginAddr(Objects.requireNonNull(environment.getActiveProfiles()[0]))+"/blade-auth/oauth/token"; |
||||||
|
HttpRequest urlRequest = HttpRequest.post(url); |
||||||
|
urlRequest.header("Authorization", "Basic c2FiZXI6c2FiZXJfc2VjcmV0"); |
||||||
|
urlRequest.header("Tenant-Id", "627683"); |
||||||
|
|
||||||
|
Map<String, Object> params = new HashMap<>(); |
||||||
|
params.put("grant_type", "local_server"); |
||||||
|
params.put("scope", "all"); |
||||||
|
params.put("username", account); |
||||||
|
params.put("tenantId", "627683"); |
||||||
|
HttpResponse execute = urlRequest.form(params).execute(); |
||||||
|
String body = execute.body(); |
||||||
|
data = JSONObject.parseObject(body); |
||||||
|
bladeRedis.setEx(CacheNames.LOCAL_SERVER_USER+account,data,3200L); |
||||||
|
} |
||||||
|
return data; |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,48 @@ |
|||||||
|
package com.logpm.factory.aspect; |
||||||
|
|
||||||
|
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder; |
||||||
|
import lombok.extern.slf4j.Slf4j; |
||||||
|
import org.aspectj.lang.ProceedingJoinPoint; |
||||||
|
import org.aspectj.lang.annotation.Around; |
||||||
|
import org.aspectj.lang.annotation.Aspect; |
||||||
|
import org.springblade.core.secure.utils.AuthUtil; |
||||||
|
import org.springframework.stereotype.Component; |
||||||
|
|
||||||
|
@Aspect |
||||||
|
@Component |
||||||
|
@Slf4j |
||||||
|
public class RabbitAnnotationAspect { |
||||||
|
|
||||||
|
|
||||||
|
/** |
||||||
|
* 该方法为一个切面，用于对标记了@RabbitListener注解的方法进行增强，主要增加了日志记录和数据源上下文处理的功能。 |
||||||
|
* |
||||||
|
* @param joinPoint 切面的连接点，表示当前被拦截的方法。 |
||||||
|
* @return 返回被拦截方法的执行结果。 |
||||||
|
* @throws Throwable 如果执行过程中发生异常，则抛出。 |
||||||
|
*/ |
||||||
|
@Around("@annotation(org.springframework.amqp.rabbit.annotation.RabbitListener)") |
||||||
|
public Object rabbitMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
// 在方法执行前，获取当前租户ID并设置到数据源上下文中
|
||||||
|
String tenantId = AuthUtil.getTenantId(); |
||||||
|
DynamicDataSourceContextHolder.push(tenantId); |
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
// 在方法执行后，从数据源上下文中移除租户ID
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
@Around("@annotation(org.springframework.amqp.rabbit.annotation.RabbitHandler)") |
||||||
|
public Object rabbitHandlerMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
// 在方法执行前，获取当前租户ID并设置到数据源上下文中
|
||||||
|
String tenantId = AuthUtil.getTenantId(); |
||||||
|
DynamicDataSourceContextHolder.push(tenantId); |
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
// 在方法执行后，从数据源上下文中移除租户ID
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,82 @@ |
|||||||
|
package com.logpm.patch.aspect; |
||||||
|
|
||||||
|
import cn.hutool.http.HttpRequest; |
||||||
|
import cn.hutool.http.HttpResponse; |
||||||
|
import com.alibaba.fastjson.JSONObject; |
||||||
|
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder; |
||||||
|
import lombok.AllArgsConstructor; |
||||||
|
import org.aspectj.lang.ProceedingJoinPoint; |
||||||
|
import org.aspectj.lang.annotation.Around; |
||||||
|
import org.aspectj.lang.annotation.Aspect; |
||||||
|
import org.springblade.common.cache.CacheNames; |
||||||
|
import org.springblade.common.constant.LauncherConstant; |
||||||
|
import org.springblade.core.redis.cache.BladeRedis; |
||||||
|
import org.springblade.core.tool.utils.ThreadLocalUtil; |
||||||
|
import org.springframework.core.env.Environment; |
||||||
|
import org.springframework.http.HttpHeaders; |
||||||
|
import org.springframework.mock.web.MockHttpServletRequest; |
||||||
|
import org.springframework.stereotype.Component; |
||||||
|
import org.springframework.web.context.request.RequestContextHolder; |
||||||
|
import org.springframework.web.context.request.ServletRequestAttributes; |
||||||
|
|
||||||
|
import java.util.HashMap; |
||||||
|
import java.util.Map; |
||||||
|
import java.util.Objects; |
||||||
|
|
||||||
|
@Aspect |
||||||
|
@Component |
||||||
|
@AllArgsConstructor |
||||||
|
public class JobAnnotationAspect { |
||||||
|
private final BladeRedis bladeRedis; |
||||||
|
private final Environment environment; |
||||||
|
|
||||||
|
@Around("@annotation(com.xxl.job.core.handler.annotation.XxlJob)") |
||||||
|
public Object xxlJobAnnotationMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
|
||||||
|
|
||||||
|
JSONObject jsonObject = mockLogin(); |
||||||
|
MockHttpServletRequest mockRequest = new MockHttpServletRequest(); |
||||||
|
mockRequest.addHeader("Blade-Auth", "bearer "+jsonObject.get("access_token")); |
||||||
|
RequestContextHolder.setRequestAttributes(new ServletRequestAttributes(mockRequest)); |
||||||
|
|
||||||
|
HttpHeaders httpHeaders = new HttpHeaders(); |
||||||
|
httpHeaders.add("Blade-Auth","bearer "+jsonObject.get("access_token") ); |
||||||
|
httpHeaders.add( "Authorization", "Basic c2FiZXI6c2FiZXJfc2VjcmV0"); |
||||||
|
ThreadLocalUtil.put("bladeContext", httpHeaders); |
||||||
|
|
||||||
|
DynamicDataSourceContextHolder.push(jsonObject.getString("tenant_id")); |
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
// 在方法执行后，从数据源上下文中移除租户ID
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
|
||||||
|
return result; |
||||||
|
|
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
private JSONObject mockLogin(){ |
||||||
|
|
||||||
|
String account ="shujutongbu"; |
||||||
|
JSONObject data =bladeRedis.get(CacheNames.LOCAL_SERVER_USER+account); |
||||||
|
if(Objects.isNull(data)){ |
||||||
|
String url = "http://"+ LauncherConstant.loginAddr(Objects.requireNonNull(environment.getActiveProfiles()[0]))+"/blade-auth/oauth/token"; |
||||||
|
HttpRequest urlRequest = HttpRequest.post(url); |
||||||
|
urlRequest.header("Authorization", "Basic c2FiZXI6c2FiZXJfc2VjcmV0"); |
||||||
|
urlRequest.header("Tenant-Id", "627683"); |
||||||
|
|
||||||
|
Map<String, Object> params = new HashMap<>(); |
||||||
|
params.put("grant_type", "local_server"); |
||||||
|
params.put("scope", "all"); |
||||||
|
params.put("username", account); |
||||||
|
params.put("tenantId", "627683"); |
||||||
|
HttpResponse execute = urlRequest.form(params).execute(); |
||||||
|
String body = execute.body(); |
||||||
|
data = JSONObject.parseObject(body); |
||||||
|
bladeRedis.setEx(CacheNames.LOCAL_SERVER_USER+account,data,3200L); |
||||||
|
} |
||||||
|
return data; |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,48 @@ |
|||||||
|
package com.logpm.patch.aspect; |
||||||
|
|
||||||
|
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder; |
||||||
|
import lombok.extern.slf4j.Slf4j; |
||||||
|
import org.aspectj.lang.ProceedingJoinPoint; |
||||||
|
import org.aspectj.lang.annotation.Around; |
||||||
|
import org.aspectj.lang.annotation.Aspect; |
||||||
|
import org.springblade.core.secure.utils.AuthUtil; |
||||||
|
import org.springframework.stereotype.Component; |
||||||
|
|
||||||
|
@Aspect |
||||||
|
@Component |
||||||
|
@Slf4j |
||||||
|
public class RabbitAnnotationAspect { |
||||||
|
|
||||||
|
|
||||||
|
/** |
||||||
|
* 该方法为一个切面，用于对标记了@RabbitListener注解的方法进行增强，主要增加了日志记录和数据源上下文处理的功能。 |
||||||
|
* |
||||||
|
* @param joinPoint 切面的连接点，表示当前被拦截的方法。 |
||||||
|
* @return 返回被拦截方法的执行结果。 |
||||||
|
* @throws Throwable 如果执行过程中发生异常，则抛出。 |
||||||
|
*/ |
||||||
|
@Around("@annotation(org.springframework.amqp.rabbit.annotation.RabbitListener)") |
||||||
|
public Object rabbitMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
// 在方法执行前，获取当前租户ID并设置到数据源上下文中
|
||||||
|
String tenantId = AuthUtil.getTenantId(); |
||||||
|
DynamicDataSourceContextHolder.push(tenantId); |
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
// 在方法执行后，从数据源上下文中移除租户ID
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
@Around("@annotation(org.springframework.amqp.rabbit.annotation.RabbitHandler)") |
||||||
|
public Object rabbitHandlerMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
// 在方法执行前，获取当前租户ID并设置到数据源上下文中
|
||||||
|
String tenantId = AuthUtil.getTenantId(); |
||||||
|
DynamicDataSourceContextHolder.push(tenantId); |
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
// 在方法执行后，从数据源上下文中移除租户ID
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,48 @@ |
|||||||
|
package com.logpm.trunkline.aspect; |
||||||
|
|
||||||
|
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder; |
||||||
|
import lombok.extern.slf4j.Slf4j; |
||||||
|
import org.aspectj.lang.ProceedingJoinPoint; |
||||||
|
import org.aspectj.lang.annotation.Around; |
||||||
|
import org.aspectj.lang.annotation.Aspect; |
||||||
|
import org.springblade.core.secure.utils.AuthUtil; |
||||||
|
import org.springframework.stereotype.Component; |
||||||
|
|
||||||
|
@Aspect |
||||||
|
@Component |
||||||
|
@Slf4j |
||||||
|
public class RabbitAnnotationAspect { |
||||||
|
|
||||||
|
|
||||||
|
/** |
||||||
|
* 该方法为一个切面，用于对标记了@RabbitListener注解的方法进行增强，主要增加了日志记录和数据源上下文处理的功能。 |
||||||
|
* |
||||||
|
* @param joinPoint 切面的连接点，表示当前被拦截的方法。 |
||||||
|
* @return 返回被拦截方法的执行结果。 |
||||||
|
* @throws Throwable 如果执行过程中发生异常，则抛出。 |
||||||
|
*/ |
||||||
|
@Around("@annotation(org.springframework.amqp.rabbit.annotation.RabbitListener)") |
||||||
|
public Object rabbitMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
// 在方法执行前，获取当前租户ID并设置到数据源上下文中
|
||||||
|
String tenantId = AuthUtil.getTenantId(); |
||||||
|
DynamicDataSourceContextHolder.push(tenantId); |
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
// 在方法执行后，从数据源上下文中移除租户ID
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
@Around("@annotation(org.springframework.amqp.rabbit.annotation.RabbitHandler)") |
||||||
|
public Object rabbitHandlerMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
// 在方法执行前，获取当前租户ID并设置到数据源上下文中
|
||||||
|
String tenantId = AuthUtil.getTenantId(); |
||||||
|
DynamicDataSourceContextHolder.push(tenantId); |
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
// 在方法执行后，从数据源上下文中移除租户ID
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,48 @@ |
|||||||
|
package com.logpm.warehouse.aspect; |
||||||
|
|
||||||
|
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder; |
||||||
|
import lombok.extern.slf4j.Slf4j; |
||||||
|
import org.aspectj.lang.ProceedingJoinPoint; |
||||||
|
import org.aspectj.lang.annotation.Around; |
||||||
|
import org.aspectj.lang.annotation.Aspect; |
||||||
|
import org.springblade.core.secure.utils.AuthUtil; |
||||||
|
import org.springframework.stereotype.Component; |
||||||
|
|
||||||
|
@Aspect |
||||||
|
@Component |
||||||
|
@Slf4j |
||||||
|
public class RabbitAnnotationAspect { |
||||||
|
|
||||||
|
|
||||||
|
/** |
||||||
|
* 该方法为一个切面，用于对标记了@RabbitListener注解的方法进行增强，主要增加了日志记录和数据源上下文处理的功能。 |
||||||
|
* |
||||||
|
* @param joinPoint 切面的连接点，表示当前被拦截的方法。 |
||||||
|
* @return 返回被拦截方法的执行结果。 |
||||||
|
* @throws Throwable 如果执行过程中发生异常，则抛出。 |
||||||
|
*/ |
||||||
|
@Around("@annotation(org.springframework.amqp.rabbit.annotation.RabbitListener)") |
||||||
|
public Object rabbitMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
// 在方法执行前，获取当前租户ID并设置到数据源上下文中
|
||||||
|
String tenantId = AuthUtil.getTenantId(); |
||||||
|
DynamicDataSourceContextHolder.push(tenantId); |
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
// 在方法执行后，从数据源上下文中移除租户ID
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
@Around("@annotation(org.springframework.amqp.rabbit.annotation.RabbitHandler)") |
||||||
|
public Object rabbitHandlerMethods(ProceedingJoinPoint joinPoint) throws Throwable { |
||||||
|
// 在方法执行前，获取当前租户ID并设置到数据源上下文中
|
||||||
|
String tenantId = AuthUtil.getTenantId(); |
||||||
|
DynamicDataSourceContextHolder.push(tenantId); |
||||||
|
// 执行原方法
|
||||||
|
Object result = joinPoint.proceed(); |
||||||
|
// 在方法执行后，从数据源上下文中移除租户ID
|
||||||
|
DynamicDataSourceContextHolder.poll(); |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
Loading…
Reference in new issue