高阶函数的使用

问题

字节跳动面试时问题：原函数例如fetchData是一个异步函数，尝试从服务器端获取一些信息并返回一个Promise。写一个新的函数可以自动重试一定次数，并且在使用上和原函数没有区别。

思路

这个问题其实不是很难，不过可能是太菜了紧张的原因，当时答得不是很好。不过思路还是很明确的，内部通过闭包来计数，一旦成功获得数据就返回，否则就继续尝试，直到重试次数达到上限位置。

function retry(fetch, n) {
    let i = 0
    function tryFetch(err) {
        if (i > n) {
            return "reach try limit:" + err
        } else {
            fetch().then(data => {
                return data
            }).catch(err => {
                i ++
                tryFetch(err)
            })
        }
    }
}

当时差不多就是那么答的，有几个问题是，函数调用方式与原来不通，按道理应该返回一个Promise，更准确的说是返回一个返回Promise的函数，如果有什么问题也应该在外面能catch住。另一方面，也许fetch函数要接受参数，也应该传递进去才行。

解决

修改后的函数如下

function retry(fetch, n) {
    return function() {
        let args = arguments
        return new Promise((rseolve, reject) => {
            let i = 0
            tryFetch()
            function tryFetch(err) {
                console.log(i)
                if (i > n) {
                    reject("reach max try" + err)
                } else {
                    fetch(...args).then(data => {
                        rseolve(data)
                    }).catch(err => {
                        i ++
                        tryFetch(err)
                    })
                }
            }
        })
    }
}

最后自己写了个fetch进行测试

function fetch() {
    console.log(arguments)
    return new Promise((rseolve, reject) => {
        console.log('trying...')
        setTimeout(function() {
            if (Math.random() > 0.9) {
                console.log('resolved')
                rseolve('resolved')
            } else {
                console.log('rejected')
                reject('rejected')
            }
        }, 5000)
    })
}

结果符合预期，问题解决。当然也可以返回async function，不过和Promise本质上是一个思路。