|
|
|
|
@ -1,13 +1,24 @@
|
|
|
|
|
package cn.iocoder.yudao.framework.security.core.service;
|
|
|
|
|
|
|
|
|
|
import cn.hutool.core.collection.CollUtil;
|
|
|
|
|
import cn.hutool.core.collection.ListUtil;
|
|
|
|
|
import cn.hutool.core.util.ArrayUtil;
|
|
|
|
|
import cn.hutool.core.util.HashUtil;
|
|
|
|
|
import cn.hutool.core.util.ObjectUtil;
|
|
|
|
|
import cn.iocoder.yudao.framework.common.core.KeyValue;
|
|
|
|
|
import cn.iocoder.yudao.framework.common.pojo.CommonResult;
|
|
|
|
|
import cn.iocoder.yudao.framework.common.util.cache.CacheUtils;
|
|
|
|
|
import cn.iocoder.yudao.framework.security.core.LoginUser;
|
|
|
|
|
import cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils;
|
|
|
|
|
import cn.iocoder.yudao.module.system.api.permission.PermissionApi;
|
|
|
|
|
import com.google.common.cache.CacheLoader;
|
|
|
|
|
import com.google.common.cache.LoadingCache;
|
|
|
|
|
import lombok.AllArgsConstructor;
|
|
|
|
|
import lombok.SneakyThrows;
|
|
|
|
|
|
|
|
|
|
import java.time.Duration;
|
|
|
|
|
import java.util.Arrays;
|
|
|
|
|
import java.util.List;
|
|
|
|
|
|
|
|
|
|
import static cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils.getLoginUserId;
|
|
|
|
|
|
|
|
|
|
@ -21,16 +32,45 @@ public class SecurityFrameworkServiceImpl implements SecurityFrameworkService {
|
|
|
|
|
|
|
|
|
|
private final PermissionApi permissionApi;
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 针对 {@link #hasAnyRoles(String...)} 的缓存
|
|
|
|
|
*/
|
|
|
|
|
private final LoadingCache<KeyValue<Long, List<String>>, Boolean> hasAnyRolesCache = CacheUtils.buildAsyncReloadingCache(
|
|
|
|
|
Duration.ofMinutes(1L), // 过期时间 1 分钟
|
|
|
|
|
new CacheLoader<KeyValue<Long, List<String>>, Boolean>() {
|
|
|
|
|
@Override
|
|
|
|
|
public Boolean load(KeyValue<Long, List<String>> key) {
|
|
|
|
|
CommonResult<Boolean> hasAnyRolesResult = permissionApi.hasAnyRoles(key.getKey(),
|
|
|
|
|
key.getValue().toArray(new String[0]));
|
|
|
|
|
hasAnyRolesResult.checkError();
|
|
|
|
|
return hasAnyRolesResult.getData();
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 针对 {@link #hasAnyPermissions(String...)} 的缓存
|
|
|
|
|
*/
|
|
|
|
|
private final LoadingCache<KeyValue<Long, List<String>>, Boolean> hasAnyPermissionsCache = CacheUtils.buildAsyncReloadingCache(
|
|
|
|
|
Duration.ofMinutes(1L), // 过期时间 1 分钟
|
|
|
|
|
new CacheLoader<KeyValue<Long, List<String>>, Boolean>() {
|
|
|
|
|
@Override
|
|
|
|
|
public Boolean load(KeyValue<Long, List<String>> key) {
|
|
|
|
|
CommonResult<Boolean> hasAnyPermissionsResult = permissionApi.hasAnyPermissions(key.getKey(),
|
|
|
|
|
key.getValue().toArray(new String[0]));
|
|
|
|
|
hasAnyPermissionsResult.checkError();
|
|
|
|
|
return hasAnyPermissionsResult.getData();
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public boolean hasPermission(String permission) {
|
|
|
|
|
return hasAnyPermissions(permission);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
@SneakyThrows
|
|
|
|
|
public boolean hasAnyPermissions(String... permissions) {
|
|
|
|
|
CommonResult<Boolean> hasAnyPermissionsResult = permissionApi.hasAnyPermissions(getLoginUserId(), permissions);
|
|
|
|
|
hasAnyPermissionsResult.checkError();
|
|
|
|
|
return hasAnyPermissionsResult.getData();
|
|
|
|
|
return hasAnyPermissionsCache.get(new KeyValue<>(getLoginUserId(), Arrays.asList(permissions)));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
@ -39,10 +79,9 @@ public class SecurityFrameworkServiceImpl implements SecurityFrameworkService {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
@SneakyThrows
|
|
|
|
|
public boolean hasAnyRoles(String... roles) {
|
|
|
|
|
CommonResult<Boolean> hasAnyRolesResult = permissionApi.hasAnyRoles(getLoginUserId(), roles);
|
|
|
|
|
hasAnyRolesResult.checkError();
|
|
|
|
|
return hasAnyRolesResult.getData();
|
|
|
|
|
return hasAnyRolesCache.get(new KeyValue<>(getLoginUserId(), Arrays.asList(roles)));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
|