From b3a6d0f8ff2b430ca41ff76cf347f9fcc9ed18c6 Mon Sep 17 00:00:00 2001
From: xingyuv <xingyu4j@vip.qq.com>
Date: Wed, 22 Mar 2023 17:41:18 +0800
Subject: [PATCH] feat: ws

---
 src/views/infra/webSocket/index.vue | 101 +++++++++++++++++++++++++++-
 1 file changed, 100 insertions(+), 1 deletion(-)

diff --git a/src/views/infra/webSocket/index.vue b/src/views/infra/webSocket/index.vue
index 3b64cfc4..b0d2d38a 100644
--- a/src/views/infra/webSocket/index.vue
+++ b/src/views/infra/webSocket/index.vue
@@ -1,3 +1,102 @@
 <template>
-  <div>开发中</div>
+  <PageWrapper title="WebSocket 示例">
+    <div class="flex">
+      <div class="w-1/3 bg-white p-4">
+        <div class="flex items-center">
+          <span class="text-lg font-medium mr-4"> 连接状态: </span>
+          <Tag :color="getTagColor">{{ status }}</Tag>
+        </div>
+        <hr class="my-4" />
+
+        <div class="flex">
+          <Input v-model:value="server" disabled>
+            <template #addonBefore> 服务地址 </template>
+          </Input>
+          <a-button :type="getIsOpen ? 'danger' : 'primary'" @click="toggle">
+            {{ getIsOpen ? '关闭连接' : '开启连接' }}
+          </a-button>
+        </div>
+        <p class="text-lg font-medium mt-4">设置</p>
+        <hr class="my-4" />
+
+        <InputTextArea placeholder="需要发送到服务器的内容" :disabled="!getIsOpen" v-model:value="state.sendValue" allowClear />
+
+        <a-button type="primary" block class="mt-4" :disabled="!getIsOpen" @click="handlerSend"> 发送 </a-button>
+      </div>
+
+      <div class="w-2/3 bg-white ml-4 p-4">
+        <span class="text-lg font-medium mr-4"> 消息记录: </span>
+        <hr class="my-4" />
+
+        <div class="max-h-80 overflow-auto">
+          <ul>
+            <li v-for="item in getList" class="mt-2" :key="item.time">
+              <div class="flex items-center">
+                <span class="mr-2 text-primary font-medium">收到消息:</span>
+                <span>{{ formatToDateTime(item.time) }}</span>
+              </div>
+              <div>
+                {{ item.res }}
+              </div>
+            </li>
+          </ul>
+        </div>
+      </div>
+    </div>
+  </PageWrapper>
 </template>
+<script lang="ts" setup>
+import { reactive, watchEffect, computed, ref } from 'vue'
+import { Tag, Input } from 'ant-design-vue'
+import { PageWrapper } from '@/components/Page'
+import { useWebSocket } from '@vueuse/core'
+import { formatToDateTime } from '@/utils/dateUtil'
+import { useUserStore } from '@/store/modules/user'
+const InputTextArea = Input.TextArea
+
+const userStore = useUserStore()
+
+const state = reactive({
+  sendValue: '',
+  recordList: [] as { id: number; time: number; res: string }[]
+})
+const server = ref(
+  (import.meta.env.VITE_GLOB_API_URL.replace('/admin-api', '') + '/websocket/message').replace('http', 'ws') +
+    '?userId=' +
+    userStore.getUserInfo.user.id
+)
+const { status, data, send, close, open } = useWebSocket(server.value, {
+  autoReconnect: false,
+  heartbeat: true
+})
+watchEffect(() => {
+  if (data.value) {
+    try {
+      const res = JSON.parse(data.value)
+      state.recordList.push(res)
+    } catch (error) {
+      state.recordList.push({
+        res: data.value,
+        id: Math.ceil(Math.random() * 1000),
+        time: new Date().getTime()
+      })
+    }
+  }
+})
+const getIsOpen = computed(() => status.value === 'OPEN')
+const getTagColor = computed(() => (getIsOpen.value ? 'success' : 'red'))
+const getList = computed(() => {
+  return [...state.recordList].reverse()
+})
+function handlerSend() {
+  send(state.sendValue)
+  state.sendValue = ''
+}
+function toggle() {
+  if (getIsOpen.value) {
+    close()
+  } else {
+    open()
+  }
+}
+</script>