update 优化 支持脱敏传多角色多权限标识

dev
疯狂的狮子Li 1 year ago
parent c7eb77d730
commit 5f3ce86836

@ -22,7 +22,13 @@ import java.lang.annotation.Target;
public @interface Sensitive { public @interface Sensitive {
SensitiveStrategy strategy(); SensitiveStrategy strategy();
String roleKey() default ""; /**
*
*/
String[] roleKey() default {};
String perms() default ""; /**
*
*/
String[] perms() default {};
} }

@ -13,6 +13,6 @@ public interface SensitiveService {
/** /**
* *
*/ */
boolean isSensitive(String roleKey, String perms); boolean isSensitive(String[] roleKey, String[] perms);
} }

@ -26,8 +26,8 @@ import java.util.Objects;
public class SensitiveHandler extends JsonSerializer<String> implements ContextualSerializer { public class SensitiveHandler extends JsonSerializer<String> implements ContextualSerializer {
private SensitiveStrategy strategy; private SensitiveStrategy strategy;
private String roleKey; private String[] roleKey;
private String perms; private String[] perms;
@Override @Override
public void serialize(String value, JsonGenerator gen, SerializerProvider serializers) throws IOException { public void serialize(String value, JsonGenerator gen, SerializerProvider serializers) throws IOException {

@ -1,7 +1,7 @@
package org.dromara.demo.service.impl; package org.dromara.demo.service.impl;
import cn.dev33.satoken.stp.StpUtil; import cn.dev33.satoken.stp.StpUtil;
import org.dromara.common.core.utils.StringUtils; import cn.hutool.core.util.ArrayUtil;
import org.dromara.common.satoken.utils.LoginHelper; import org.dromara.common.satoken.utils.LoginHelper;
import org.dromara.common.sensitive.core.SensitiveService; import org.dromara.common.sensitive.core.SensitiveService;
import org.dromara.common.tenant.helper.TenantHelper; import org.dromara.common.tenant.helper.TenantHelper;
@ -21,19 +21,19 @@ public class SensitiveServiceImpl implements SensitiveService {
* *
*/ */
@Override @Override
public boolean isSensitive(String roleKey, String perms) { public boolean isSensitive(String[] roleKey, String[] perms) {
if (!LoginHelper.isLogin()) { if (!LoginHelper.isLogin()) {
return true; return true;
} }
boolean roleExist = StringUtils.isNotBlank(roleKey); boolean roleExist = ArrayUtil.isNotEmpty(roleKey);
boolean permsExist = StringUtils.isNotBlank(perms); boolean permsExist = ArrayUtil.isNotEmpty(perms);
if (roleExist && permsExist) { if (roleExist && permsExist) {
if (StpUtil.hasRole(roleKey) && StpUtil.hasPermission(perms)) { if (StpUtil.hasRoleOr(roleKey) && StpUtil.hasPermissionOr(perms)) {
return false; return false;
} }
} else if (roleExist && StpUtil.hasRole(roleKey)) { } else if (roleExist && StpUtil.hasRoleOr(roleKey)) {
return false; return false;
} else if (permsExist && StpUtil.hasPermission(perms)) { } else if (permsExist && StpUtil.hasPermissionOr(perms)) {
return false; return false;
} }

@ -1,7 +1,7 @@
package org.dromara.system.service.impl; package org.dromara.system.service.impl;
import cn.dev33.satoken.stp.StpUtil; import cn.dev33.satoken.stp.StpUtil;
import org.dromara.common.core.utils.StringUtils; import cn.hutool.core.util.ArrayUtil;
import org.dromara.common.satoken.utils.LoginHelper; import org.dromara.common.satoken.utils.LoginHelper;
import org.dromara.common.sensitive.core.SensitiveService; import org.dromara.common.sensitive.core.SensitiveService;
import org.dromara.common.tenant.helper.TenantHelper; import org.dromara.common.tenant.helper.TenantHelper;
@ -22,19 +22,19 @@ public class SysSensitiveServiceImpl implements SensitiveService {
* *
*/ */
@Override @Override
public boolean isSensitive(String roleKey, String perms) { public boolean isSensitive(String[] roleKey, String[] perms) {
if (!LoginHelper.isLogin()) { if (!LoginHelper.isLogin()) {
return true; return true;
} }
boolean roleExist = StringUtils.isNotEmpty(roleKey); boolean roleExist = ArrayUtil.isNotEmpty(roleKey);
boolean permsExist = StringUtils.isNotEmpty(perms); boolean permsExist = ArrayUtil.isNotEmpty(perms);
if (roleExist && permsExist) { if (roleExist && permsExist) {
if (StpUtil.hasRole(roleKey) && StpUtil.hasPermission(perms)) { if (StpUtil.hasRoleOr(roleKey) && StpUtil.hasPermissionOr(perms)) {
return false; return false;
} }
} else if (roleExist && StpUtil.hasRole(roleKey)) { } else if (roleExist && StpUtil.hasRoleOr(roleKey)) {
return false; return false;
} else if (permsExist && StpUtil.hasPermission(perms)) { } else if (permsExist && StpUtil.hasPermissionOr(perms)) {
return false; return false;
} }

Loading…
Cancel
Save