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