mmosaic/mosaicmap
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ws reconnect

Browse Source
This commit is contained in:
Tsuki 2025-08-25 00:14:25 +08:00
parent 14f611e796
commit 19d020faec
1 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
app/ws-context.tsx
View File

@ -54,7 +54,7 @@ export function WSProvider({ children }: MapProviderProps) {
reconnectAttempts: 0 reconnectAttempts: 0
}); });
const [isOnline, setIsOnline] = useState(typeof window !== 'undefined' ? navigator.onLine : true); const [isOnline, setIsOnline] = useState(typeof window !== 'undefined' ? navigator.onLine : true);
const { data, loading, error, restart } = useSubscription(SUBSCRIPTION_QUERY, { const { data, loading, error, restart } = useSubscription(SUBSCRIPTION_QUERY, {
errorPolicy: 'all', errorPolicy: 'all',
onError: (error) => { onError: (error) => {
@ -91,7 +91,7 @@ export function WSProvider({ children }: MapProviderProps) {
useEffect(() => { useEffect(() => {
const unsubscribe = subscribeToConnectionState((state) => { const unsubscribe = subscribeToConnectionState((state) => {
setConnectionState(state); setConnectionState(state);
switch (state.status) { switch (state.status) {
case 'connecting': case 'connecting':
setWsStatus(WsStatus.CONNECTING); setWsStatus(WsStatus.CONNECTING);
@ -113,16 +113,25 @@ export function WSProvider({ children }: MapProviderProps) {
// 监听订阅状态变化 // 监听订阅状态变化
useEffect(() => { useEffect(() => {
if (loading && wsStatus !== WsStatus.CONNECTING) { if (loading) {
setWsStatus(WsStatus.CONNECTING); // 只在第一次加载时设置为 connecting，避免覆盖 WebSocket 的实际状态
if (connectionState.status === 'disconnected') {
setWsStatus(WsStatus.CONNECTING);
}
} else if (error && !loading) { } else if (error && !loading) {
if (isOnline) { console.error('WebSocket subscription error:', error);
if (isOnline && connectionState.status !== 'reconnecting') {
setTimeout(() => { setTimeout(() => {
restart(); restart();
}, 5000); }, 5000);
} }
} else if (!loading && !error && data) {
// 确保在有数据时状态是 connected
if (wsStatus !== WsStatus.CONNECTED) {
setWsStatus(WsStatus.CONNECTED);
}
} }
}, [loading, error, isOnline, restart, wsStatus]); }, [loading, error, data, isOnline, restart, wsStatus, connectionState.status]);
const forceReconnect = () => { const forceReconnect = () => {
Sentry.startSpan({ Sentry.startSpan({