kd-aiot-frontend-eam/src/utils/token-util.js

/**
 * token 操作封装
 */
import { TOKEN_STORE_NAME } from '@/config/setting';

/**
 * 获取缓存的 token
 */
export function getToken() {
  const token = localStorage.getItem(TOKEN_STORE_NAME);
  if (!token) {
    return sessionStorage.getItem(TOKEN_STORE_NAME);
  }
  return token;
}

/**
 * 缓存 token
 * @param token token
 * @param remember 是否永久存储
 */
export function setToken(token, remember) {
  removeToken();
  if (token) {
    if (remember) {
      localStorage.setItem(TOKEN_STORE_NAME, token);
    } else {
      sessionStorage.setItem(TOKEN_STORE_NAME, token);
    }
  }
}

/**
 * 获取缓存的 当前登陆角色
 */
export function getCurrentUser() {
  let currentUser = sessionStorage.getItem('currentUser');
  if (!currentUser) {
    currentUser = localStorage.getItem('currentUser');
  }
  return JSON.parse(currentUser);
}

/**
 * 缓存 当前登陆角色
 * @param token token
 * @param remember 是否永久存储
 */
export function setCurrentUser(currentUser, remember) {
  removeCurrentUser()
  if (currentUser) {
    if (remember) {
      localStorage.setItem('currentUser', JSON.stringify(currentUser));
    } else {
      sessionStorage.setItem('currentUser', JSON.stringify(currentUser));
    }
  }
}
/**
 * 移除 token
 */
export function removeToken() {
  localStorage.removeItem(TOKEN_STORE_NAME);
  sessionStorage.removeItem(TOKEN_STORE_NAME);
}
/**
 * 移除 currentUser
 */
export function removeCurrentUser() {
  localStorage.removeItem('currentUser');
  sessionStorage.removeItem('currentUser');
}