ws.ts公司

import { IWSClient } from "./types/IChannel";

export default class WSHandler {
  sockets: Object = {};

  bindChannelWS(channel: any, client: IWSClient) {
    if (this.sockets[channel] === undefined) {
      this.sockets[channel] = [client];
    } else {
      this.sockets[channel].push(client);
    }
  }

  publishToChannel(channel: string, message: string) {
    if (this.sockets[channel] !== undefined) {
      this.sockets[channel].forEach(function (client: IWSClient) {
        console.log(JSON.parse(message));
        client.send(message);
      });
    }
  }
}

索引.ts

export {};
const redis = require("redis");
const WebSocket = require("ws");
import { IWSClient } from "./types/IChannel";
import WSHandler from "./ws";

const subscriber = redis.createClient();
subscriber.psubscribe("*:*:*");
const wss = new WebSocket.Server({ port: 8080 });

var wsHandler = new WSHandler();

subscriber.on("pmessage", function (
  pattern: any,
  channel: string,
  message: string
) {
  wsHandler.publishToChannel(channel, message);
});

wss.on("connection", function connection(ws: IWSClient) {
  ws.on("message", function incoming(_subMessages: string) {
    let subMessages: Array<string> = JSON.parse(_subMessages);
    console.log("Server Received: ", subMessages);
    subMessages.forEach((channel) => wsHandler.bindChannelWS(channel, ws));
  });

  ws.send(JSON.stringify({ connection: "Initiated" }));
});