From d20d38e22b06559d758c568769017e2acf632583 Mon Sep 17 00:00:00 2001
From: zuoxiao <470321431@qq.com>
Date: 星期三, 23 四月 2025 14:06:48 +0800
Subject: [PATCH] Merge branch 'master' of http://8.140.179.55:20000/r/pipIrr-SV
---
pipIrr-platform/pipIrr-common/src/main/java/com/dy/common/aop/SsoPowerAspect.java | 144 ++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 144 insertions(+), 0 deletions(-)
diff --git a/pipIrr-platform/pipIrr-common/src/main/java/com/dy/common/aop/SsoPowerAspect.java b/pipIrr-platform/pipIrr-common/src/main/java/com/dy/common/aop/SsoPowerAspect.java
new file mode 100644
index 0000000..0458c3f
--- /dev/null
+++ b/pipIrr-platform/pipIrr-common/src/main/java/com/dy/common/aop/SsoPowerAspect.java
@@ -0,0 +1,144 @@
+package com.dy.common.aop;
+
+import com.dy.common.contant.Constant;
+import com.dy.common.multiDataSource.DataSourceContext;
+import com.dy.common.webFilter.UserTokenContext;
+import com.dy.common.webUtil.BaseResponseUtils;
+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.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.core.annotation.Order;
+import org.springframework.http.HttpEntity;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpMethod;
+import org.springframework.http.ResponseEntity;
+import org.springframework.stereotype.Component;
+import org.springframework.web.client.RestTemplate;
+import org.springframework.web.util.UriComponentsBuilder;
+
+import java.lang.reflect.Method;
+import java.util.Objects;
+
+@Slf4j
+@Aspect
+@Order(Constant.AspectOrderSsoAutho)
+@Component
+public class SsoPowerAspect {
+
+ @Value("${pipIrr.global.dev}")
+ public String isDevStage ;//鏄惁涓哄紑鍙戦樁娈�
+
+ @Value("${pipIrr.sso.checkUrl}")
+ public String SsoCheckUrl ;
+
+ private RestTemplate restTemplate;
+
+ @Autowired
+ public void setRestTemplate(RestTemplate restTemplate){
+ this.restTemplate = restTemplate ;
+ }
+
+ @Pointcut("@annotation(com.dy.common.aop.SsoPowerAop)")
+ public void ssoPowerPointCut() {
+ }
+
+ @Around("ssoPowerPointCut()")
+ public Object execute(ProceedingJoinPoint point) throws Throwable {
+ if(isDevStage != null && !isDevStage.trim().equals("") && isDevStage.trim().equalsIgnoreCase("true")){
+ return point.proceed();
+ }else{
+ MethodSignature signature = (MethodSignature) point.getSignature();
+ Method method = signature.getMethod();
+ SsoPowerAop aop = method.getAnnotation(SsoPowerAop.class) ;
+ if (Objects.nonNull(aop)){
+ String power = aop.value() ;
+ if(power.trim().equals("")){
+ power = aop.power() ;
+ }
+ if(power.trim().equals("-1")){
+ //涓嶈繘琛岀櫥褰曞強鏉冮檺楠岃瘉
+ return point.proceed();
+ }else{
+ String[] allPower = aop.ifAllPower() ;
+ String[] anyPower = aop.ifAnyPower() ;
+ String token = UserTokenContext.get() ;
+ Object rObj = this.check(token, power, allPower, anyPower);
+ if(rObj != null){
+ if(rObj instanceof SsoVo ssoVo){
+ if(ssoVo.logined){
+ if(ssoVo.hasPower){
+ if(!StringUtils.isNullOrEmpty(ssoVo.dataSourceName)){
+ DataSourceContext.set(ssoVo.dataSourceName);
+ Object obj = point.proceed();
+ DataSourceContext.remove();
+ return obj ;
+ }else{
+ //鏃犳暟鎹簮
+ return BaseResponseUtils.buildError("鍚庣绯荤粺鍑洪敊锛屾湭寰楀埌褰撳墠鐧诲綍鐢ㄦ埛鎵�灞炴満鏋勬爣绛撅紙鏁版嵁婧愬悕锛�") ;
+ }
+ }else{
+ //鏃犳潈闄�
+ return BaseResponseUtils.buildNoPower() ;
+ }
+ }else{
+ //鏈櫥褰�
+ return BaseResponseUtils.buildToLogin() ;
+ }
+ }else{
+ return rObj ;
+ }
+ }else{
+ return BaseResponseUtils.buildError("鍚庣绯荤粺鍑洪敊锛宑heck鏂规硶杩斿洖null") ;
+ }
+ }
+ }else{
+ //宸茬粡杩涘叆娉ㄨВ澶勭悊浜嗭紝杩樺緱涓嶅埌娉ㄨВ锛岃繖绉嶆儏鍐垫槸涓嶅彲鑳界殑銆�
+ return BaseResponseUtils.buildError("鍚庣绯荤粺鍑洪敊锛孌yAop娉ㄨВ涓簄ull") ;
+ }
+ }
+ }
+
+ /**
+ * 璋冪敤SSO绯荤粺杩涜楠岃瘉
+ * @param token Header涓殑鐢ㄦ埛token
+ * @param power 涓�涓潈闄�
+ * @param allPower 澶氫釜鏉冮檺
+ * @param anyPower 澶氫釜鏉冮檺
+ * @return 杩斿洖瀵硅薄
+ */
+ private Object check(String token, String power, String[] allPower, String[] anyPower){
+ if(!StringUtils.isNullOrEmpty(SsoCheckUrl)){
+ String url = UriComponentsBuilder.fromUriString(SsoCheckUrl)
+ .queryParam("token", token)
+ .queryParam("power", power)
+ .queryParam("allPower", (Object) allPower)
+ .queryParam("anyPower", (Object) anyPower)
+ .build()
+ .toUriString();
+ HttpHeaders headers = new HttpHeaders();
+ HttpEntity<?> httpEntity = new HttpEntity<>(headers);
+ ResponseEntity<SsoVo> response = null;
+ try {
+ // 閫氳繃Get鏂瑰紡璋冪敤鎺ュ彛
+ response = restTemplate.exchange(url, HttpMethod.GET, httpEntity, SsoVo.class);
+ } catch (Exception e) {
+ e.printStackTrace();
+ return BaseResponseUtils.buildError("鍚庣绯荤粺鍑洪敊锛岃闂崟鐐圭櫥褰曠郴缁熷紓甯�");
+ }
+ if(response == null){
+ return BaseResponseUtils.buildError("鍚庣绯荤粺鍑洪敊锛岃闂崟鐐圭櫥褰曠郴缁熷紓甯�");
+ }else{
+ return response.getBody();
+ }
+ }else {
+ return BaseResponseUtils.buildError("鍚庣绯荤粺鍑洪敊锛屾湭寰楀埌SsoCheckUrl");
+ }
+ }
+
+}
--
Gitblit v1.8.0