Jayden/AFFiNE-Mirror
1
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-02-15 05:37:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: remove awareness state before window unload (#4752)

Browse Source
This commit is contained in:
Flrande
2023-10-30 03:50:18 +08:00
committed by GitHub
parent 2a4495f7ee
commit 2aa4b4c1f3
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
packages/frontend/workspace/src/affine/index.ts
View File

@@ -5,6 +5,7 @@ import {
applyAwarenessUpdate, applyAwarenessUpdate,
type Awareness, type Awareness,
encodeAwarenessUpdate, encodeAwarenessUpdate,
removeAwarenessStates,
} from 'y-protocols/awareness'; } from 'y-protocols/awareness';
import type { DocDataSource } from 'y-provider'; import type { DocDataSource } from 'y-provider';
import type { Doc } from 'yjs'; import type { Doc } from 'yjs';
@@ -190,6 +191,10 @@ function setupAffineAwareness(
.catch(err => console.error(err)); .catch(err => console.error(err));
}; };
const windowBeforeUnloadHandler = () => {
removeAwarenessStates(awareness, [awareness.clientID], 'window unload');
};
conn.on('server-awareness-broadcast', awarenessBroadcast); conn.on('server-awareness-broadcast', awarenessBroadcast);
conn.on('new-client-awareness-init', newClientAwarenessInitHandler); conn.on('new-client-awareness-init', newClientAwarenessInitHandler);
awareness.on('update', awarenessUpdate); awareness.on('update', awarenessUpdate);
@@ -198,9 +203,12 @@ function setupAffineAwareness(
conn.emit('awareness-init', rootDoc.guid); conn.emit('awareness-init', rootDoc.guid);
}); });
window.addEventListener('beforeunload', windowBeforeUnloadHandler);
return () => { return () => {
awareness.off('update', awarenessUpdate); awareness.off('update', awarenessUpdate);
conn.off('server-awareness-broadcast', awarenessBroadcast); conn.off('server-awareness-broadcast', awarenessBroadcast);
conn.off('new-client-awareness-init', newClientAwarenessInitHandler); conn.off('new-client-awareness-init', newClientAwarenessInitHandler);
window.removeEventListener('unload', windowBeforeUnloadHandler);
}; };
} }