refactor(sse): 重构 SSE 分层并同步调整通知中心与示例页

- src/composables/sse: 删除,连接实现下沉到 utils/sse,事件名枚举迁到 enums/sse
- src/utils/sse: 新增 SSE 连接与事件订阅实现
- src/stores/dict: 改用 utils/sse 订阅字典变更
- src/stores/user: 登出清理改用 utils/sse 的 cleanupSse
- src/views/dashboard: 在线用户订阅下沉为页面级 composable
- src/components/NoticeCenter: 由 NoticeDropdown 改名
- src/views/demo: 示例页面按 component/route/table 归类
- src/router: 同步示例详情页路由路径
This commit is contained in:
Ray.Hao
2026-09-01 07:22:19 +08:00
parent d36ccf54d5
commit d136266acd
47 changed files with 207 additions and 430 deletions

View File

@@ -1,8 +1,3 @@
// SSE 服务
export { setupSse, cleanupSseServices } from "./sse";
export { useSse, useDictSync, useOnlineUsers, cleanupSse, SseConnectionState, SseTopics } from "./sse";
export type { DictChangeMessage, DictChangeCallback, SseTopic } from "./sse";
// 表格相关
export { useTableSelection } from "./useTableSelection";
export { usePageTable } from "./usePageTable";

View File

@@ -1,37 +0,0 @@
import { useDictSync } from "./useDictSync";
import { useOnlineUsers } from "./useOnlineUsers";
import { useSse, cleanupSse } from "./useSse";
/**
* 初始化所有 SSE 服务
*/
export function setupSse() {
const sse = useSse();
sse.connect();
const dictSync = useDictSync();
dictSync.initialize();
const onlineUsers = useOnlineUsers();
onlineUsers.initialize();
}
/**
* 清理所有 SSE 连接
*/
export function cleanupSseServices() {
const dictSync = useDictSync();
dictSync.cleanup();
const onlineUsers = useOnlineUsers();
onlineUsers.cleanup();
cleanupSse();
}
export { useDictSync } from "./useDictSync";
export { useOnlineUsers } from "./useOnlineUsers";
export { useSse, cleanupSse, SseConnectionState } from "./useSse";
export { SseTopics } from "./sseTopics";
export type { DictChangeMessage, DictChangeCallback } from "./useDictSync";
export type { SseTopic } from "./sseTopics";

View File

@@ -1,17 +0,0 @@
/** SSE 事件名常量,与后端 SseTopics.java 一一对应 */
export const SseTopics = {
/** 字典变更事件 */
DICT: "dict",
/** 在线用户数事件 */
ONLINE_USERS: "online-users",
/** 系统消息事件 */
SYSTEM: "system",
/** 心跳事件 */
PING: "ping",
/** 通知事件 */
NOTICE: "notice",
/** 通知撤回事件 */
NOTICE_REVOKE: "notice-revoke",
} as const;
export type SseTopic = (typeof SseTopics)[keyof typeof SseTopics];

View File

@@ -1,88 +0,0 @@
import { useDictStoreHook } from "@/stores/dict";
import { useSse } from "./useSse";
import { SseTopics } from "./sseTopics";
/** 字典变更消息体 */
export interface DictChangeMessage {
/** 字典编码 */
dictCode: string;
}
/** 字典变更回调函数类型 */
export type DictChangeCallback = (message: DictChangeMessage) => void;
let globalInstance: ReturnType<typeof createDictSyncComposable> | null = null;
function createDictSyncComposable() {
const dictStore = useDictStoreHook();
const sse = useSse();
const callbacks: DictChangeCallback[] = [];
let unsubscribe: (() => void) | null = null;
let initialized = false; // 防止重复初始化导致重复订阅
// 处理字典变更消息:清除指定字典缓存,并通知所有已注册回调
const handleDictChange = (data: DictChangeMessage) => {
const { dictCode } = data;
if (!dictCode) {
console.warn("[DictSync] 收到无效的字典变更消息:缺少 dictCode");
return;
}
dictStore.removeDictItem(dictCode);
callbacks.forEach((cb) => {
try {
cb(data);
} catch (err) {
console.error("[DictSync] 回调执行失败:", err);
}
});
};
// 订阅 SSE 字典变更事件(幂等:重复调用不会产生重复订阅)
const initialize = () => {
if (initialized) return;
initialized = true;
unsubscribe = sse.on(SseTopics.DICT, handleDictChange);
};
// 取消 SSE 订阅并清空所有回调
const cleanup = () => {
initialized = false;
unsubscribe?.();
unsubscribe = null;
callbacks.length = 0;
};
// 注册字典变更回调,返回取消注册函数
const onDictChange = (cb: DictChangeCallback) => {
callbacks.push(cb);
return () => {
const idx = callbacks.indexOf(cb);
if (idx !== -1) callbacks.splice(idx, 1);
};
};
return {
isConnected: sse.isConnected,
connectionState: sse.connectionState,
initialize,
cleanup,
onDictChange,
};
}
/**
* 字典同步组合式函数(单例模式)
*
* 监听 SSE 字典变更事件,收到变更时自动清除对应字典缓存,
* 并通知所有已注册的回调函数。
*
* @returns 字典同步实例,包含连接状态、初始化、清理和回调注册方法
*/
export function useDictSync() {
if (!globalInstance) {
globalInstance = createDictSyncComposable();
}
return globalInstance;
}

