解决承诺后返回数组

Question

我正在尝试解决我的问题，但我没有任何想法。 我想创建函数来根据类型变量返回值数组。

我想检查发送给函数的参数是字符串数组还是webelements数组，我想将此数组返回给其他函数进行排序。

这是我的代码：

  private static returnByType(unSortedElements) {
      var i = 0;
      var unSorted = [];
      if (typeof unSortedElements[0] === 'string') {
          unSorted = unSortedElements;
          return unSorted;
      } else {
          return unSortedElements.each((eachName) => {
              eachName.getText().then((name) => {
                  unSorted[i] = name;
                  i++;
              })
          })

      }

  }

如果我将字符串数组传递给这个函数，它会返回带有这个字符串的数组，但是当我传递一个 webelements 列表时，我得到一个空数组（在 else 块中）。 我应该如何从 else 块返回 unSorted？

Answer 1

如果你在处理promise，你不能只返回值，它永远是一个promise。你的代码应该是这样的：

private static returnByType(unSortedElements) {
  var i = 0;
  var unSorted = [];
  if (typeof unSortedElements[0] === 'string') {
      unSorted = unSortedElements;
      // Promise.resolve(unSorted) to unify with else
      return Promise.resolve(unSorted);
  } else {
      // return type will be Promise<[]>
      return Promise.all(unSortedElements.map((eachName) => eachName.getText()));
  }

}

函数的返回类型现在将是Promise<[]>。然后你可以像这样使用它：

returnByType(args).then((unsorted) => ...);

• Promise.all() 参考documentation。
• Promise.resolve() 参考documentation。