链接承诺在webdriver.io中的while循环

时间:2016-03-25 23:05:35

标签: javascript selenium promise es6-promise webdriver-io

我希望通过捕获视口大小的切片来获取完整网页的屏幕截图。它几乎完成了,但我对承诺很新,我正在寻找正确的方法。

这是我的代码。问题是调用client.execute(...)。然后(...)不会在循环迭代之间等待它自己。最后的结局'两者都没有等到之前的那个'那就是为什么它被注释掉了。

...
var client = webdriverio.remote(options);
...
client    
  ...
  .then(function() {

    var yTile = 0;
    var heightCaptured = 0;

    while(heightCaptured < documentSize.height) {
      var tileFile  = 'screenshot-' + yTile + '.png';

      client
      .execute(function(heightCaptured) {
        window.scrollTo(0, heightCaptured);
      }, heightCaptured)
      .then(function() {
        console.log('captured: ' + tileFile);
        client.saveScreenshot('./' + tileFile);

        return client;
      });

      heightCaptured += viewportSize.height;
      yTile++;
    }

  })
  //.client.end()
  ;

在这种情况下使用promises的正确方法是什么?

感谢。

1 个答案:

答案 0 :(得分:5)

您不能使用while外观链接不确定数量的异步操作,因为while循环将立即运行完成,但您需要在每次异步执行后做出循环决策。

相反,您可以创建一个内部函数next(),它返回一个promise并重复调用它,将每个函数链接到上一个,直到完成并在循环内决定是否通过返回将另一个调用链接到next()它位于先前的.then()处理程序中,或者您可以通过返回常规值(而非承诺)来结束链。

...
var client = webdriverio.remote(options);
...
client
    ...
    .then(function () {
        var yTile = 0;
        var heightCaptured = 0;

        function next() {
            if (heightCaptured < documentSize.height) {
                var tileFile = 'screenshot-' + yTile + '.png';

                // return promise to chain it automatically to prior promise
                return client.execute(function (heightCaptured) {
                    window.scrollTo(0, heightCaptured);
                }, heightCaptured).then(function () {
                    console.log('captured: ' + tileFile);

                    // increment state variables
                    heightCaptured += viewportSize.height;
                    yTile++;

                    // return this promise to so it is also chained properly
                    // when this is done, call next again in the .then() handler
                    return client.saveScreenshot('./' + tileFile).then(next);
                });

            } else {
                // Done now, end the promise chain by returning a final value
                // Might also consider returning yTile so the caller knows
                // how many screen shots were saved
                return client;
            }
        }
        // start the loop
        return next();
    }).then(function () {
        // done here
    }, function (err) {
        // error here
    });

作为参考,如果您在.then()处理程序中并且从.then()处理程序返回一个promise,那么该promise将链接到先前的promise。如果您返回一个值,那么promise链就会终止,并且该值将作为整个链的最终解析值返回。

因此,在此示例中,由于next()会返回一个承诺,您可以在return next();处理程序中重复调用.then(),这会将所有屏幕截图链接到一个顺序链中,直到你最终只返回一个值,而不是一个承诺,这将结束链。