MySQL数据权限的实现详情

private static final String SQL_AND = ” and “;

private static final String SQL_OR = ” or “;

private static final String SQL_JOINT = ” (%s) “;

/**
* 获取带有数据权限的SQL
* @param identityId 身份ID
*/
public String getPermissionSql(String identityId) {
//————————获取当前身份的数据规则————————————
List<RoleComponentRuleDTO> conditionList = getCurrentIdentyPermission(identityId);
if (CollectionUtils.isEmpty(conditionList)) {
//没有权限
return “1 = 0”;
}
//存在权限
//对当前身份根据规则编码分组-去除不同角色中相同编码且规则值为ALL的规则 并根据角色id分组
Map<String, List<RoleComponentRuleDTO>> ruleMap = getRuleMapByRoleId(conditionList);

StringBuilder sb = new StringBuilder();
String roleSql;
if (MapUtils.isNotEmpty(ruleMap)) {
//按角色拼接SQL
for (Map.Entry<String, List<RoleComponentRuleDTO>> entry : ruleMap.entrySet()) {

List<RoleComponentRuleDTO> lists = entry.getValue();

// 同角色之间使用 AND
roleSql = buildRoleSql(lists);

//角色之间使用 OR
if (StringUtils.isNotEmpty(roleSql)) {
jointSqlByRoles(sb, roleSql);
}
}

}
return sb.toString();
}

private static List<RoleComponentRuleDTO> getCurrentIdentyPermission(String identityId) {
//—————————-获取所有数据规则—————————–
List<RoleComponentRuleDTO> roleRuleList = DataPermissionUtils.loadDataSearchCondition();
if(CollectionUtils.isEmpty(roleRuleList)){
return null;
}
//—————————–过滤掉不属于当前身份的规则———————————–
return roleRuleList.stream()
.filter(item -> item.getIdentityId().equals(identityId))
.collect(Collectors.toList());
}

/**
* 构建单角色SQL
*/
private static String buildRoleSql(List<RoleComponentRuleDTO> lists) {
StringBuilder roleSql = new StringBuilder();
for (RoleComponentRuleDTO item : lists) {
//如果出现全选 则 代表全部，不需要限定范围
if (“ALL”.equals(item.getRuleValue())) {
continue;
}
//将规则转换成SQL
String filedSql = convertRuleToSql(item);

roleSql.append(SQL_AND).append(filedSql);
}
return roleSql.toString();
}

/**
* 将单一规则转化成SQL，默认全部使用 In
* ruleCode : area_test
* ruleValue : 区域1，区域2，区域3
* @param rule 规则值
*/
private static String convertRuleToSql(RoleComponentRuleDTO rule) {
String whereCondition = ” in “;
String ruleValueConvert = getInConditionValue(rule.getRuleValue());
return rule.getRuleCode() + whereCondition + ruleValueConvert;
}

/**
* IN字符串转换
* 区域1, 区域2, 区域3 –> (“区域1″,”区域2″,”区域3”)
* 江西大区 –> (“江西大区”)
*/
private static String getInConditionValue(String ruleValue) {
String[] temp = ruleValue.split(“,”);
StringBuilder res = new StringBuilder();
for (String string : temp) {
res.append(“,'”).append(string).append(“‘”);
}
return “(” + res.substring(1) + “)”;
}

/**
* 拼接单角色的SQL
* @param sqlBuilder 总的SQL
* @param roleSql 单角色SQL
*/
private static void jointSqlByRoles(StringBuilder sqlBuilder, String roleSql) {
roleSql = roleSql.replaceFirst(SQL_AND, “”);
if (StringUtils.isEmpty(sqlBuilder.toString())) {
sqlBuilder.append(String.format(SQL_JOINT, roleSql));
} else {
sqlBuilder.append(SQL_OR).append(String.format(SQL_JOINT, roleSql));
}
}

/**
*
* 1. 对当前身份根据规则编码分组-去除不同角色中相同编码且规则值为ALL的规则
* 2. 对角色进行分组
* @param conditionList 数据规则
* @return 分组后的规则list
*/
private static Map<String, List<RoleComponentRuleDTO>> getRuleMapByRoleId(List<RoleComponentRuleDTO> conditionList) {
//——–过滤掉不属于当前身份的规则,并对条件编码进行分组———————————–
Map<String, List<RoleComponentRuleDTO>> conditionMap = conditionList.stream().collect(Collectors.groupingBy(RoleComponentRuleDTO::getRuleCode));

//——–相同编码分组中存在ALL的排除掉———————————————–
List<RoleComponentRuleDTO> newRoleRuleList = new ArrayList<>();
if (MapUtils.isNotEmpty(conditionMap)) {
for (Map.Entry<String, List<RoleComponentRuleDTO>> entry : conditionMap.entrySet()) {
boolean flag = true;
List<RoleComponentRuleDTO> lists = entry.getValue();
for (RoleComponentRuleDTO item : lists) {
if (“ALL”.equals(item.getRuleValue())) {
flag = false;
break;
}
}

if (flag) {
newRoleRuleList.addAll(lists);
}
}
}
if (CollectionUtils.isNotEmpty(newRoleRuleList)) {
return newRoleRuleList.stream().collect(Collectors.groupingBy(RoleComponentRuleDTO::getRoleId));
}
return Maps.newHashMap();
}
}

Dao层接受service层传入已经生成好的sql语句，作为查询条件直接拼接在业务语句之后。

小结

以上，就是数据权限的实现过程，其实代码实现并不复杂，主要还是得理解其中的实现原理。如果你也有数据权限的需求，不妨参考一下。

到此这篇关于MySQL数据权限的实现详情的文章就介绍到这了,更多相关SQL数据权限内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持！