From 425386f6f403d8dee0db2632ad3598039d833f4b Mon Sep 17 00:00:00 2001
From: 疯狂的狮子Li <15040126243@163.com>
Date: 星期五, 26 七月 2024 16:06:17 +0800
Subject: [PATCH] add 增加 sse 推送功能
---
src/layout/index.vue | 2 ++
src/utils/sse.ts | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
2 files changed, 50 insertions(+), 0 deletions(-)
diff --git a/src/layout/index.vue b/src/layout/index.vue
index 29fb5ff..e7bec82 100644
--- a/src/layout/index.vue
+++ b/src/layout/index.vue
@@ -27,6 +27,7 @@
import useAppStore from '@/store/modules/app';
import useSettingsStore from '@/store/modules/settings';
import { initWebSocket } from '@/utils/websocket';
+import { initSSE } from "@/utils/sse";
const settingsStore = useSettingsStore();
const theme = computed(() => settingsStore.theme);
@@ -69,6 +70,7 @@
onMounted(() => {
let protocol = window.location.protocol === 'https:' ? 'wss://' : 'ws://';
initWebSocket(protocol + window.location.host + import.meta.env.VITE_APP_BASE_API + '/resource/websocket');
+ initSSE(import.meta.env.VITE_APP_BASE_API + '/sse')
});
const handleClickOutside = () => {
diff --git a/src/utils/sse.ts b/src/utils/sse.ts
new file mode 100644
index 0000000..7db4aa7
--- /dev/null
+++ b/src/utils/sse.ts
@@ -0,0 +1,48 @@
+import { getToken } from '@/utils/auth';
+import { ElNotification } from 'element-plus';
+import useNoticeStore from '@/store/modules/notice';
+
+let message = '';
+
+// 鍒濆鍖�
+export const initSSE = (url: any) => {
+ if (import.meta.env.VITE_APP_WEBSOCKET === 'false') {
+ return;
+ }
+ url = url + '?Authorization=Bearer ' + getToken() + '&clientid=' + import.meta.env.VITE_APP_CLIENT_ID
+ const {
+ data,
+ error
+ } = useEventSource(url, [], {
+ autoReconnect: {
+ retries: 10,
+ delay: 3000,
+ onFailed() {
+ console.log('Failed to connect after 10 retries')
+ },
+ }
+ });
+
+ watch(error, () => {
+ console.log('SSE connection error:', error.value)
+ error.value = null;
+ });
+
+ watch(data, () => {
+ if (!data.value) return;
+ useNoticeStore().addNotice({
+ message: data.value,
+ read: false,
+ time: new Date().toLocaleString()
+ });
+ ElNotification({
+ title: '娑堟伅',
+ message: data.value,
+ type: 'success',
+ duration: 3000
+ });
+ data.value = null;
+ });
+};
+
+
--
Gitblit v1.9.3