@forchange/mock/lib/socket/index.js

const Koa = require("koa");
const path = require("path");
const EventEmitter = require("events");

const route = require("koa-route");
const views = require("koa-views");
const websocket = require("koa-websocket");

const parseSocket = require("./parse/socket");
const cors = require('./../util/cors')

class HttpEmitter extends EventEmitter {}

var socketList = {};
let socketMap;

function emitterSocketEvent(route, message) {
  if (socketList[route]) {
    let routeList = socketList[route];
    routeList.map(item => {
      item.httpEmitter.emit("sendMessage", JSON.stringify(message));
    });
  }
}

function emitterSocketClose(route) {
  if (socketList[route]) {
    let routeList = socketList[route];
    routeList.map(item => {
      item.httpEmitter.emit("close", route);
    });
    delete socketList[route];
  }
}

function socketServer(socketConfig) {
  const port = socketConfig.port;
  const host = socketConfig.host;
  const directory = socketConfig.directory;
  const app = websocket(new Koa());

  if (directory) {
    socketMap = parseSocket(directory);
  }

  app.on("error", err => {
    console.log("出现了错误   ", err);
  });

  app.use(
    views(path.resolve(__dirname), {
      extension: "html"
    })
  );

  app.use(cors)

  app.use(
    route.get("/", async ctx => {
      await ctx.render("index", {
        socketList
      });
    })
  );

  app.use(
    route.get("/mock/sendMessage", async ctx => {
      // todo 触发http事件
      emitterSocketEvent(ctx.query.url, ctx.query.message);
      ctx.response.status = 200;
      ctx.response.body = {
        code: 0
      };
    })
  );

  app.use(
    route.get("/mock/closeSocket", async ctx => {
      // todo 触发http事件
      emitterSocketClose(ctx.query.url);
      ctx.response.status = 200;
      ctx.response.body = {
        msg: "socket close"
      };
    })
  );

  app.use(
    route.get("/mock/refresh", async ctx => {
      ctx.response.status = 200;
      ctx.response.body = {
        socketList: socketList,
        port: port
      };
    })
  );

  app.ws.use(
    route.all("*", ctx => {
      let httpEmitter = new HttpEmitter();

      if (socketList[ctx.url]) {
        socketList[ctx.url].push({
          url: ctx.url,
          httpEmitter: httpEmitter
        });
      } else {
        socketList[ctx.url] = [
          {
            url: ctx.url,
            httpEmitter: httpEmitter
          }
        ];
      }
      httpEmitter.on("sendMessage", function(data) {
        console.log("🐸🐸🐸 服务器发送的数据  ", data);
        ctx.websocket.send(data);
      });
      ctx.websocket.on("message", async message => {
        console.log("🐸🐸🐸🐸 服务器收到的数据  ", message);
        if (socketMap && socketMap[ctx.url]) {
          let returnMsg = await socketMap[ctx.url](message);
          emitterSocketEvent(ctx.url, returnMsg);
        }
      });
      httpEmitter.on("close", function() {
        console.log("socket close");
        ctx.websocket.close();
      });
    })
  );

  app.listen(port, host);

  console.log(`socket server is running on: ${host}:${port}`);
}

module.exports = {
  socketServer,
  emitterSocketEvent
};