1. 程式人生 > 其它 >實現無感重新整理token

實現無感重新整理token

前言

最近在做需求的時候,涉及到登入token,產品提出一個問題:能不能讓token過期時間長一點,我頻繁的要去登入。

前端:後端,你能不能把token 過期時間設定的長一點。

後端:可以,但是那樣做不安全,你可以用更好的方法。

前端:什麼方法?

後端:給你重新整理token的介面,定時去重新整理token

前端:好,讓我思考一下

需求

當token過期的時候,重新整理token,前端需要做到無感重新整理token,即刷token時要做到使用者無感知,避免頻繁登入。 實現思路

  • 方法一

後端返回過期時間,前端判斷token過期時間,去呼叫重新整理token介面

缺點:需要後端額外提供一個token過期時間的欄位;使用了本地時間判斷,若本地時間被篡改,特別是本地時間比伺服器時間慢時,攔截會失敗。

  • 方法二

寫個定時器,定時重新整理token介面

缺點:浪費資源,消耗效能,不建議採用。

  • 方法三

在響應攔截器中攔截,判斷token 返回過期後,呼叫重新整理token介面

實現

axios的基本骨架,利用service.interceptors.response 進行攔截

import axios from 'axios'

service.interceptors.response.use(
  response => {
    if (response.data.code === 409) {
        return refreshToken({ refreshToken: localStorage.getItem('refreshToken'), token: getToken() }).then(res => {
          const { token } = res.data
          setToken(token)
          response.headers.Authorization = `${token}`
        }).catch(err => {
          removeToken()
          router.push('/login')
          return Promise.reject(err)
        })
    }
    return response && response.data
  },
  (error) => {
    Message.error(error.response.data.msg)
    return Promise.reject(error)
  }
)

問題解決

問題一:如何防止多次重新整理token

我們通過一個變數isRefreshing 去控制是否在重新整理token的狀態。

import axios from 'axios'

service.interceptors.response.use(
  response => {
    if (response.data.code === 409) {
      if (!isRefreshing) {
        isRefreshing = true
        return refreshToken({ refreshToken: localStorage.getItem('refreshToken'), token: getToken() }).then(res => {
          const { token } = res.data
          setToken(token)
          response.headers.Authorization = `${token}`
        }).catch(err => {
          removeToken()
          router.push('/login')
          return Promise.reject(err)
        }).finally(() => {
          isRefreshing = false
        })
      }
    }
    return response && response.data
  },
  (error) => {
    Message.error(error.response.data.msg)
    return Promise.reject(error)
  }
)

問題二:同時發起兩個或者兩個以上的請求時,其他介面怎麼解決

當第二個過期的請求進來,token正在重新整理,我們先將這個請求存到一個數組佇列中,想辦法讓這個請求處於等待中,一直等到重新整理token後再逐個重試清空請求佇列。 那麼如何做到讓這個請求處於等待中呢?為了解決這個問題,我們得藉助Promise。將請求存進佇列中後,同時返回一個Promise,讓這個Promise一直處於Pending狀態(即不呼叫resolve),此時這個請求就會一直等啊等,只要我們不執行resolve,這個請求就會一直在等待。當重新整理請求的介面返回來後,我們再呼叫resolve,逐個重試。最終程式碼:

import axios from 'axios'

// 是否正在重新整理的標記
let isRefreshing = false
//重試佇列
let requests = []
service.interceptors.response.use(
  response => {
  //約定code 409 token 過期
    if (response.data.code === 409) {
      if (!isRefreshing) {
        isRefreshing = true
        //呼叫重新整理token的介面
        return refreshToken({ refreshToken: localStorage.getItem('refreshToken'), token: getToken() }).then(res => {
          const { token } = res.data
          // 替換token
          setToken(token)
          response.headers.Authorization = `${token}`
           // token 重新整理後將陣列的方法重新執行
          requests.forEach((cb) => cb(token))
          requests = [] // 重新請求完清空
          return service(response.config)
        }).catch(err => {
        //跳到登入頁
          removeToken()
          router.push('/login')
          return Promise.reject(err)
        }).finally(() => {
          isRefreshing = false
        })
      } else {
        // 返回未執行 resolve 的 Promise
        return new Promise(resolve => {
          // 用函式形式將 resolve 存入,等待重新整理後再執行
          requests.push(token => {
            response.headers.Authorization = `${token}`
            resolve(service(response.config))
          })
        })
      }
    }
    return response && response.data
  },
  (error) => {
    Message.error(error.response.data.msg)
    return Promise.reject(error)
  }
)

作者:zz
連結:https://juejin.cn/post/6983582201690456071
來源:掘金
著作權歸作者所有。商業轉載請聯絡作者獲得授權,非商業轉載請註明出處。