fix: initialize

toeverything · Dec 18, 2024 · 4cda4d2 · 4cda4d2
1 parent 7564ada
commit 4cda4d2
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/packages/affine/block-surface/src/view/mindmap.ts b/packages/affine/block-surface/src/view/mindmap.ts
@@ -26,6 +26,10 @@ export class MindMapView extends GfxElementModelView<MindmapElementModel> {
 
   private _initCollapseButtons() {
     const updateButtons = requestThrottledConnectedFrame(() => {
+      if (!this.isConnected) {
+        return;
+      }
+
       const visited = new Set<LocalShapeElementModel>();
 
       this.model.traverse(node => {
@@ -63,6 +67,8 @@ export class MindMapView extends GfxElementModelView<MindmapElementModel> {
     this.disposable.add(() => {
       this.model.children.unobserve(updateButtons);
     });
+
+    updateButtons();
   }
 
   private _needToUpdateButtonStyle(options: {