Spaces:
Running
Running
File size: 4,273 Bytes
78c921d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 |
"use strict";
// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements. See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership. The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the License for the
// specific language governing permissions and limitations
// under the License.
Object.defineProperty(exports, "__esModule", { value: true });
exports.toNodeStream = void 0;
const tslib_1 = require("tslib");
const stream_1 = require("stream");
const compat_js_1 = require("../../util/compat.js");
/** @ignore */
function toNodeStream(source, options) {
if ((0, compat_js_1.isAsyncIterable)(source)) {
return new AsyncIterableReadable(source[Symbol.asyncIterator](), options);
}
if ((0, compat_js_1.isIterable)(source)) {
return new IterableReadable(source[Symbol.iterator](), options);
}
/* istanbul ignore next */
throw new Error(`toNodeStream() must be called with an Iterable or AsyncIterable`);
}
exports.toNodeStream = toNodeStream;
/** @ignore */
class IterableReadable extends stream_1.Readable {
constructor(it, options) {
super(options);
this._iterator = it;
this._pulling = false;
this._bytesMode = !options || !options.objectMode;
}
_read(size) {
const it = this._iterator;
if (it && !this._pulling && (this._pulling = true)) {
this._pulling = this._pull(size, it);
}
}
_destroy(e, cb) {
const it = this._iterator;
let fn;
it && (fn = e != null && it.throw || it.return);
fn === null || fn === void 0 ? void 0 : fn.call(it, e);
cb && cb(null);
}
_pull(size, it) {
const bm = this._bytesMode;
let r = null;
while (this.readable && !(r = it.next(bm ? size : null)).done) {
if (size != null) {
size -= (bm && ArrayBuffer.isView(r.value) ? r.value.byteLength : 1);
}
if (!this.push(r.value) || size <= 0) {
break;
}
}
if (((r === null || r === void 0 ? void 0 : r.done) || !this.readable) && (this.push(null) || true)) {
it.return && it.return();
}
return !this.readable;
}
}
/** @ignore */
class AsyncIterableReadable extends stream_1.Readable {
constructor(it, options) {
super(options);
this._iterator = it;
this._pulling = false;
this._bytesMode = !options || !options.objectMode;
}
_read(size) {
const it = this._iterator;
if (it && !this._pulling && (this._pulling = true)) {
(() => tslib_1.__awaiter(this, void 0, void 0, function* () { return this._pulling = yield this._pull(size, it); }))();
}
}
_destroy(e, cb) {
const it = this._iterator;
let fn;
it && (fn = e != null && it.throw || it.return);
(fn === null || fn === void 0 ? void 0 : fn.call(it, e).then(() => cb && cb(null))) || (cb && cb(null));
}
_pull(size, it) {
return tslib_1.__awaiter(this, void 0, void 0, function* () {
const bm = this._bytesMode;
let r = null;
while (this.readable && !(r = yield it.next(bm ? size : null)).done) {
if (size != null) {
size -= (bm && ArrayBuffer.isView(r.value) ? r.value.byteLength : 1);
}
if (!this.push(r.value) || size <= 0) {
break;
}
}
if (((r === null || r === void 0 ? void 0 : r.done) || !this.readable) && (this.push(null) || true)) {
it.return && it.return();
}
return !this.readable;
});
}
}
//# sourceMappingURL=iterable.js.map
|