feat: add ws support (#1058)

midwayjs · Jun 7, 2021 · e73cfcb · e73cfcb
1 parent c2bea86
commit e73cfcb
Show file tree

Hide file tree

Showing 24 changed files with 1,322 additions and 2 deletions.
diff --git a/packages/decorator/src/interface.ts b/packages/decorator/src/interface.ts
@@ -233,7 +233,7 @@ export enum MidwayFrameworkType {
   MS_GRPC = '@midwayjs/grpc',
   MS_RABBITMQ = '@midwayjs/rabbitmq',
   WS_IO = '@midwayjs/socketio',
-  WSS = '',
+  WS = '@midwayjs/ws',
   SERVERLESS_APP = '@midwayjs/serverless-app',
   CUSTOM = '',
   EMPTY = 'empty',

diff --git a/packages/decorator/src/ws/webSocketEvent.ts b/packages/decorator/src/ws/webSocketEvent.ts
@@ -91,6 +91,25 @@ export function WSEmit(
   };
 }
 
+export function WSBroadCast(
+  messageName = '',
+  roomName: string | string[] = []
+): MethodDecorator {
+  return (target: any, propertyKey: string, descriptor: PropertyDescriptor) => {
+    attachClassMetadata(
+      WS_EVENT_KEY,
+      {
+        eventType: WSEventTypeEnum.BROADCAST,
+        propertyName: propertyKey,
+        messageEventName: messageName,
+        roomName: [].concat(roomName),
+        descriptor,
+      },
+      target.constructor
+    );
+  };
+}
+
 /**
  * @deprecated please use @OnWSDisConnection
  */

diff --git a/packages/mock/package.json b/packages/mock/package.json
@@ -34,7 +34,8 @@
     "@types/amqplib": "*",
     "amqplib": "*",
     "socket.io": "^4.0.0",
-    "socket.io-client": "^4.0.0"
+    "socket.io-client": "^4.0.0",
+    "ws": "^7.4.5"
   },
   "dependencies": {
     "@midwayjs/bootstrap": "^2.10.18",

diff --git a/packages/mock/src/client/ws.ts b/packages/mock/src/client/ws.ts
@@ -0,0 +1,16 @@
+import * as WebSocket from 'ws';
+import * as http from 'http';
+import * as url from 'url';
+
+export async function createWebSocketClient(
+  address: string | url.URL,
+  options?: WebSocket.ClientOptions | http.ClientRequestArgs
+) {
+  const WebSocket = require('ws');
+  const client = new WebSocket(address, options);
+  return new Promise<WebSocket>(resolve => {
+    client.on('open', () => {
+      resolve(client);
+    });
+  });
+}
diff --git a/packages/mock/src/index.ts b/packages/mock/src/index.ts
@@ -8,6 +8,7 @@ export {
 export * from './client/http';
 export * from './client/rabbitMQ';
 export * from './client/socketio';
+export * from './client/ws';
 
 /**
  * @deprecated below