test(mock): update socket example code

liangskyli · Dec 24, 2022 · 1e7f6d0 · 1e7f6d0
1 parent dbe7dea
commit 1e7f6d0
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 10 deletions.
diff --git a/packages/mock/mock/socket/sock.ts b/packages/mock/mock/socket/sock.ts
@@ -1,7 +1,10 @@
-import type { Socket } from 'socket.io';
 import mockjs from 'mockjs';
+import type {
+  ISocketDefaultController,
+  ISocketNamespaceController,
+} from '../../src';
 
-const socketController = (socket: Socket) => {
+const socketDefaultController: ISocketDefaultController = (socket) => {
   const data = mockjs.mock({
     'list|2': [{ name: '@city', 'value|1-100': 50, 'type|0-2': 1 }],
     a: '1',
@@ -20,4 +23,24 @@ const socketController = (socket: Socket) => {
   });
 };
 
-export default socketController;
+export default socketDefaultController;
+
+const socketNamespaceController: ISocketNamespaceController = () => {
+  return {
+    '/namespace': (socket) => {
+      socket.emit('toClient', { data: 'init' });
+
+      // 定时推送数据
+      let toClient = 1;
+      setInterval(() => {
+        socket.emit('toClient', toClient++);
+      }, 5000);
+
+      // 接收客户端数据
+      socket.on('toServer', (clientData) => {
+        console.log('from client:', clientData);
+      });
+    },
+  };
+};
+export { socketNamespaceController };
diff --git a/packages/mock/mock/socket/sock2.js b/packages/mock/mock/socket/sock2.js
@@ -1,4 +1,4 @@
-const socketController = (socket) => {
+const socketDefaultController = (socket) => {
   const data = { a: 112 };
   // 数据发送客户端
   socket.emit('toClient', data);
@@ -9,4 +9,4 @@ const socketController = (socket) => {
   });
 };
 
-exports.default = socketController;
+exports.default = socketDefaultController;
diff --git a/packages/mock/test/mock/socket/sock2.js b/packages/mock/test/mock/socket/sock2.js
@@ -1,4 +1,4 @@
-const socketController = (socket) => {
+const socketDefaultController = (socket) => {
   const data = { a: 112 };
   // 数据发送客户端
   socket.emit('toClient', data);
@@ -9,4 +9,4 @@ const socketController = (socket) => {
   });
 };
 
-exports.default = socketController;
+exports.default = socketDefaultController;
diff --git a/packages/mock/test/mock/socket/sock3.ts b/packages/mock/test/mock/socket/sock3.ts
@@ -1,6 +1,6 @@
-import type { Socket } from 'socket.io';
+import type { ISocketDefaultController } from '../../../src';
 
-const socketController = (socket: Socket) => {
+const socketDefaultController: ISocketDefaultController = (socket) => {
   const data = { a: 112 };
   // 数据发送客户端
   socket.emit('toClient', data);
@@ -11,4 +11,4 @@ const socketController = (socket: Socket) => {
   });
 };
 
-export default socketController;
+export default socketDefaultController;