From 7ce92488d62ee9a26f2e76fdf0a5e84ab115f623 Mon Sep 17 00:00:00 2001 From: liurunyu <lry9898@163.com> Date: 星期二, 07 十一月 2023 10:52:03 +0800 Subject: [PATCH] 重写登录用户权限验证部分 --- pipIrr-platform/pipIrr-web/pipIrr-web-sso/src/main/java/com/dy/sso/busi/SsoCtrl.java | 126 ++++++++++++++++++++++++++--------------- 1 files changed, 80 insertions(+), 46 deletions(-) diff --git a/pipIrr-platform/pipIrr-web/pipIrr-web-sso/src/main/java/com/dy/sso/busi/SsoCtrl.java b/pipIrr-platform/pipIrr-web/pipIrr-web-sso/src/main/java/com/dy/sso/busi/SsoCtrl.java index 957ef19..9d20d68 100644 --- a/pipIrr-platform/pipIrr-web/pipIrr-web-sso/src/main/java/com/dy/sso/busi/SsoCtrl.java +++ b/pipIrr-platform/pipIrr-web/pipIrr-web-sso/src/main/java/com/dy/sso/busi/SsoCtrl.java @@ -199,62 +199,22 @@ BaUser userPo = this.sv.getByUuid(token) ; SsoVo vo = new SsoVo(); if(userPo != null){ + vo.dataSourceName = userPo.orgTag ; vo.logined = true ; vo.hasPower = false ;//榛樿鏄棤鏉冮檺 if(userPo.supperAdmin != null && userPo.supperAdmin == 1){ vo.hasPower = true ; }else{ if(userPo.privileges != null && userPo.privileges.size() > 0){ - if(privilege != null && !privilege.trim().equals("")){ - int intPri = Integer.parseInt(privilege) ; - for(Integer pri : userPo.privileges){ - if(pri == intPri){ - vo.hasPower = true ; - break ; - } - } - }else{ - if(allPrivilege != null && allPrivilege.length > 0){ - int intPri ; - boolean ok = false ; - boolean allOk = true ; - for(String strPri : allPrivilege){ - intPri = Integer.parseInt(strPri) ; - for(Integer pri : userPo.privileges){ - if(pri == intPri){ - ok = true ; - break ; - } - } - if(!ok){ - allOk = false ; - break ; - } - } - if(allOk){ - vo.hasPower = true ; - } - }else{ - int intPri ; - if(anyPrivilege != null && anyPrivilege.length > 0){ - for(String strPri : anyPrivilege){ - intPri = Integer.parseInt(strPri) ; - for(Integer pri : userPo.privileges){ - if(pri == intPri){ - vo.hasPower = true ; - break ; - } - } - if(vo.hasPower){ - break ; - } - } - } + vo.hasPower = this.hasOnePrivilege(privilege, userPo) ; + if(!vo.hasPower){ + vo.hasPower = this.hasAllPrivilege(allPrivilege, userPo) ; + if(!vo.hasPower){ + vo.hasPower = this.hasAnyPrivilege(anyPrivilege, userPo) ; } } } } - vo.dataSourceName = userPo.getOrgTag() ; }else{ vo.logined = false ; vo.hasPower = false ; @@ -262,5 +222,79 @@ return vo ; } + /** + * 鍒ゆ柇鐧诲綍鐢ㄦ埛鏄惁鎷ユ湁鎸囧畾鐨勪竴涓潈闄� + * @param privilege 鎸囧畾鐨勪竴涓潈闄� + * @param userPo 褰撳墠鐧诲綍鐢ㄦ埛 + * @return 鏄惁鏈夋潈闄� + */ + private boolean hasOnePrivilege(String privilege, BaUser userPo){ + boolean hasPrivilege = false ; + if (privilege != null && !privilege.trim().equals("")) { + int intPri = Integer.parseInt(privilege); + for (Integer pri : userPo.privileges) { + if (pri == intPri) { + hasPrivilege = true; + break; + } + } + } + return hasPrivilege ; + } + + /** + * 鍒ゆ柇鐧诲綍鐢ㄦ埛鏄惁鎷ユ湁鎸囧畾鐨勫涓潈闄� + * @param allPrivilege 鎸囧畾鐨勫涓潈闄� + * @param userPo 褰撳墠鐧诲綍鐢ㄦ埛 + * @return 鏄惁鏈夋潈闄� + */ + private boolean hasAllPrivilege(String[] allPrivilege, BaUser userPo){ + boolean hasPrivilege = false ; + if(allPrivilege != null && allPrivilege.length > 0){ + int intPri ; + int count = 0 ; + for(String strPri : allPrivilege){ + intPri = Integer.parseInt(strPri) ; + for(Integer pri : userPo.privileges){ + if(pri == intPri){ + count++ ; + break ; + } + } + } + if(count == allPrivilege.length){ + hasPrivilege = true ; + } + } + return hasPrivilege ; + } + + + /** + * 鍒ゆ柇鐧诲綍鐢ㄦ埛鏄惁鎷ユ湁鎸囧畾鐨勬煇涓潈闄� + * @param anyPrivilege 鎸囧畾鐨勫涓潈闄� + * @param userPo 褰撳墠鐧诲綍鐢ㄦ埛 + * @return 鏄惁鏈夋潈闄� + */ + private boolean hasAnyPrivilege(String[] anyPrivilege, BaUser userPo){ + boolean hasPrivilege = false ; + int intPri ; + if(anyPrivilege != null && anyPrivilege.length > 0){ + for(String strPri : anyPrivilege){ + intPri = Integer.parseInt(strPri) ; + for(Integer pri : userPo.privileges){ + if(pri == intPri){ + hasPrivilege = true ; + break ; + } + } + if(hasPrivilege){ + break ; + } + } + } + return hasPrivilege ; + } + } -- Gitblit v1.8.0