content: /*
   * Copyright 1999-2025 Alibaba Group Holding Ltd.
   *
   * 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
   *
   *      http://www.apache.org/licenses/LICENSE-2.0
   *
   * 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 com.alibaba.nacos.plugin.auth.impl.roles;
  
  import com.alibaba.nacos.api.model.Page;
  import com.alibaba.nacos.common.utils.ConcurrentHashSet;
  import com.alibaba.nacos.core.utils.Loggers;
  import com.alibaba.nacos.plugin.auth.impl.persistence.PermissionInfo;
  import com.alibaba.nacos.plugin.auth.impl.persistence.RoleInfo;
  import org.springframework.scheduling.annotation.Scheduled;
  
  import java.util.ArrayList;
  import java.util.HashSet;
  import java.util.List;
  import java.util.Map;
  import java.util.Set;
  import java.util.concurrent.ConcurrentHashMap;
  
  /**
   * Nacos abstract cached role service.
   *
   * @author xiweng.yy
   */
  public abstract class AbstractCachedRoleService implements NacosRoleService {
      
      protected static final int DEFAULT_PAGE_NO = 1;
      
      private volatile Set<String> roleSet = new ConcurrentHashSet<>();
      
      private volatile Map<String, List<RoleInfo>> roleInfoMap = new ConcurrentHashMap<>();
      
      private volatile Map<String, List<PermissionInfo>> permissionInfoMap = new ConcurrentHashMap<>();
      
      protected Set<String> getCachedRoleSet() {
          return roleSet;
      }
      
      protected Map<String, List<RoleInfo>> getCachedRoleInfoMap() {
          return roleInfoMap;
      }
      
      protected Map<String, List<PermissionInfo>> getCachedPermissionInfoMap() {
          return permissionInfoMap;
      }
      
      @Scheduled(initialDelay = 5000, fixedDelay = 15000)
      protected void reload() {
          try {
              List<RoleInfo> roleInfoPage = getAllRoles();
              Set<String> tmpRoleSet = new HashSet<>(16);
              Map<String, List<RoleInfo>> tmpRoleInfoMap = new ConcurrentHashMap<>(16);
              for (RoleInfo roleInfo : roleInfoPage) {
                  if (!tmpRoleInfoMap.containsKey(roleInfo.getUsername())) {
                      tmpRoleInfoMap.put(roleInfo.getUsername(), new ArrayList<>());
                  }
                  tmpRoleInfoMap.get(roleInfo.getUsername()).add(roleInfo);
                  tmpRoleSet.add(roleInfo.getRole());
              }
              
              Map<String, List<PermissionInfo>> tmpPermissionInfoMap = new ConcurrentHashMap<>(16);
              for (String role : tmpRoleSet) {
                  Page<PermissionInfo> permissionInfoPage = getPermissions(role, DEFAULT_PAGE_NO, Integer.MAX_VALUE);
                  tmpPermissionInfoMap.put(role, permissionInfoPage.getPageItems());
              }
              
              roleSet = tmpRoleSet;
              roleInfoMap = tmpRoleInfoMap;
              permissionInfoMap = tmpPermissionInfoMap;
          } catch (Exception e) {
              Loggers.AUTH.warn("[LOAD-ROLES] load failed", e);
          }
      }
  }
  