1-招聘; 2-求职; 3-教师; 存取token

src/utils/auth.js

@@ -9,32 +9,37 @@ export const getIsEnterprise = () => {
   return PATH_NAME.includes(ENTERPRISE_PATH)
 }
 
+
+const ACCESS_TOKEN_NAME = ['ENT_ACCESS_TOKEN', 'PER_ACCESS_TOKEN', 'TEA_ACCESS_TOKEN']
+const REFRESH_TOKEN_NAME = ['ENT_REFRESH_TOKEN', 'PER_REFRESH_TOKEN', 'TEA_REFRESH_TOKEN']
+
 // 获取token
-export const getToken = (index = 2) => { // index=1: 使用招聘token; index=2: 使用求职token
-  const arr = ['ENT_ACCESS_TOKEN', 'PER_ACCESS_TOKEN']
-  return localStorage.getItem(arr[index-1])
+export const getToken = (index = 2) => { // index: 1-招聘; 2-求职; 3-教师
+  return localStorage.getItem(ACCESS_TOKEN_NAME[index-1])
 }
 
 // 设置token
-export const setToken = (token, index = 2) => { // index=1: 招聘token; index=2: 求职token
-  return localStorage.setItem((index === 1 ? 'ENT_ACCESS_TOKEN' : 'PER_ACCESS_TOKEN'), token)
+export const setToken = (token, index = 2) => { // index: 1-招聘; 2-求职; 3-教师
+  return localStorage.setItem(ACCESS_TOKEN_NAME[index-1], token)
 }
 
 // 清除token
 export const removeToken = () => {
-  localStorage.removeItem('ENT_ACCESS_TOKEN')
-  localStorage.removeItem('PER_ACCESS_TOKEN')
-  localStorage.removeItem('ENT_REFRESH_TOKEN')
-  localStorage.removeItem('PER_REFRESH_TOKEN')
+  ACCESS_TOKEN_NAME.forEach(item => localStorage.removeItem(item))
+  REFRESH_TOKEN_NAME.forEach(item => localStorage.removeItem(item))
+  // localStorage.removeItem('ENT_ACCESS_TOKEN')
+  // localStorage.removeItem('PER_ACCESS_TOKEN')
+  // localStorage.removeItem('ENT_REFRESH_TOKEN')
+  // localStorage.removeItem('PER_REFRESH_TOKEN')
 }
 // 获取刷新token
 export const getRefreshToken = (index) => {
-  return localStorage.getItem(index === 1 ? 'ENT_REFRESH_TOKEN' : 'PER_REFRESH_TOKEN')
+  return localStorage.getItem(ACCESS_TOKEN_NAME[index-1])
 }
 
 // 设置刷新token
 export const setRefreshToken = (refreshToken, index = 2) => { // index=1: 招聘token; index=2: 求职token
-  return localStorage.setItem((index === 1 ? 'ENT_REFRESH_TOKEN' : 'PER_REFRESH_TOKEN'), refreshToken)
+  return localStorage.setItem((ACCESS_TOKEN_NAME[index-1]), refreshToken)
 }
 
 // 租户ID