fix: don't clear elementsCache on cleanup (#830)

TanStack · Sep 4, 2024 · 9627ecd · 9627ecd
1 parent 8409909
commit 9627ecd
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/packages/virtual-core/src/index.ts b/packages/virtual-core/src/index.ts
@@ -370,7 +370,10 @@ export class Virtualizer<
     }
 
     return {
-      disconnect: () => get()?.disconnect(),
+      disconnect: () => {
+        get()?.disconnect()
+        _ro = null
+      },
       observe: (target: Element) =>
         get()?.observe(target, { box: 'border-box' }),
       unobserve: (target: Element) => get()?.unobserve(target),
@@ -444,10 +447,9 @@ export class Virtualizer<
   private cleanup = () => {
     this.unsubs.filter(Boolean).forEach((d) => d!())
     this.unsubs = []
+    this.observer.disconnect()
     this.scrollElement = null
     this.targetWindow = null
-    this.observer.disconnect()
-    this.elementsCache.clear()
   }
 
   _didMount = () => {
@@ -477,6 +479,10 @@ export class Virtualizer<
         this.targetWindow = this.scrollElement?.window ?? null
       }
 
+      this.elementsCache.forEach((cached) => {
+        this.observer.observe(cached)
+      })
+
       this._scrollToOffset(this.getScrollOffset(), {
         adjustments: undefined,
         behavior: undefined,