add 增加 sse 推送功能

疯狂的狮子Li

2024-07-26 425386f6f403d8dee0db2632ad3598039d833f4b

add 增加 sse 推送功能

 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 = () => {

 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;
  });
};