前端如何實現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)
  }
)

相關文章