递归异步生成器函数的返回类型

Question

定义以下递归异步函数生成器的正确类型是什么？

async function* recursive_f(n: number): AsyncGenerator<number, ?, ?> {
    n++
    if(n>0) {
        yield n
        return await recursive_f(n)
    }

    return Promise.resolve()
}

以下代码未按预期工作。它应该递归地减少 n 并产生一次负数，而不是通过 if 条件并结束。

async function* recursive_f(n: number): AsyncGenerator<number> {
    
    if (n < 0) {
        yield n
    }

    return await recursive_f(n--) // 'await' has no effect on the type of this expression.
}

async function main_function() {
    console.log(`main_function`)
    let num = 10
    for await (const n of recursive_f(num)) {
        console.log(n)
    }
}

main_function().then(console.log, console.error)

Answer 1

如果没有包含用例的 minimal reproducible example，我无法判断以下内容是否适合您的需求，因此请检查一下。

---

您可以像这样定义recursive type alias：

type RecursiveF = AsyncGenerator<number, void | RecursiveF>;

这意味着：RecursiveF 是一个异步生成器函数，它产生 number，并返回 void 或 RecursiveF。在实践中这似乎很难使用，因为 void 类型的值不能真正被检查。但这取决于您和您的用例。

不管怎样，现在你可以用它来注释recursiveF的返回类型了：

async function* recursive_f(n: number): RecursiveF {
  n++
  if (n > 0) {
    yield n
    return recursive_f(n)
  }

  return Promise.resolve()
}

这在编译器同意的意义上是有效的。能不能满足你的需求，你自己看看。

Playground link to code