Add BrokerOptions to customize Heartbeat interval and message

src/broker.ts

@@ -1,10 +1,14 @@
 import WebSocket from 'ws';
 
+export interface BrokerOptions {
+  heartbeatInterval?: number; // in milliseconds
+  heartbeatMessage?: string;
+}
 export class Broker {
   private subsByWsId = new Map<string, Subscription[]>();
   private subsByChannel = new Map<string, Subscription>();
 
-  constructor() {
+  constructor(private options: BrokerOptions = {}) {
     this.heartbeat();
   }
 
@@ -67,13 +71,13 @@ export class Broker {
     for (const subs of this.subsByWsId.values()) {
       for (const sub of subs) {
         try {
-          sub.send('HEARTBEAT');
+          sub.send(this.options.heartbeatMessage || 'HEARTBEAT');
         } catch(err) {
           // continue
         }
       }
     }
-    setTimeout(() => this.heartbeat(), 120_000);
+    setTimeout(() => this.heartbeat(), this.options.heartbeatInterval || 120_000);
   }
 }
 

src/index.ts

@@ -1,11 +1,11 @@
 import http from 'http';
 import WebSocket from 'ws';
 import { connect, upgrade } from './webSocket';
-import { Broker } from './broker';
+import { Broker, BrokerOptions } from './broker';
 import { consoleLogger, Logger } from './logger';
 import * as net from 'net';
 
-export function createBroker<T extends http.IncomingMessage>(server: PubSubServer<T>, getChannel: (request: T) => Promise<string>, logger: Logger = consoleLogger): Broker {
+export function createBroker<T extends http.IncomingMessage>(server: PubSubServer<T>, getChannel: (request: T) => Promise<string>, logger: Logger = consoleLogger, options: BrokerOptions = {}): Broker {
   const wss = new WebSocket.Server({ noServer: true });
   const broker = new Broker();