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

Vincent.W發表於2019-01-24
AI

async/await 優雅的錯誤處理方法
一般情況下 async/await 在錯誤處理方面，主要使用 try/catch，像這樣

const fetchData = () => {
    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 = () => {
    return new Promise((resolve, reject) => {
        setTimeout(() => {
            resolve('fetch data is A')
        }, 1000)
    })
}

const fetchDataB = () => {
    return new Promise((resolve, reject) => {
        setTimeout(() => {
            resolve('fetch data is B')
        }, 1000)
    })
}

const fetchDataC = () => {
    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 = () => {
        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 = () => {
        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 = () => {
        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])
}
複製程式碼

以上。

相關文章