asyncToGenerator 实现

摘自 blog.csdn.net/weixin_33826897/arti...

const getData = () => new Promise(resolve => setTimeout(() => resolve("data"), 1000))

function * testG() {

    // await被编译成了

    const data = yield getData()

    console.log('data: ', data);  

    const data2 = yield getData()  

    console.log('data2: ', data2);  

    return 'success'

}

function asyncToGenerator(generatorFunc) {  

    // 返回的是一个新的函数

      return function() {

        // 先调用generator函数 生成迭代器

        // 对应 var gen = testG()

        const gen = generatorFunc.apply(this, arguments)

        // 返回一个promise 因为外部是用.then的方式 或者await的方式去使用这个函数的返回值的

        // var test = asyncToGenerator(testG)

        // test().then(res => console.log(res))    

        return new Promise((resolve, reject) => {          

            // 内部定义一个step函数 用来一步一步的跨过yield的阻碍      

            // key有next和throw两种取值，分别对应了gen的next和throw方法      

            // arg参数则是用来把promise resolve出来的值交给下一个yield      

            function step(key, arg) {        

                let generatorResult                

                // 这个方法需要包裹在try catch中        

                // 如果报错了 就把promise给reject掉 外部通过.catch可以获取到错误        

                try {          

                    generatorResult = gen[key](arg)        

                } catch (error) {          

                    return reject(error)        

                }        

                // gen.next() 得到的结果是一个 { value, done } 的结构        

                const { value, done } = generatorResult        

                if (done) {

                    // 如果已经完成了 就直接resolve这个promise          

                    // 这个done是在最后一次调用next后才会为true          

                    // 以本文的例子来说 此时的结果是 { done: true, value: 'success' }          

                    // 这个value也就是generator函数最后的返回值          

                    return resolve(value)        

                } else {

                    // 除了最后结束的时候外，每次调用gen.next()          

                    // 其实是返回 { value: Promise, done: false } 的结构，          

                    // 这里要注意的是Promise.resolve可以接受一个promise为参数          

                    // 并且这个promise参数被resolve的时候，这个then才会被调用          

                    return Promise.resolve(     

                        // 这个value对应的是yield后面的promise            

                        value          

                        ).then(            

                            // value这个promise被resove的时候，就会执行next            

                            // 并且只要done不是true的时候 就会递归的往下解开promise            

                            // 对应

                            //  gen.next().value.then(value => {            

                            //    gen.next(value).value.then(value2 => {            

                            //       gen.next()            

                            //            

                            //      

                            // 此时done为true了 整个promise被resolve了           

                            //      

                            // 最外部的test().then(res => console.log(res))的then就开始执行了            

                            //    })            

                            // })           

                        function onResolve(val) {              

                            step("next", val)      

                        },            

                        // 如果promise被reject了 就再次进入step函数            

                        // 不同的是，这次的try catch中调用的是gen.throw(err)            

                        // 那么自然就被catch到 然后把promise给reject掉啦            

                        function onReject(err) {              

                            step("throw", err)            

                        },)        

                    }      

                }      

                step("next")

            })  

        }}

var test = asyncToGenerator(testG)

test().then((res) => {

    console.log(res);

})

本作品采用《CC 协议》，转载必须注明作者和本文链接