Carina
3 years ago
parent
d948915ed8
commit
61ec20b260
@ -0,0 +1,232 @@
|
|||||||
|
/**
|
||||||
|
* Copyright 2020 OPSLI 快速开发平台 https://www.opsli.com
|
||||||
|
* <p>
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); you may not
|
||||||
|
* use this file except in compliance with the License. You may obtain a copy of
|
||||||
|
* the License at
|
||||||
|
* <p>
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
* <p>
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
* License for the specific language governing permissions and limitations under
|
||||||
|
* the License.
|
||||||
|
*/
|
||||||
|
package org.opsli.core.persistence.querybuilder.chain;
|
||||||
|
|
||||||
|
import cn.hutool.core.util.ReflectUtil;
|
||||||
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
||||||
|
import com.google.common.collect.Lists;
|
||||||
|
import org.apache.commons.lang3.StringUtils;
|
||||||
|
import org.opsli.api.wrapper.system.role.RoleModel;
|
||||||
|
import org.opsli.api.wrapper.system.user.UserModel;
|
||||||
|
import org.opsli.api.wrapper.system.user.UserOrgRefModel;
|
||||||
|
import org.opsli.common.constants.MyBatisConstants;
|
||||||
|
import org.opsli.common.utils.FieldUtil;
|
||||||
|
import org.opsli.common.utils.ListDistinctUtil;
|
||||||
|
import org.opsli.core.base.entity.BaseEntity;
|
||||||
|
import org.opsli.core.persistence.querybuilder.conf.WebQueryConf;
|
||||||
|
import org.opsli.core.utils.UserUtil;
|
||||||
|
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 数据权限赋值处理
|
||||||
|
*
|
||||||
|
* @author Parker
|
||||||
|
* @date 2020-09-13 19:36
|
||||||
|
*/
|
||||||
|
public class QueryDataPermsHandler implements QueryBuilderChain{
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 子 责任链
|
||||||
|
*/
|
||||||
|
private QueryBuilderChain queryBuilderChain;
|
||||||
|
|
||||||
|
|
||||||
|
public QueryDataPermsHandler(){}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 构造函数
|
||||||
|
* @param queryBuilderChain 责任链
|
||||||
|
*/
|
||||||
|
public QueryDataPermsHandler(QueryBuilderChain queryBuilderChain){
|
||||||
|
this.queryBuilderChain = queryBuilderChain;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public <T extends BaseEntity> QueryWrapper<T> handler(Class<T> entityClazz, QueryWrapper<T> wrapper) {
|
||||||
|
// 执行 子 责任链
|
||||||
|
if(queryBuilderChain != null){
|
||||||
|
wrapper = queryBuilderChain.handler(entityClazz, wrapper);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 自身责任 -- 判断组织
|
||||||
|
boolean flag = ReflectUtil.hasField(entityClazz, MyBatisConstants.FIELD_ORG_GROUP);
|
||||||
|
if(flag) {
|
||||||
|
String fieldName = FieldUtil.humpToUnderline(MyBatisConstants.FIELD_ORG_GROUP);
|
||||||
|
// 处理查询条件
|
||||||
|
handleDataPermsCondition(fieldName, wrapper);
|
||||||
|
}
|
||||||
|
|
||||||
|
return wrapper;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public <T extends BaseEntity> QueryWrapper<T> handler(Class<T> entityClazz, WebQueryConf webQueryConf, QueryWrapper<T> wrapper) {
|
||||||
|
// 执行 子 责任链
|
||||||
|
if(queryBuilderChain != null){
|
||||||
|
wrapper = queryBuilderChain.handler(entityClazz, webQueryConf, wrapper);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 自身责任 -- 判断组织
|
||||||
|
boolean flag = ReflectUtil.hasField(entityClazz, MyBatisConstants.FIELD_ORG_GROUP);
|
||||||
|
if(flag) {
|
||||||
|
String fieldName = webQueryConf.get(MyBatisConstants.FIELD_ORG_GROUP);
|
||||||
|
if(StringUtils.isEmpty(fieldName)){
|
||||||
|
fieldName = FieldUtil.humpToUnderline(MyBatisConstants.FIELD_ORG_GROUP);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 处理查询条件
|
||||||
|
handleDataPermsCondition(fieldName, wrapper);
|
||||||
|
}
|
||||||
|
|
||||||
|
return wrapper;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 处理 数据权限 条件
|
||||||
|
*
|
||||||
|
* @param field 查询字段
|
||||||
|
* @param queryWrapper 组织集合
|
||||||
|
*/
|
||||||
|
private static <T extends BaseEntity> QueryWrapper<T> handleDataPermsCondition(String field, QueryWrapper<T> queryWrapper) {
|
||||||
|
|
||||||
|
// 1. 当前用户
|
||||||
|
UserModel currUser = UserUtil.getUser();
|
||||||
|
|
||||||
|
// 2. 当前用户 组织机构集合
|
||||||
|
List<UserOrgRefModel> userOrgRefModelList = UserUtil.getOrgListByUserId(currUser.getId());
|
||||||
|
List<String> orgIdGroupList = Lists.newArrayListWithCapacity(userOrgRefModelList.size());
|
||||||
|
for (UserOrgRefModel userOrgRefModel : userOrgRefModelList) {
|
||||||
|
orgIdGroupList.add(userOrgRefModel.getOrgIds());
|
||||||
|
}
|
||||||
|
// 组织机构集合 去重
|
||||||
|
orgIdGroupList = ListDistinctUtil.distinct(orgIdGroupList);
|
||||||
|
|
||||||
|
// 3. 获得查询类型
|
||||||
|
// 如果是超级管理员 则查询类型为全部
|
||||||
|
ConditionType conditionType = ConditionType.SELF;
|
||||||
|
if(StringUtils.equals(UserUtil.SUPER_ADMIN, currUser.getUsername())){
|
||||||
|
conditionType = ConditionType.ALL;
|
||||||
|
}else{
|
||||||
|
// 如果不是超级管理员 则获得当前用户的默认角色下的 授权数据权限类型
|
||||||
|
RoleModel defRole = UserUtil.getUserDefRoleByUserId(currUser.getId());
|
||||||
|
if(null != defRole){
|
||||||
|
conditionType = ConditionType.getConditionType(defRole.getDataScope());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 4. 如果查询字段为空 则默认
|
||||||
|
if(StringUtils.isBlank(field)){
|
||||||
|
field = FieldUtil.humpToUnderline(MyBatisConstants.FIELD_ORG_GROUP);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 常量
|
||||||
|
final ConditionType finalConditionType = conditionType;
|
||||||
|
final String finalField = field;
|
||||||
|
final List<String> finalOrgIdGroupList = orgIdGroupList;
|
||||||
|
|
||||||
|
// 查询 全部
|
||||||
|
if(ConditionType.ALL.equals(finalConditionType)){
|
||||||
|
// ..
|
||||||
|
}else {
|
||||||
|
queryWrapper.and(wra -> {
|
||||||
|
// 查询 本部门
|
||||||
|
if(ConditionType.DEPT.equals(finalConditionType)){
|
||||||
|
wra.in(finalField, finalOrgIdGroupList);
|
||||||
|
}
|
||||||
|
// 部门及以下
|
||||||
|
else if(ConditionType.DEPT_AND_BELOW.equals(finalConditionType)){
|
||||||
|
wra.and(wraConfine -> {
|
||||||
|
// 增加右模糊 查询条件
|
||||||
|
for (int i = 0; i < finalOrgIdGroupList.size(); i++) {
|
||||||
|
// 右模糊匹配
|
||||||
|
wraConfine.likeRight(
|
||||||
|
finalField, finalOrgIdGroupList.get(i));
|
||||||
|
|
||||||
|
if(i < finalOrgIdGroupList.size() - 1){
|
||||||
|
wraConfine.or();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}else {
|
||||||
|
// 查自身
|
||||||
|
wra.in(FieldUtil.humpToUnderline(MyBatisConstants.FIELD_CREATE_BY), currUser.getId());
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
return queryWrapper;
|
||||||
|
}
|
||||||
|
|
||||||
|
// =================================
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 条件类型
|
||||||
|
*/
|
||||||
|
public enum ConditionType{
|
||||||
|
|
||||||
|
/** 自身 */
|
||||||
|
SELF("0", "自身"),
|
||||||
|
|
||||||
|
/** 部门 */
|
||||||
|
DEPT("1", "本部门"),
|
||||||
|
|
||||||
|
/** 部门及以下 */
|
||||||
|
DEPT_AND_BELOW("2", "本部门及以下"),
|
||||||
|
|
||||||
|
/** 全部 */
|
||||||
|
ALL("3", "全部");
|
||||||
|
|
||||||
|
|
||||||
|
/** 值 */
|
||||||
|
private final String value;
|
||||||
|
|
||||||
|
/** 描述 */
|
||||||
|
private final String describe;
|
||||||
|
|
||||||
|
ConditionType(String value, String describe){
|
||||||
|
this.value = value;
|
||||||
|
this.describe = describe;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getValue() {
|
||||||
|
return value;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getDescribe() {
|
||||||
|
return describe;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获得 类型
|
||||||
|
* @param value 值
|
||||||
|
* @return AlgSource
|
||||||
|
*/
|
||||||
|
public static ConditionType getConditionType(String value) {
|
||||||
|
ConditionType[] var1 = values();
|
||||||
|
for (ConditionType source : var1) {
|
||||||
|
if(source.value.equals(value)){
|
||||||
|
return source;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// 如果条件类型为空 则默认 查看自身数据
|
||||||
|
return ConditionType.SELF;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
@ -1,71 +0,0 @@
|
|||||||
/**
|
|
||||||
* Copyright 2020 OPSLI 快速开发平台 https://www.opsli.com
|
|
||||||
* <p>
|
|
||||||
* Licensed under the Apache License, Version 2.0 (the "License"); you may not
|
|
||||||
* use this file except in compliance with the License. You may obtain a copy of
|
|
||||||
* the License at
|
|
||||||
* <p>
|
|
||||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
* <p>
|
|
||||||
* Unless required by applicable law or agreed to in writing, software
|
|
||||||
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
||||||
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
||||||
* License for the specific language governing permissions and limitations under
|
|
||||||
* the License.
|
|
||||||
*/
|
|
||||||
package org.opsli.core.persistence.querybuilder.chain;
|
|
||||||
|
|
||||||
import cn.hutool.core.util.ReflectUtil;
|
|
||||||
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
|
||||||
import org.apache.commons.lang3.StringUtils;
|
|
||||||
import org.opsli.api.wrapper.system.user.UserModel;
|
|
||||||
import org.opsli.common.constants.MyBatisConstants;
|
|
||||||
import org.opsli.common.utils.FieldUtil;
|
|
||||||
import org.opsli.core.base.entity.BaseEntity;
|
|
||||||
import org.opsli.core.utils.UserUtil;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 组织赋值处理
|
|
||||||
*
|
|
||||||
* @author Parker
|
|
||||||
* @date 2020-09-13 19:36
|
|
||||||
*/
|
|
||||||
public class QueryOrgHandler implements QueryBuilderChain{
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 子 责任链
|
|
||||||
*/
|
|
||||||
private QueryBuilderChain queryBuilderChain;
|
|
||||||
|
|
||||||
|
|
||||||
public QueryOrgHandler(){}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 构造函数
|
|
||||||
* @param queryBuilderChain 责任链
|
|
||||||
*/
|
|
||||||
public QueryOrgHandler(QueryBuilderChain queryBuilderChain){
|
|
||||||
this.queryBuilderChain = queryBuilderChain;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public <T extends BaseEntity> QueryWrapper<T> handler(Class<T> entityClazz, QueryWrapper<T> wrapper) {
|
|
||||||
// 执行 子 责任链
|
|
||||||
if(queryBuilderChain != null){
|
|
||||||
wrapper = queryBuilderChain.handler(entityClazz, wrapper);
|
|
||||||
}
|
|
||||||
|
|
||||||
// 自身责任 -- 判断组织
|
|
||||||
boolean flag = ReflectUtil.hasField(entityClazz, MyBatisConstants.FIELD_ORG_GROUP);
|
|
||||||
if(flag) {
|
|
||||||
// 1. 获得 用户 数据权限
|
|
||||||
|
|
||||||
// 2. 获得 角色 数据权限
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
return wrapper;
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
Loading…
Reference in new issue