123

.env

@@ -19,3 +19,7 @@ MYSQL_DB = chicken_breeding
 
 # salt
 JWT_SECRET = 'eyJ0e2223XAiOiJKV1Q33iLCJhbGciOiJsdJ9.eyJqd4GkiOiJjYjc4M2I5Yi1kMGYxLTRjMWEtYmQyMC0zOTM'
+
+# websocket
+WEB_SOCKET_PORT = 4000
+WEB_SOCKET_SERVER_HOST = 127.0.0.1

package.json

@@ -20,7 +20,8 @@
     "moment": "^2.29.4",
     "mysql2": "^3.2.0",
     "sequelize": "^6.29.1",
-    "uuid": "^9.0.0"
+    "uuid": "^9.0.0",
+    "ws": "^8.13.0"
   },
   "devDependencies": {
     "nodemon": "^2.0.21"

src/main.js

@@ -2,15 +2,42 @@ const {
   APP_PORT,
   TCP_PORT,
   TCP_SERVER_HOST,
+  WEB_SOCKET_PORT,
+  WEB_SOCKET_SERVER_HOST,
 } = require("./config/config.default");
+const WebSocket = require("ws");
 
 const app = require("./app/");
 const server = require("./tcp/server");
 
 app.listen(APP_PORT, () => {
-  console.log(`app server is running on http://localhost:${APP_PORT}`);
+  console.log(`nodejs server is running on http://localhost:${APP_PORT}`);
 });
 
-server.listen(TCP_PORT, TCP_SERVER_HOST, () => {
-  console.log(`tcp server is running on http://localhost:${TCP_PORT}`);
+// server.listen(TCP_PORT, TCP_SERVER_HOST, () => {
+//   console.log(`tcp server is running on http://localhost:${TCP_PORT}`);
+// });
+
+// websocket server
+const WebSocketServer = WebSocket.Server;
+
+//在4000端口上打开了一个WebSocket Server，该实例由变量wss引用。
+const wss = new WebSocketServer({
+  port: WEB_SOCKET_PORT,
+});
+
+wss.on("connection", function (ws) {
+  //在connection事件中，回调函数会传入一个WebSocket的实例，表示这个WebSocket连接。
+  console.log(
+    `websocket server is connected on ws:/${WEB_SOCKET_SERVER_HOST}:${WEB_SOCKET_PORT}`
+  );
+  ws.on("message", function (message) {
+    console.log(`[SERVER] Received:${message}`);
+    ws.send(`server received : ${message}`, (err) => {
+      // 对接收来的消息进行处理分发
+      if (err) {
+        console.log(`[SERVER] error:${err}`);
+      }
+    });
+  });
 });

src/websocket/client.js

@@ -0,0 +1,17 @@
+const WebSocket = require("ws");
+let ws = new WebSocket("ws://localhost:4000");
+
+// 打开WebSocket连接后立刻发送一条消息:
+ws.on("open", function () {
+  console.log(`[CLIENT] open()`);
+  ws.send(
+    JSON.stringify({
+      a: 1,
+    })
+  );
+});
+
+// 响应收到的消息:
+ws.on("message", function (message) {
+  console.log(`[CLIENT] Received: ${message}`);
+});

yarn.lock

@@ -1043,6 +1043,11 @@ wrappy@1:
   resolved "https://registry.yarnpkg.com/wrappy/-/wrappy-1.0.2.tgz#b5243d8f3ec1aa35f1364605bc0d1036e30ab69f"
   integrity sha512-l4Sp/DRseor9wL6EvV2+TuQn63dMkPjZ/sp9XkghTEbV9KlPS1xUsZ3u7/IQO4wxtcFB4bgpQPRcR3QCvezPcQ==
 
+ws@^8.13.0:
+  version "8.13.0"
+  resolved "https://registry.npmmirror.com/ws/-/ws-8.13.0.tgz#9a9fb92f93cf41512a0735c8f4dd09b8a1211cd0"
+  integrity sha512-x9vcZYTrFPC7aSIbj7sRCYo7L/Xb8Iy+pW0ng0wt2vCJv7M9HOMy0UoN3rr+IFC7hb7vXoqS+P9ktyLLLhO+LA==
+
 yallist@^4.0.0:
   version "4.0.0"
   resolved "https://registry.yarnpkg.com/yallist/-/yallist-4.0.0.tgz#9bb92790d9c0effec63be73519e11a35019a3a72"