@litert/televoke
Version:
A simple RPC service framework.
106 lines • 3.9 kB
JavaScript
;
/**
* Copyright 2025 Angus.Fenying <fenying@litert.org>
*
* 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
*
* https://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.createWebsocketGateway = createWebsocketGateway;
exports.createCustomWebsocketGateway = createCustomWebsocketGateway;
const LibWS = require("@litert/websocket");
const dWS = require("./WebSocket.decl");
const node_events_1 = require("node:events");
const WebSocket_Transporter_1 = require("./WebSocket.Transporter");
class WebSocketGateway extends node_events_1.EventEmitter {
constructor(registerListener, _server, timeout) {
super();
this._server = _server;
this._onUpgrade = (req, socket) => {
let apn = req.headers[LibWS.H1_HDR_NAME_WS_PROTOCOL];
if (!apn) {
this._wsServer.reject({
'request': req,
'socket': socket,
});
return;
}
if (!Array.isArray(apn)) {
apn = [apn];
}
if (!apn.includes(dWS.WEBSOCKET_SUB_PROTOCOL)) {
this._wsServer.reject({
'request': req,
'socket': socket,
});
return;
}
const ws = this._wsServer.accept({
'request': req,
'socket': socket,
'subProtocol': dWS.WEBSOCKET_SUB_PROTOCOL,
});
this._server.registerChannel(new WebSocket_Transporter_1.WebSocketTransporter(dWS.WS_PROTOCOL_NAME, ws));
};
this._listener = registerListener({
'onErrorCallback': (e) => this.emit('error', e),
'onUpgradeCallback': this._onUpgrade,
});
this._wsServer = LibWS.createServer({
frameReceiveMode: LibWS.EFrameReceiveMode.LITE,
timeout,
});
}
get running() {
return this._listener.running;
}
async start() {
if (this.running) {
return;
}
await this._listener.start?.();
}
async stop() {
if (!this.running) {
return;
}
await this._listener.stop?.();
}
}
/**
* Create a WebSocket gateway, binding to a simple built-in HTTP(S) listener.
*
* > When using built-in HTTP server, the api will ignore other headers, path and query string in the URL.
*
* @param listener The HTTP(S) listener object.
* @param server The televoke server object.
*/
function createWebsocketGateway(listener, server, options = {}) {
return new WebSocketGateway((o) => {
listener.on('error', o.onErrorCallback);
listener.setUpgradeProcessor(o.onUpgradeCallback);
return listener;
}, server, options.timeout);
}
/**
* Create a WebSocket gateway, binding to a custom HTTP server.
*
* > When using a custom HTTP server, it's able to preprocess the request before passing to the server, like
* > authentication, rate limiting, custom routing, etc.
*
* @param registerListener The function to register the listener to the custom HTTP server.
* @param server The server to process the requests.
*/
function createCustomWebsocketGateway(registerListener, server) {
return new WebSocketGateway(registerListener, server);
}
//# sourceMappingURL=WebSocket.Server.js.map