12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182 |
- "use strict";
- Object.defineProperty(exports, "__esModule", {
- value: true
- });
- exports.PipeTransport = void 0;
- var _utils = require("../utils");
- /**
- * Copyright (c) Microsoft Corporation.
- *
- * Licensed 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.
- */
- class PipeTransport {
- constructor(pipeWrite, pipeRead, closeable, endian = 'le') {
- this._pipeWrite = void 0;
- this._data = Buffer.from([]);
- this._waitForNextTask = (0, _utils.makeWaitForNextTask)();
- this._closed = false;
- this._bytesLeft = 0;
- this.onmessage = void 0;
- this.onclose = void 0;
- this._endian = void 0;
- this._closeableStream = void 0;
- this._pipeWrite = pipeWrite;
- this._endian = endian;
- this._closeableStream = closeable;
- pipeRead.on('data', buffer => this._dispatch(buffer));
- pipeRead.on('close', () => {
- this._closed = true;
- if (this.onclose) this.onclose();
- });
- this.onmessage = undefined;
- this.onclose = undefined;
- }
- send(message) {
- if (this._closed) throw new Error('Pipe has been closed');
- const data = Buffer.from(message, 'utf-8');
- const dataLength = Buffer.alloc(4);
- if (this._endian === 'be') dataLength.writeUInt32BE(data.length, 0);else dataLength.writeUInt32LE(data.length, 0);
- this._pipeWrite.write(dataLength);
- this._pipeWrite.write(data);
- }
- close() {
- // Let it throw.
- this._closeableStream.close();
- }
- _dispatch(buffer) {
- this._data = Buffer.concat([this._data, buffer]);
- while (true) {
- if (!this._bytesLeft && this._data.length < 4) {
- // Need more data.
- break;
- }
- if (!this._bytesLeft) {
- this._bytesLeft = this._endian === 'be' ? this._data.readUInt32BE(0) : this._data.readUInt32LE(0);
- this._data = this._data.slice(4);
- }
- if (!this._bytesLeft || this._data.length < this._bytesLeft) {
- // Need more data.
- break;
- }
- const message = this._data.slice(0, this._bytesLeft);
- this._data = this._data.slice(this._bytesLeft);
- this._bytesLeft = 0;
- this._waitForNextTask(() => {
- if (this.onmessage) this.onmessage(message.toString('utf-8'));
- });
- }
- }
- }
- exports.PipeTransport = PipeTransport;
|