修改websocket日志

back/app.ts

@@ -9,7 +9,7 @@ import Logger from './loaders/logger';
 async function startServer() {
   const app = express();
 
-  await require('./loaders').default({ expressApp: app });
+  await require('./loaders/app').default({ expressApp: app });
 
   const server = app
     .listen(config.port, () => {
@@ -24,7 +24,7 @@ async function startServer() {
       process.exit(1);
     });
 
-  await require('./loaders/sock').default({ server });
+  await require('./loaders/server').default({ server });
 }
 
 startServer();

back/loaders/app.ts

@@ -5,8 +5,6 @@ import initData from './initData';
 import { Application } from 'express';
 
 export default async ({ expressApp }: { expressApp: Application }) => {
-  Logger.info('✌️ DB loaded and connected!');
-
   await dependencyInjectorLoader({
     models: [],
   });

back/loaders/server.ts

@@ -0,0 +1,16 @@
+import { Server } from 'http';
+import Logger from './logger';
+import Sock from './sock';
+
+export default async ({ server }: { server: Server }) => {
+  await Sock({ server });
+  Logger.info('✌️ Sock loaded');
+
+  process.on('SIGINT', () => {
+    Logger.info('✌️ Server need close');
+    server.close(() => {
+      Logger.info('✌️ Server closed');
+      process.exit(0);
+    });
+  });
+};

back/loaders/sock.ts

@@ -1,4 +1,4 @@
-import sockjs from 'sockjs';
+import sockJs from 'sockjs';
 import { Server } from 'http';
 import Logger from './logger';
 import { Container } from 'typedi';
@@ -8,7 +8,7 @@ import fs from 'fs';
 import { getPlatform } from '../config/util';
 
 export default async ({ server }: { server: Server }) => {
-  const echo = sockjs.createServer({ prefix: '/api/ws' });
+  const echo = sockJs.createServer({ prefix: '/api/ws', log: () => {} });
   const sockService = Container.get(SockService);
 
   echo.on('connection', (conn) => {
@@ -18,7 +18,7 @@ export default async ({ server }: { server: Server }) => {
     if (data) {
       const { token = '', tokens = {} } = JSON.parse(data);
       if (headerToken === token || tokens[platform] === headerToken) {
-        Logger.info('✌️ Sockjs connection success');
+        conn.write('hanhh');
         sockService.addClient(conn);
 
         conn.on('data', (message) => {

src/layouts/index.tsx

@@ -127,13 +127,15 @@ export default function (props: any) {
     ws.current = new SockJS(
       `${location.origin}/api/ws?token=${localStorage.getItem(config.authKey)}`,
     );
-    ws.current.onopen = (e: any) => {
-      console.log('websocket连接成功', e);
+
+    ws.current.onmessage = (e: any) => {
+      if (e.data === 'hanhh') {
+        console.log('websocket连接成功', e);
+      } else {
+        console.log('websocket连接失败', e);
+      }
     };
 
-    ws.current.onclose = (e: any) => {
-      console.log('websocket已关闭', e);
-    };
     const wsCurrent = ws.current;
 
     return () => {