Как изменить простой цикл ожидания? - PullRequest
Новая
       36

Как изменить простой цикл ожидания?

0 голосов
/ 15 сентября 2018

Я играю с некоторыми новыми функциями JavaScript, такими как async / await и генераторы.У меня есть функция readPages с подписью 

async function* readPages(....): AsyncIterableIterator<string> {}

, и я хочу объединить результат этой функции с некоторым разделителем.Вот как я это делаю сейчас 

let array = new Array<string>();

for await (const page of readPages(...))
    array.push(page);

let result = array.join(pagesDelimiter);

Думаю, это довольно многословно.Можно ли сделать это лучше?

Вот полный код для справки 

import * as fs from 'fs';
import { PDFJSStatic, PDFDocumentProxy } from 'pdfjs-dist';
const PDFJS: PDFJSStatic = require('pdfjs-dist');
PDFJS.disableWorker = true;

async function* readPages(doc: PDFDocumentProxy, wordsDelimiter = '\t'): AsyncIterableIterator<string> {
    for (let i = 1; i <= doc.numPages; i++) {
        const page = await doc.getPage(i);
        const textContent = await page.getTextContent();
        yield textContent.items.map(item => item.str).join(wordsDelimiter);
    }
}

async function pdfToText(filename: string, pagesDelimiter = '\n', wordsDelimiter = '\t') {
    const data = new Uint8Array(fs.readFileSync(filename));
    const doc = await PDFJS.getDocument(data);

    const array = new Array<string>();

    for await (const page of readPages(doc, wordsDelimiter))
        array.push(page);

    return array.join(pagesDelimiter);
}

pdfToText('input.pdf').then(console.log);

1 Ответ

0 голосов
/ 15 сентября 2018

ОК, я немного больше играю с этим кодом и думаю, что в настоящее время невозможно справиться с этой задачей лучше, чем с циклом for-await-of. Но , вы можете скрыть этот цикл за прототипом функции ... 

declare global {
    interface AsyncIterableIterator<T> {
        toPromise(): Promise<T[]>;
    }
}

(async function* (): any {})().constructor.prototype.toPromise = async function<T>(this: AsyncIterableIterator<T>): Promise<T[]> {
    let result = new Array<T>();

    for await (const item of this)
        result.push(item);

    return result;
};

так мой код 

const array = new Array<string>();

for await (const page of readPages(...))
    array.push(page);

const result = array.join(pagesDelimiter);

становится 

const array = await readPages(...).toPromise();
const result = array.join(pagesDelimiter);

Да, и я знаю, что прототипирование сомнительно. Но было интересно, как создать прототип асинхронного итератора: -).

...