jtzx-crm/src/main/java/cc/uncarbon/helper/RolePermissionCacheHelper.java

package cc.uncarbon.helper;

import cc.uncarbon.framework.core.context.UserContextHolder;
import cn.hutool.core.collection.CollUtil;

import java.util.*;

import lombok.RequiredArgsConstructor;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

/**
 * 将角色对应权限，缓存至 Redis
 * 参考文章: https://sa-token.dev33.cn/doc/index.html#/fun/jur-cache
 *
 * @author Uncarbon
 */
@Component
@RequiredArgsConstructor
public class RolePermissionCacheHelper {

    private final RedisTemplate<String, Set<String>> stringSetRedisTemplate;

    private static final String CACHE_KEY_ROLE_PERMISSIONS = "Authorization:rolePermissions:roleId_%s";


    /**
     * 从缓存中取得当前用户拥有的所有权限名集合
     *
     * @return List<String>
     */
    public List<String> getUserPermissions() {
        Set<Long> rolesIds = UserContextHolder.getUserContext().getRolesIds();
        // aka * 64
        List<String> ret = new ArrayList<>(rolesIds.size() << 6);

        rolesIds.forEach(
                roleId -> {
                    String cacheKey = String.format(CACHE_KEY_ROLE_PERMISSIONS, roleId);
                    ret.addAll(CollUtil.emptyIfNull(stringSetRedisTemplate.opsForValue().get(cacheKey)));
                }
        );

        return ret;
    }

    /**
     * 覆盖更新角色对应权限至 Redis
     *
     * @param map key=角色ID value=权限集合
     */
    public void putCache(Map<Long, Set<String>> map) {
        Set<Map.Entry<Long, Set<String>>> entries = map.entrySet();
        entries.forEach(
                entry -> this.putCache(entry.getKey(), entry.getValue())
        );
    }

    /**
     * 覆盖更新角色对应权限至 Redis
     *
     * @param roleId 角色ID
     * @param newPermissions 新权限名集合
     */
    public void putCache(Long roleId, Set<String> newPermissions) {
        String cacheKey = String.format(CACHE_KEY_ROLE_PERMISSIONS, roleId);
        stringSetRedisTemplate.opsForValue().set(cacheKey, newPermissions);
    }

    /**
     * 删除角色ID对应的权限缓存
     *
     * @param roleIds 角色ID集合
     */
    public void deleteCache(Collection<Long> roleIds) {
        roleIds.forEach(
                roleId -> {
                    String cacheKey = String.format(CACHE_KEY_ROLE_PERMISSIONS, roleId);
                    stringSetRedisTemplate.delete(cacheKey);
                }
        );
    }
}