Question

我有一个通过回调批量传输数据的功能。

每个批次将在获取另一个批次之前等待回调函数，整个函数返回一个在所有批次完成时解析的承诺。

（我正在使用TypeScript注释来帮助提高可读性）

async function callbackStream(fn: (batch: Array<number>) => Promise<void>) {}

如何将此功能转换为异步生成器，一次产生一个值？

async function* generatorStream(): AsyncIterableIterator<number> {}

事实证明这是一项非常艰巨的任务。

我已经玩弄了这个问题并且我已经构建了一些有用的东西，但它非常复杂，我无法证明合并这些代码并让我的团队中的其他人处理它。

这是我目前的实施：

我正在使用这个辅助函数创建了一个“延迟”的承诺，这有助于在回调周围传递承诺。

interface DeferredPromise<T> {
    resolve: (value: T) => void
    reject: (error: any) => void
    promise: Promise<T>
}

function deferred<T>(): DeferredPromise<T> {
    let resolve
    let reject
    const promise = new Promise<T>((res, rej) => {
        resolve = res
        reject = rej
    })
    return {
        resolve: resolve as (value: T) => void,
        reject: reject as (error: any) => void,
        promise,
    }
}

接下来，我有了这个逻辑毛球，将promise回调线性化为一个链，其中每个promise解决了具有下一个函数的批处理，该函数将返回另一个获取下一批的promise。

type Done = { done: true }
type More = { done: false; value: Array<number>; next: () => Promise<Result> }
type Result = More | Done

async function chainedPromises() {
    let deferred = PromiseUtils.deferred<Result>()

    callbackStream(async batch => {
        const next = PromiseUtils.deferred<null>()
        deferred.resolve({
            done: false,
            value: batch,
            next: () => {
                deferred = PromiseUtils.deferred<Result>()
                next.resolve(null)
                return deferred.promise
            },
        })
        await next.promise
    }).then(() => {
        deferred.resolve({ done: true })
    })

    return deferred.promise
}

从这里开始，创建一次生成一个项目的生成器并不是很困难：

async function* generatorStream(): AsyncIterableIterator<number> {
    let next = chainedPromises
    while (true) {
        const result = await next()
        if (result.done) {
            return
        }
        for (const item of result.value) {
            yield item
        }
        next = result.next
    }
}

我认为我们都同意中间chainedPromises功能非常令人困惑和错综复杂。 有什么办法可以让callbackStream以一种易于理解和易于理解的方式转换为generatorStream吗？我不介意使用图书馆，但我也很欣赏第一原则的简单实现。

Answer 1

不，我认为没有办法以易于理解和易于理解的方式实现这种转变。但是，我建议删除deferred s（你永远不会reject）并只使用promise构造函数。另外，我宁愿立即实现异步生成器。

function queue() {
    let resolve = () => {};
    const q = {
        put() {
            resolve();
            q.promise = new Promise(r => { resolve = r; });
        },
        promise: null,
    }
    q.put(); // generate first promise
    return q;
}
function toAsyncIterator(callbackStream) {
    const query = queue();
    const result = queue();
    const end = callbackStream(batch => {
        result.put(batch);
        return query.promise;
    }).then(value => ({value, done: true}));
    end.catch(e => void e); // prevent unhandled promise rejection warnings
    return {
        [Symbol.asyncIterator]() { return this; },
        next(x) {
            query.put(x);
            return Promise.race([
                end,
                result.promise.then(value => ({value, done:false})
            ]);
        }
    }
}
async function* batchToAsyncIterator(batchCallbackStream) {
    for await (const batch of toAsyncIterator(batchCallbackStream)) {
        // for (const val of batch) yield val;
        // or simpler:
        yield* batch;
    }
}

Answer 2

您需要一个事件存储桶，下面是一个示例：

function bucket() {
  const stack = [],
    iterate = bucket();

  var next;

  async function* bucket() {
    while (true) {
      yield new Promise((res) => {
        if (stack.length > 0) {
          return res(stack.shift());
        }

        next = res;
      });
    }
  }

  iterate.push = (itm) => {
    if (next) {
      next(itm);
      next = false;
      return;
    }

    stack.push(itm);
  }

  return iterate;
}

;
(async function() {
  let evts = new bucket();

  setInterval(() => {
    evts.push(Date.now());
    evts.push(Date.now() + '++');
  }, 1000);

  for await (let evt of evts) {
    console.log(evt);
  }
})();

Answer 3

如果有打字稿解决方案会行得通吗？

当回调被更快地调用时，它应该处理条件，然后两次解决promise。回调可以是具有此签名callback(error, result, index)的方法设置为在不带参数的情况下调用回调时完成。用法：

asAsyncOf(this.storage, this.storage.each);

解决方案：

function asAsyncOf<T1, T2, T3, T4, Y>(c, fn: { (a: T1, a1: T2, a2: T3, a3: T4, cb: { (err?, res?: Y, index?: number): boolean }): void }, a: T1, a1: T2, a2: T3, a3: T4): AsyncGenerator<Y>
function asAsyncOf<T1, T2, T3, Y>(c, fn: { (a: T1, a1: T2, a2: T3, cb: { (err?, res?: Y, index?: number): boolean }): void }, a: T1, a1: T2, a3: T3): AsyncGenerator<Y>
function asAsyncOf<T1, T2, Y>(c, fn: { (a: T1, a1: T2, cb: {(err?, res?: Y, index?: number): boolean}): void}, a: T1, a1: T2): AsyncGenerator<Y>
function asAsyncOf<T, Y>(c, fn: { (a: T, cb: { (err?, res?: Y, index?: number): boolean }): void }, a: T): AsyncGenerator<Y>
function asAsyncOf<Y>(c, fn: { (cb: {(err?, res?: Y, index?: number): boolean}): void}): AsyncGenerator<Y>
async function* asAsyncOf(context, fn, ...args) {
    let next = (result?) => { };
    let fail = (err) => { };
    let finish = {};
    const items = [];
    let started = true;
    try {
        fn.apply(context, [...args, function (err, result, index) {
            const nextArgs = [].slice.call(arguments, 0);
            if (nextArgs.length === 0) {
                started = false;
                next(finish);
                return true;
            }
            if (err) {
                fail(err);
                return true;
            }
            items.push(result);
            next(result);
        }]);
    } catch (ex) {
        fail(ex);
    }
    while (true) {
        const promise = started ? new Promise((resolve, error) => {
            next = resolve;
            fail = error;
        }) : Promise.resolve(finish);
        const record = await promise;
        if (record === finish) {
            while (items.length) {
                const item = items.shift();
                yield item;
            }
            return;
        }
        while (items.length) {
            const item = items.shift();
            yield item;
        }
    }
}
export { asAsyncOf };

如何将Node.js异步流回调转换为异步生成器？

3 个答案: