Как изменить простой цикл ожидания?
Я играю с некоторыми новыми функциями 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 ответ
Хорошо, я играю с этим кодом немного больше, и я думаю, что в настоящее время невозможно справиться с этой задачей лучше, чем с 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);
Да, и я знаю, что прототипирование сомнительно. Но было интересно, как создать прототип асинхронного итератора:-).