subscribeChecker

This commit is contained in:
Pan Qiancheng 2024-11-02 15:26:16 +08:00
parent 3061840097
commit 2073c18d18
1 changed files with 9 additions and 0 deletions

View File

@ -16,12 +16,15 @@ import { CheckerInfo, TriggerInfo } from '../types';
import { import {
getCheckerCountByEntity, getCheckerCountByEntity,
getCheckersInfoByEntity, getCheckersInfoByEntity,
subscribeChecker,
} from '../utils/checkers'; } from '../utils/checkers';
class OakTreeDataProvider implements vscode.TreeDataProvider<TreeItem> { class OakTreeDataProvider implements vscode.TreeDataProvider<TreeItem> {
private disposeGlobal: (() => void) | null = null; private disposeGlobal: (() => void) | null = null;
private disposeComponentSub: (() => void) | null = null; private disposeComponentSub: (() => void) | null = null;
private disposeTriggerSub: (() => void) | null = null; private disposeTriggerSub: (() => void) | null = null;
// checker的更新
private disposeCheckerSub: (() => void) | null = null;
private showAllEntities: boolean = true; // 控制是否显示全部实体类 private showAllEntities: boolean = true; // 控制是否显示全部实体类
// 切换显示全部实体类的方法 // 切换显示全部实体类的方法
@ -59,6 +62,9 @@ class OakTreeDataProvider implements vscode.TreeDataProvider<TreeItem> {
this.disposeTriggerSub = subscribeTrigger(() => { this.disposeTriggerSub = subscribeTrigger(() => {
this.refresh(); this.refresh();
}); });
this.disposeCheckerSub = subscribeChecker(() => {
this.refresh();
});
} }
getTreeItem( getTreeItem(
@ -156,6 +162,9 @@ class OakTreeDataProvider implements vscode.TreeDataProvider<TreeItem> {
if (this.disposeTriggerSub) { if (this.disposeTriggerSub) {
this.disposeTriggerSub(); this.disposeTriggerSub();
} }
if (this.disposeCheckerSub) {
this.disposeCheckerSub();
}
} }
} }