@stomp/stompjs/src/augment-websocket.ts

STOMP client for Javascript and Typescript

import { IStompSocket } from './types';

/**
 * @internal
 */
export function augmentWebsocket(
  webSocket: IStompSocket,
  debug: (msg: string) => void
) {
  webSocket.terminate = function () {
    const noOp = () => {};

    // set all callbacks to no op
    this.onerror = noOp;
    this.onmessage = noOp;
    this.onopen = noOp;

    const ts = new Date();

    const origOnClose = this.onclose;

    // Track delay in actual closure of the socket
    this.onclose = closeEvent => {
      const delay = new Date().getTime() - ts.getTime();
      debug(
        `Discarded socket closed after ${delay}ms, with code/reason: ${closeEvent.code}/${closeEvent.reason}`
      );
    };

    this.close();

    origOnClose.call(this, {
      code: 4001,
      reason: 'Heartbeat failure, discarding the socket',
      wasClean: false,
    });
  };
}