serverless-offline/src/events/websocket/http-routes/connections/ConnectionsController.js

Emulate AWS λ and API Gateway locally when developing your Serverless project

export default class ConnectionsController {
  #webSocketClients = null

  constructor(webSocketClients) {
    this.#webSocketClients = webSocketClients
  }

  send(connectionId, payload) {
    // TODO, is this correct?
    if (!payload) {
      return null
    }

    const clientExisted = this.#webSocketClients.send(
      connectionId,
      // payload is a Buffer
      payload.toString("utf8"),
    )

    return clientExisted
  }

  remove(connectionId) {
    const clientExisted = this.#webSocketClients.close(connectionId)

    return clientExisted
  }
}