View File

@@ -1,50 +0,0 @@
import { ref, readonly } from "vue";
import { useSse } from "./useSse";
import { SseTopics } from "./sseTopics";
let globalInstance: ReturnType<typeof createOnlineUsersComposable> | null = null;
function createOnlineUsersComposable() {
const onlineUserCount = ref(0);
const lastUpdateTime = ref(0);
const sse = useSse();
let unsubscribe: (() => void) | null = null;
const handleOnlineUsersMessage = (count: number) => {
if (!Number.isFinite(count) || count < 0) return;
onlineUserCount.value = count;
lastUpdateTime.value = Date.now();
};
const initialize = () => {
unsubscribe = sse.on(SseTopics.ONLINE_USERS, handleOnlineUsersMessage);
};
const cleanup = () => {
if (unsubscribe) {
unsubscribe();
unsubscribe = null;
}
onlineUserCount.value = 0;
lastUpdateTime.value = 0;
};
return {
onlineUserCount: readonly(onlineUserCount),
lastUpdateTime: readonly(lastUpdateTime),
isConnected: sse.isConnected,
connectionState: sse.connectionState,
initialize,
cleanup,
};
}
/** 在线用户数组合式函数(单例模式) */
export function useOnlineUsers() {
if (!globalInstance) {
globalInstance = createOnlineUsersComposable();
}
return globalInstance;
}

View File

