async/await 優雅的錯誤處理方法

try/catch
,像這樣
const fetchData = async () => { return new Promise((resolve, reject) => { setTimeout(() => { resolve('fetch data is me') }, 1000) }) } (async () => { try { const data = await fetchData() console.log('data is ->', data) } catch(err) { console.log('err is ->', err) } })() 複製程式碼
這麼看,感覺倒是沒什麼問題,如果是這樣呢?有多個非同步操作,需要對每個非同步返回的 error 錯誤狀態進行不同的處理,以下是示例程式碼
const fetchDataA = async () => { return new Promise((resolve, reject) => { setTimeout(() => { resolve('fetch data is A') }, 1000) }) } const fetchDataB = async () => { return new Promise((resolve, reject) => { setTimeout(() => { resolve('fetch data is B') }, 1000) }) } const fetchDataC = async () => { return new Promise((resolve, reject) => { setTimeout(() => { resolve('fetch data is C') }, 1000) }) } (async () => { try { const dataA = await fetchDataA() console.log('dataA is ->', dataA) } catch(err) { console.log('err is ->', err) } try { const dataB = await fetchDataB() console.log('dataB is ->', dataB) } catch(err) { console.log('err is ->', err) } try { const dataC = await fetchDataC() console.log('dataC is ->', dataC) } catch(err) { console.log('err is ->', err) } })() 複製程式碼
這樣寫程式碼裡充斥著 try/catch
,有程式碼潔癖的你能忍受的了嗎?這時可能會想到只用一個 try/catch
呢?
// ... 這裡 fetch 函式省略 (async () => { try { const dataA = await fetchDataA() console.log('dataA is ->', dataA) const dataB = await fetchDataB() console.log('dataB is ->', dataB) const dataC = await fetchDataC() console.log('dataC is ->', dataC) } catch(err) { console.log('err is ->', err) // 難道要定義 err 型別,然後判斷嗎?? /** * if (err.type === 'dataA') { *console.log('dataA err is', err) * } * ...... * */ } })() 複製程式碼
如果是這樣寫只會增加編碼的複雜度,而且要多寫程式碼。這個時候就應該想想怎麼優雅的解決, async/await
本質就是 promise
的語法糖,既然是 promise
那麼就可以使用 then
函數了
(async () => { const fetchData = async () => { return new Promise((resolve, reject) => { setTimeout(() => { resolve('fetch data is me') }, 1000) }) } const data = await fetchData().then(data => data ).catch(err => err) console.log(data) })() 複製程式碼
在上面寫法中,如果 fetchData 返回 resolve 正確結果時,data 是我們要的結果,如果是 reject 了,發生錯誤了,那麼 data 是錯誤結果,這顯然是行不通的,再對其完善。
(async () => { const fetchData = async () => { return new Promise((resolve, reject) => { setTimeout(() => { resolve('fetch data is me') }, 1000) }) } const [err, data] = await fetchData().then(data => [null, data] ).catch(err => [err, null]) console.log('err', err) console.log('data', data) // err null // data fetch data is me })() 複製程式碼
這樣是不是好很多了呢,但是問題又來了,不能每個 await 都寫這麼長,寫著也不方便也不優雅,再優化一下
(async () => { const fetchData = async () => { return new Promise((resolve, reject) => { setTimeout(() => { resolve('fetch data is me') }, 1000) }) } // 抽離成公共方法 const awaitWrap = (promise) => { return promise .then(data => [null, data]) .catch(err => [err, null]) } const [err, data] = await awaitWrap(fetchData()) console.log('err', err) console.log('data', data) // err null // data fetch data is me })() 複製程式碼
將對 await
處理的方法抽離成公共的方法,在使用 await
呼叫 awaitWrap
這樣的方法是不是更優雅了呢。如果使用 typescript 實現大概是這個樣子
function awaitWrap<T, U = any>(promise: Promise<T>): Promise<[U | null, T | null]> { return promise .then<[null, T]>((data: T) => [null, data]) .catch<[U, null]>(err => [err, null]) } 複製程式碼
以上。