@litert/televoke 1.0.0-1 → 1.0.0-11

package/src/lib/transporters/legacy-http/LegacyHttp.Server.ts

@@ -27,7 +27,7 @@ const encoder = new v1.TvEncoderV1();
 const INVALID_REQUEST_RESPONSE = Buffer.from(encoder.encodeApiErrorResponse(
     'null',
     Shared.Encodings.v1.EResponseCode.MALFORMED_ARGUMENTS,
-    'INVALID REQUEST',
+    '"INVALID REQUEST"',
     0
 ));
 
@@ -76,6 +76,19 @@ class LegacyHttpGateway extends EventEmitter implements dT.IGateway {
             .setLegacyApiProcessor(this._onRequest);
     }
 
+    private _sendResponse(resp: Http.ServerResponse, data: string): void {
+
+        try {
+
+            resp.setHeader('content-length', Buffer.byteLength(data));
+            resp.end(data);
+        }
+        catch (e) {
+
+            this.emit('error', e);
+        }
+    }
+
     public get running(): boolean {
 
         return this._listener.running;
@@ -159,7 +172,6 @@ class LegacyHttpGateway extends EventEmitter implements dT.IGateway {
                     refuseBadRequest(resp);
                     this.emit('error', new Shared.errors.invalid_packet({
                         reason: 'invalid_json',
-                        data: buf,
                     }, e));
                     return;
                 }
@@ -169,7 +181,6 @@ class LegacyHttpGateway extends EventEmitter implements dT.IGateway {
                     refuseBadRequest(resp);
                     this.emit('error', new Shared.errors.invalid_packet({
                         reason: 'malformed_json',
-                        data: input,
                     }));
                     return;
                 }
@@ -185,48 +196,54 @@ class LegacyHttpGateway extends EventEmitter implements dT.IGateway {
 
                         if (result instanceof Shared.errors.app_error) {
 
-                            const data = encoder.encodeApiErrorResponse(
+                            this._sendResponse(resp, encoder.encodeApiErrorResponse(
                                 input.rid,
                                 v1.EResponseCode.FAILURE,
                                 result.message,
                                 recvAt
-                            );
-
-                            resp.setHeader('content-length', Buffer.byteLength(data));
-                            resp.end(data);
+                            ));
                         }
                         else if (result instanceof Shared.errors.api_not_found) {
 
-                            const data = encoder.encodeApiErrorResponse(
+                            this._sendResponse(resp, encoder.encodeApiErrorResponse(
                                 input.rid,
                                 v1.EResponseCode.API_NOT_FOUND,
                                 'null',
                                 recvAt
-                            );
-
-                            resp.setHeader('content-length', Buffer.byteLength(data));
-                            resp.end(data);
+                            ));
                         }
                         else {
 
-                            const data = encoder.encodeApiErrorResponse(
+                            this._sendResponse(resp, encoder.encodeApiErrorResponse(
                                 input.rid,
                                 v1.EResponseCode.SYSTEM_ERROR,
                                 'null',
                                 recvAt
-                            );
-
-                            resp.setHeader('content-length', Buffer.byteLength(data));
-                            resp.end(data);
+                            ));
                         }
 
                         return;
                     }
 
-                    const data = encoder.encodeApiOkResponse(input.rid, result ?? null, recvAt);
+                    let data: string;
+
+                    try {
+
+                        data = encoder.encodeApiOkResponse(input.rid, result ?? null, recvAt);
+                    }
+                    catch (e) {
+
+                        data = encoder.encodeApiErrorResponse(
+                            input.rid,
+                            v1.EResponseCode.SYSTEM_ERROR,
+                            'null',
+                            recvAt
+                        );
+
+                        this.emit('error', new Shared.errors.unprocessable_error({ api: input.api }, e));
+                    }
 
-                    resp.setHeader('content-length', Buffer.byteLength(data));
-                    resp.end(data);
+                    this._sendResponse(resp, data);
 
                 }, input.api, input.args, new LegacyHttpTransporter(req));
             });

package/src/lib/transporters/websocket/WebSocket.Server.ts

@@ -28,6 +28,7 @@ class WebSocketGateway extends EventEmitter implements dT.IGateway {
     public constructor(
         private readonly _listener: Listener.IHttpListener,
         private readonly _server: dT.IServer,
+        timeout?: number,
     ) {
 
         super();
@@ -38,6 +39,7 @@ class WebSocketGateway extends EventEmitter implements dT.IGateway {
 
         this._wsServer = LibWS.createServer({
             liteFrameMode: true,
+            timeout,
         });
     }
 
@@ -103,6 +105,18 @@ class WebSocketGateway extends EventEmitter implements dT.IGateway {
     }
 }
 
+export interface IWebSocketGatewayOptions {
+
+    /**
+     * The timeout of WebSocket connections.
+     *
+     * > Set to 0 to disable timeout.
+     *
+     * @default 60000
+     */
+    timeout?: number;
+}
+
 /**
  * Create a WebSocket gateway.
  *
@@ -112,7 +126,8 @@ class WebSocketGateway extends EventEmitter implements dT.IGateway {
 export function createWebsocketGateway(
     listener: Listener.IHttpListener,
     server: dT.IServer,
+    options: IWebSocketGatewayOptions = {}
 ): dT.IGateway {
 
-    return new WebSocketGateway(listener, server);
+    return new WebSocketGateway(listener, server, options.timeout);
 }