前端如何實現token的無感重新整理

xiangzhihong 發表於 2021-10-23
前端

通常,對於一些需要記錄使用者行為的系統,在進行網路請求的時候都會要求傳遞一下登入的token。不過,為了介面資料的安全,伺服器的token一般不會設定太長,根據需要一般是1-7天的樣子,token過期後就需要重新登入。不過,頻繁的登入會造成體驗不好的問題,因此,需要體驗好的話,就需要定時去重新整理token,並替換之前的token。

要做到token的無感重新整理,主要有3種方案:

方案一:

後端返回過期時間,前端每次請求就判斷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

為了防止多次重新整理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正在重新整理,我們先將這個請求存到一個陣列佇列中,想辦法讓這個請求處於等待中,一直等到重新整理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)
  }
)