fix: providers get disconnected after opening the setting and close it (#4429)

Co-authored-by: Alex Yang <himself65@outlook.com>
This commit is contained in:
Hongtao Lye 2023-09-21 22:22:23 +08:00 committed by GitHub
parent 1ddae40fb2
commit d5f4fbcdb5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -18,10 +18,12 @@ const workspaceActiveAtomWeakMap = new WeakMap<
const workspaceActiveWeakMap = new WeakMap<Workspace, boolean>();
// Whether the workspace has been enabled the passive effect (background)
const workspacePassiveEffectWeakMap = new WeakMap<Workspace, boolean>();
const workspacePassiveEffectWeakMap = new WeakMap<Workspace, number>();
export function enablePassiveProviders(workspace: Workspace) {
if (workspacePassiveEffectWeakMap.get(workspace) === true) {
const value = workspacePassiveEffectWeakMap.get(workspace);
if (value !== undefined && value !== 0) {
workspacePassiveEffectWeakMap.set(workspace, value + 1);
return;
}
const providers = workspace.providers.filter(
@ -31,13 +33,14 @@ export function enablePassiveProviders(workspace: Workspace) {
providers.forEach(provider => {
provider.connect();
});
workspacePassiveEffectWeakMap.set(workspace, true);
workspacePassiveEffectWeakMap.set(workspace, 1);
}
export function disablePassiveProviders(workspace: Workspace) {
if (workspacePassiveEffectWeakMap.get(workspace) !== true) {
return;
}
const value = workspacePassiveEffectWeakMap.get(workspace);
if (value && value > 0) {
workspacePassiveEffectWeakMap.set(workspace, value - 1);
if (value - 1 === 0) {
const providers = workspace.providers.filter(
(provider): provider is PassiveDocProvider =>
'passive' in provider && provider.passive === true
@ -46,6 +49,9 @@ export function disablePassiveProviders(workspace: Workspace) {
provider.disconnect();
});
workspacePassiveEffectWeakMap.delete(workspace);
}
return;
}
}
export async function waitForWorkspace(workspace: Workspace) {