@@ -1,341 +0,0 @@
import { AuthStorage } from "@/utils/auth";
import { useUserStoreHook } from "@/stores/user";
/** SSE 连接配置选项 */
export interface UseSseOptions {
/** SSE 连接地址,默认走 VITE_APP_BASE_API 代理 */
url?: string;
/** 是否在控制台打印调试日志 */
debug?: boolean;
/** 连接超时时间(ms),默认 10000 */
connectionTimeout?: number;
/** 重连间隔基数(ms),实际间隔 = min(基数 × 2^n, maxReconnectInterval) */
reconnectInterval?: number;
/** 重连间隔上限(ms),默认 120000 */
maxReconnectInterval?: number;
/** 最大重试次数,超过后停止重连,默认 10 */
maxReconnectAttempts?: number;
}
/** SSE 事件处理器类型 */
type EventHandler = (data: unknown) => void;
/** SSE 流解析中间状态 */
type SseParseState = {
currentEvent: string;
currentData: string;
buffer: string;
};
/** SSE 连接状态 */
export enum SseConnectionState {
DISCONNECTED = "DISCONNECTED",
CONNECTING = "CONNECTING",
CONNECTED = "CONNECTED",
}
let globalInstance: ReturnType<typeof createSseConnection> | null = null;
function createSseConnection(options: UseSseOptions = {}) {
const baseUrl = import.meta.env.VITE_APP_BASE_API;
const defaultUrl = `${baseUrl}/api/v1/sse/connect`;
const config = {
url: options.url ?? defaultUrl,
debug: options.debug ?? false,
connectionTimeout: options.connectionTimeout ?? 10000, // 连接超时 10s
reconnectInterval: options.reconnectInterval ?? 5000, // 首次重连等 5s,之后翻倍
maxReconnectInterval: options.maxReconnectInterval ?? 120000, // 重连间隔最大 2min
maxReconnectAttempts: options.maxReconnectAttempts ?? 10, // 最多重试 10 次
};
const connectionState = ref<SseConnectionState>(SseConnectionState.DISCONNECTED);
const isConnected = computed(() => connectionState.value === SseConnectionState.CONNECTED);
let abortController: AbortController | null = null;
let connectionTimeoutTimer: ReturnType<typeof setTimeout> | null = null;
let reader: ReadableStreamDefaultReader<Uint8Array> | null = null;
let isManualDisconnect = false; // 主动断开则不重连
let reconnectTimer: ReturnType<typeof setTimeout> | null = null;
let reconnectAttempts = 0;
let currentReconnectInterval = config.reconnectInterval;
let tokenRefreshed = false; // 本轮拒绝是否已刷新过令牌,防止无限刷新
const eventHandlers = new Map<string, Set<EventHandler>>();
const log = (...args: unknown[]) => {
if (config.debug) {
console.debug("[SSE]", ...args);
}
};
const logError = (...args: unknown[]) => console.error("[SSE]", ...args);
// 清除定时器并返回 null,用于链式赋值
const clearTimer = (timer: ReturnType<typeof setTimeout> | null): null => {
if (timer) {
clearTimeout(timer);
}
return null;
};
// 重置重连状态:次数归零、间隔恢复基数
const resetReconnectState = () => {
reconnectAttempts = 0;
currentReconnectInterval = config.reconnectInterval;
};
// 指数退避:当前间隔翻倍,不超过上限
const advanceReconnectState = () => {
currentReconnectInterval = Math.min(currentReconnectInterval * 2, config.maxReconnectInterval);
};
// 分发 SSE 事件:先尝试 JSON.parse,失败则传原始字符串
const flushSseEvent = (eventName: string, data: string) => {
if (!data) return;
const handlers = eventHandlers.get(eventName);
if (handlers) {
try {
const parsed = JSON.parse(data);
handlers.forEach((handler) => handler(parsed));
} catch {
handlers.forEach((handler) => handler(data));
}
}
log(`收到事件[${eventName}]:`, data);
};
// 解析单行 SSE 数据:区分 event/data/注释/空行(触发分发)
const handleSseLine = (line: string, state: SseParseState) => {
if (line.startsWith(":")) return;
if (line.startsWith("event:")) {
state.currentEvent = line.slice(6).trim() || "message";
return;
}
if (line.startsWith("data:")) {
const dataLine = line.slice(5).trim();
state.currentData = state.currentData ? `${state.currentData}\n${dataLine}` : dataLine;
return;
}
if (line === "") {
flushSseEvent(state.currentEvent, state.currentData);
state.currentEvent = "message";
state.currentData = "";
}
};
// 持续读取流数据并按行解析,异常时触发重连
const consumeSseStream = async (streamReader: ReadableStreamDefaultReader<Uint8Array>) => {
const decoder = new TextDecoder();
const state: SseParseState = { currentEvent: "message", currentData: "", buffer: "" };
try {
while (true) {
const { done, value } = await streamReader.read();
if (done) {
reader = null;
connectionState.value = SseConnectionState.DISCONNECTED;
log("SSE 连接已关闭");
return;
}
state.buffer += decoder.decode(value, { stream: true });
const lines = state.buffer.split("\n");
state.buffer = lines.pop() || "";
for (const line of lines) {
handleSseLine(line, state);
}
}
} catch (err) {
reader = null;
connectionState.value = SseConnectionState.DISCONNECTED;
if (err instanceof Error && err.name === "AbortError") {
log("SSE 流读取已主动断开");
} else {
logError("SSE 流读取错误:", err);
scheduleReconnect();
}
}
};
// 调度重连:指数退避,达到上限或主动断开时停止
const scheduleReconnect = () => {
if (isManualDisconnect) return;
if (config.maxReconnectAttempts > 0 && reconnectAttempts >= config.maxReconnectAttempts) {
log(`已达到最大重试次数 ${config.maxReconnectAttempts},停止重连`);
return;
}
reconnectAttempts++;
log(`将在 ${currentReconnectInterval}ms 后重试(${reconnectAttempts})`);
reconnectTimer = setTimeout(() => {
advanceReconnectState();
connect();
}, currentReconnectInterval);
};
// 建立连接:校验 token → fetch → 超时检测 → 消费流;401/403 先刷新令牌重连,失败则停止
const connect = () => {
isManualDisconnect = false;
if (connectionState.value !== SseConnectionState.DISCONNECTED) {
log(
connectionState.value === SseConnectionState.CONNECTED
? "SSE 已连接,跳过重复连接"
: "SSE 正在连接中,跳过重复连接"
);
return;
}
const token = AuthStorage.getAccessToken();
if (!token) {
log("未检测到有效令牌,稍后重试");
// 走统一重连调度,受 maxReconnectAttempts 上限约束
scheduleReconnect();
return;
}
connectionState.value = SseConnectionState.CONNECTING;
abortController = new AbortController();
connectionTimeoutTimer = setTimeout(() => {
if (connectionState.value === SseConnectionState.CONNECTING) {
log("SSE 连接超时");
disconnect();
}
}, config.connectionTimeout);
log("正在建立 SSE 连接...");
fetch(config.url, {
method: "GET",
headers: {
Authorization: `Bearer ${token}`,
Accept: "text/event-stream",
},
signal: abortController.signal,
})
.then(async (response) => {
if (!response.ok) {
if (response.status === 401 || response.status === 403) {
// 令牌过期:刷新后用新令牌重连,刷新失败或令牌仍无效则停止重连
if (!tokenRefreshed) {
tokenRefreshed = true;
connectionTimeoutTimer = clearTimer(connectionTimeoutTimer);
try {
const userStore = useUserStoreHook();
await userStore.refreshTokenOnce();
if (AuthStorage.getAccessToken()) {
connectionState.value = SseConnectionState.DISCONNECTED;
log(`SSE 连接被拒绝(HTTP ${response.status}),令牌已刷新,使用新令牌重连`);
connect();
return null;
}
} catch (err) {
logError("SSE 令牌刷新失败:", err);
}
}
isManualDisconnect = true;
connectionState.value = SseConnectionState.DISCONNECTED;
log(`SSE 连接被拒绝(HTTP ${response.status}),不再重连`);
return null;
}
throw new Error(`HTTP ${response.status}`);
}
connectionTimeoutTimer = clearTimer(connectionTimeoutTimer);
connectionState.value = SseConnectionState.CONNECTED;
tokenRefreshed = false;
resetReconnectState();
log("SSE 连接已建立");
return response.body?.getReader();
})
.then((streamReader) => {
if (!streamReader) return;
reader = streamReader;
return consumeSseStream(streamReader);
})
.catch((err: unknown) => {
if (err instanceof Error && err.name === "AbortError") {
log("SSE 连接已主动断开");
} else {
logError("SSE 连接错误:", err);
connectionState.value = SseConnectionState.DISCONNECTED;
scheduleReconnect();
}
});
};
// 订阅指定事件,返回取消订阅函数
const on = <T = unknown>(eventName: string, handler: (data: T) => void): (() => void) => {
if (!eventHandlers.has(eventName)) {
eventHandlers.set(eventName, new Set());
}
const wrappedHandler: EventHandler = (data) => handler(data as T);
eventHandlers.get(eventName)!.add(wrappedHandler);
log(`已订阅事件: ${eventName}`);
return () => {
const handlers = eventHandlers.get(eventName);
if (handlers) {
handlers.delete(wrappedHandler);
if (handlers.size === 0) {
eventHandlers.delete(eventName);
}
}
};
};
// 主动断开:清除定时器、取消流读取、中止请求,不触发重连
const disconnect = () => {
isManualDisconnect = true;
connectionTimeoutTimer = clearTimer(connectionTimeoutTimer);
reconnectTimer = clearTimer(reconnectTimer);
reader?.cancel();
reader = null;
abortController?.abort();
abortController = null;
connectionState.value = SseConnectionState.DISCONNECTED;
log("SSE 连接已断开");
};
// 断开连接并清空所有事件订阅
const cleanup = () => {
disconnect();
eventHandlers.clear();
log("SSE 资源已清理");
};
return {
connectionState: readonly(connectionState),
isConnected,
connect,
disconnect,
cleanup,
on,
};
}
/**
* SSE 连接组合式函数(单例模式)
*
* 基于 fetch + ReadableStream 实现,支持指数退避重连、
* 事件订阅/取消订阅、主动断开与资源清理。
*
* @param options - 连接配置选项
* @returns SSE 连接实例,包含连接状态、connect/disconnect/on/cleanup 方法
*/
export function useSse(options: UseSseOptions = {}) {
if (!globalInstance) {
globalInstance = createSseConnection(options);
}
return globalInstance;
}
/** 清理 SSE 单例:断开连接、清空订阅、释放全局引用 */
export function cleanupSse() {
if (globalInstance) {
globalInstance.cleanup();
globalInstance = null;
}
}