From 68cb0ee20ad19f96952e4d4746c1689d829f6bb8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=83=9D=E5=85=88=E7=91=9E?= <1490493387@qq.com>
Date: Sat, 3 Jun 2023 11:12:17 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20:sparkles:=20=E6=96=B0=E5=A2=9E`websock?=
 =?UTF-8?q?et`=E6=B5=8B=E8=AF=95=E9=A1=B5=E9=9D=A2(=E5=AE=9E=E9=AA=8C?=
 =?UTF-8?q?=E9=98=B6=E6=AE=B5)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Former-commit-id: 8c5a49f95da85f6a245d9a3cc60d5f925cfc0861
---
 src/api/user/types.ts         |   1 +
 src/api/websocket/index.ts    |  27 ++++++++
 src/store/modules/user.ts     |   6 ++
 src/views/dashboard/index.vue |  10 +--
 src/views/demo/websocket.vue  | 119 ++++++++++++++++++++++++++++++++++
 5 files changed, 159 insertions(+), 4 deletions(-)
 create mode 100644 src/api/websocket/index.ts
 create mode 100644 src/views/demo/websocket.vue

diff --git a/src/api/user/types.ts b/src/api/user/types.ts
index 613ad6e1..057be5f3 100644
--- a/src/api/user/types.ts
+++ b/src/api/user/types.ts
@@ -2,6 +2,7 @@
  * 登录用户信息
  */
 export interface UserInfo {
+  userId: number;
   nickname: string;
   avatar: string;
   roles: string[];
diff --git a/src/api/websocket/index.ts b/src/api/websocket/index.ts
new file mode 100644
index 00000000..867a2a23
--- /dev/null
+++ b/src/api/websocket/index.ts
@@ -0,0 +1,27 @@
+import request from "@/utils/request";
+
+/**
+ * 发送消息给所有人
+ *
+ * @param file
+ */
+export function sendToAll(message: string) {
+  return request({
+    url: "/websocket/sendToAll",
+    method: "post",
+    params: { message: message },
+  });
+}
+
+/**
+ * 发送消息给指定用户
+ *
+ * @param
+ */
+export function sendToUser(userId: number, message: string) {
+  return request({
+    url: "/websocket/sendToUser/" + userId,
+    method: "post",
+    params: { message: message },
+  });
+}
diff --git a/src/store/modules/user.ts b/src/store/modules/user.ts
index 72d806d5..d30fee09 100644
--- a/src/store/modules/user.ts
+++ b/src/store/modules/user.ts
@@ -12,6 +12,7 @@ import { useStorage } from "@vueuse/core";
 
 export const useUserStore = defineStore("user", () => {
   // state
+  const userId = ref();
   const token = useStorage("accessToken", "");
   const nickname = ref("");
   const avatar = ref("");
@@ -49,6 +50,7 @@ export const useUserStore = defineStore("user", () => {
           if (!data.roles || data.roles.length <= 0) {
             reject("getUserInfo: roles must be a non-null array!");
           }
+          userId.value = data.userId;
           nickname.value = data.nickname;
           avatar.value = data.avatar;
           roles.value = data.roles;
@@ -95,6 +97,10 @@ export const useUserStore = defineStore("user", () => {
     getInfo,
     logout,
     resetToken,
+    /**
+     * 当前登录用户ID
+     */
+    userId,
   };
 });
 
diff --git a/src/views/dashboard/index.vue b/src/views/dashboard/index.vue
index 359c9c87..7e1d7d73 100644
--- a/src/views/dashboard/index.vue
+++ b/src/views/dashboard/index.vue
@@ -1,8 +1,10 @@
-<script lang="ts">
-export default { name: "Dashboard" };
-</script>
-
 <script setup lang="ts">
+defineOptions({
+  // eslint-disable-next-line vue/no-reserved-component-names
+  name: "Dashboard",
+  inheritAttrs: false,
+});
+
 import { useUserStore } from "@/store/modules/user";
 import { useTransition, TransitionPresets } from "@vueuse/core";
 
diff --git a/src/views/demo/websocket.vue b/src/views/demo/websocket.vue
new file mode 100644
index 00000000..fae813d0
--- /dev/null
+++ b/src/views/demo/websocket.vue
@@ -0,0 +1,119 @@
+<!-- websocket 示例 -->
+<script setup lang="ts">
+import { sendToAll, sendToUser } from "@/api/websocket";
+
+const inputVal = ref("初始内容");
+
+const topicMsgs = ref<string[]>(["接收到一条主题消息"]); // 主题消息列表
+const p2pMsgs = ref<string[]>(["接收到一条点对线消息"]); // 点对点消息列表
+
+import { useUserStore } from "@/store/modules/user";
+
+const userId = useUserStore().userId;
+
+import { useWebSocket } from "@vueuse/core";
+
+const { data, status, close, send, open } = useWebSocket(
+  "ws://localhost:8989/ws",
+  {
+    onConnected(ws) {
+      console.log("订阅主题");
+      // 连接建立后发送订阅消息
+      ws.send(JSON.stringify({ type: "subscribe", topic: "/topic/all" }));
+    },
+    onMessage(ws, event) {
+      // 获取接收到的消息内容
+      const message = event.data;
+
+      // 处理消息内容
+      console.log("Received message:", message);
+    },
+  }
+);
+
+// 监听 WebSocket 连接状态变化
+watch(status, (newStatus) => {
+  if (newStatus === "OPEN") {
+    // 连接已打开，订阅主题
+    console.log(" 连接已打开，订阅主题");
+    const subscribeMessage = {
+      type: "subscribe",
+      channel: "/topic/all",
+    };
+    send(JSON.stringify(subscribeMessage));
+  } else if (newStatus === "CLOSED") {
+    // 连接已关闭，执行相应的清理操作
+    // ...
+  }
+});
+
+// 监听 WebSocket 接收到的消息
+watch(data, (newData) => {
+  console.log("Received data:", newData);
+
+  // 解析消息体
+  const message = JSON.parse(newData);
+
+  // 判断消息主题并处理
+  if (message.topic === "topic1") {
+    // 处理来自 topic1 的消息
+    console.log("Received message from topic1:", message);
+  } else if (message.topic === "topic2") {
+    // 处理来自 topic2 的消息
+    console.log("Received message from topic2:", message);
+  }
+  // 可以根据需要添加更多的判断逻辑来处理其他主题的消息
+});
+
+function handleSendToAll() {
+  sendToAll(inputVal.value);
+}
+
+function handleSendToUser() {
+  sendToUser(userId, inputVal.value);
+}
+onMounted(() => {});
+</script>
+
+<template>
+  <div class="app-container">
+    <el-link
+      href="https://gitee.com/youlaiorg/vue3-element-admin/blob/master/src/views/demo/websocket.vue"
+      type="primary"
+      target="_blank"
+      class="mb-[20px]"
+      >示例源码 请点击>>>></el-link
+    >
+
+    <div>
+      <div class="search">
+        <el-form :inline="true">
+          <el-form-item> <el-input v-model="inputVal" /></el-form-item>
+          <el-form-item
+            ><el-button @click="handleSendToUser">发送点对点消息</el-button>
+            <el-button @click="handleSendToAll">发送广播消息</el-button>
+          </el-form-item>
+        </el-form>
+      </div>
+
+      <el-row :gutter="20">
+        <el-col :span="12">
+          <el-card>
+            <template #header>点对点消息接收 </template>
+            <div v-for="(msg, index) in p2pMsgs" :key="index">
+              {{ msg }}
+            </div>
+          </el-card>
+        </el-col>
+        <el-col :span="12">
+          <el-card>
+            <template #header> 广播消息接收 </template>
+            <div v-for="(msg, index) in topicMsgs" :key="index">
+              {{ msg }}
+            </div>
+          </el-card>
+        </el-col>
+      </el-row>
+    </div>
+  </div>
+</template>