From 2073c18d18a3cb60701e61cc2c2b9e99472bdf2e Mon Sep 17 00:00:00 2001
From: QCQCQC <1220204124@zust.edu.cn>
Date: Sat, 2 Nov 2024 15:26:16 +0800
Subject: [PATCH] subscribeChecker

---
 src/plugins/oakTreePanel.ts | 9 +++++++++
 1 file changed, 9 insertions(+)

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