Question

我有一个咖喱的批处理函数，它返回一个Iterable。如果使用同步Iterable进行调用，则会获得同步Iterable，如果给出AsyncIterable，则会得到AsyncIterable。但是我无法终生获得函数实现所接受的同步重载。

这个特殊的定义导致了错误

export function batch(size: number): <T>(curriedIterable: Iterable<T>) => IterableIterator<T[]>

这是所有代码

async function* _batch<T>(size: number, iterable: AsyncIterable<T>) {
  let dataBatch: T[] = []
  for await (const data of iterable) {
    dataBatch.push(data)
    if (dataBatch.length === size) {
      yield dataBatch
      dataBatch = []
    }
  }
  if (dataBatch.length > 0) {
    yield dataBatch
  }
}

function* _syncBatch<T>(size: number, iterable: Iterable<T>) {
  let dataBatch: T[] = []
  for (const data of iterable) {
    dataBatch.push(data)
    if (dataBatch.length === size) {
      yield dataBatch
      dataBatch = []
    }
  }
  if (dataBatch.length > 0) {
    yield dataBatch
  }
}

export function batch(size: number): <T>(curriedIterable: AsyncIterable<T>) => AsyncIterableIterator<T[]>
export function batch<T>(size: number, iterable: AsyncIterable<T>): AsyncIterableIterator<T[]>
export function batch(size: number): <T>(curriedIterable: Iterable<T>) => IterableIterator<T[]>
export function batch<T>(size: number, iterable: Iterable<T>): IterableIterator<T[]>
export function batch<T>(size: number, iterable?: Iterable<T> | AsyncIterable<T>) {
  if (iterable === undefined) {
    return curriedIterable => batch(size, curriedIterable)
  }
  if (iterable[Symbol.asyncIterator]) {
    return _batch(size, iterable as AsyncIterable<T>)
  }
  return _syncBatch(size, iterable as Iterable<T>)
}

Answer 1

当我需要重载的def作为返回函数时，我对function batch(size: number):有重复的定义。可以解决以下问题。

export function batch(size: number): {
  <T>(curriedIterable: AsyncIterable<T>): AsyncIterableIterator<T[]>
  <T>(curriedIterable: Iterable<T>): IterableIterator<T[]>
}
export function batch<T>(size: number, iterable: AsyncIterable<T>): AsyncIterableIterator<T[]>
export function batch<T>(size: number, iterable: Iterable<T>): IterableIterator<T[]>

这样可以输入。

按符号过滤的咖喱函数无法在Typescript中获得兼容的函数实现

1 个答案: