113 lines
3.4 KiB
JavaScript
113 lines
3.4 KiB
JavaScript
"use strict";
|
|
|
|
Object.defineProperty(exports, "__esModule", {
|
|
value: true
|
|
});
|
|
exports.WebSocketTransport = void 0;
|
|
|
|
var _ws = _interopRequireDefault(require("ws"));
|
|
|
|
var _utils = require("../utils/utils");
|
|
|
|
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
|
|
|
|
/**
|
|
* Copyright 2018 Google Inc. All rights reserved.
|
|
* Modifications 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 WebSocketTransport {
|
|
static async connect(progress, url, headers) {
|
|
progress.log(`<ws connecting> ${url}`);
|
|
const transport = new WebSocketTransport(progress, url, headers);
|
|
let success = false;
|
|
progress.cleanupWhenAborted(async () => {
|
|
if (!success) await transport.closeAndWait().catch(e => null);
|
|
});
|
|
await new Promise((fulfill, reject) => {
|
|
transport._ws.addEventListener('open', async () => {
|
|
progress.log(`<ws connected> ${url}`);
|
|
fulfill(transport);
|
|
});
|
|
|
|
transport._ws.addEventListener('error', event => {
|
|
progress.log(`<ws connect error> ${url} ${event.message}`);
|
|
reject(new Error('WebSocket error: ' + event.message));
|
|
|
|
transport._ws.close();
|
|
});
|
|
});
|
|
success = true;
|
|
return transport;
|
|
}
|
|
|
|
constructor(progress, url, headers) {
|
|
this._ws = void 0;
|
|
this._progress = void 0;
|
|
this.onmessage = void 0;
|
|
this.onclose = void 0;
|
|
this.wsEndpoint = void 0;
|
|
this.wsEndpoint = url;
|
|
this._ws = new _ws.default(url, [], {
|
|
perMessageDeflate: false,
|
|
maxPayload: 256 * 1024 * 1024,
|
|
// 256Mb,
|
|
handshakeTimeout: progress.timeUntilDeadline(),
|
|
headers
|
|
});
|
|
this._progress = progress; // The 'ws' module in node sometimes sends us multiple messages in a single task.
|
|
// In Web, all IO callbacks (e.g. WebSocket callbacks)
|
|
// are dispatched into separate tasks, so there's no need
|
|
// to do anything extra.
|
|
|
|
const messageWrap = (0, _utils.makeWaitForNextTask)();
|
|
|
|
this._ws.addEventListener('message', event => {
|
|
messageWrap(() => {
|
|
try {
|
|
if (this.onmessage) this.onmessage.call(null, JSON.parse(event.data));
|
|
} catch (e) {
|
|
this._ws.close();
|
|
}
|
|
});
|
|
});
|
|
|
|
this._ws.addEventListener('close', event => {
|
|
this._progress && this._progress.log(`<ws disconnected> ${url}`);
|
|
if (this.onclose) this.onclose.call(null);
|
|
}); // Prevent Error: read ECONNRESET.
|
|
|
|
|
|
this._ws.addEventListener('error', () => {});
|
|
}
|
|
|
|
send(message) {
|
|
this._ws.send(JSON.stringify(message));
|
|
}
|
|
|
|
close() {
|
|
this._progress && this._progress.log(`<ws disconnecting> ${this._ws.url}`);
|
|
|
|
this._ws.close();
|
|
}
|
|
|
|
async closeAndWait() {
|
|
const promise = new Promise(f => this._ws.once('close', f));
|
|
this.close();
|
|
await promise; // Make sure to await the actual disconnect.
|
|
}
|
|
|
|
}
|
|
|
|
exports.WebSocketTransport = WebSocketTransport; |