feat(system): add createRenderEffectWithCleaning function

swordev · May 15, 2022 · 1678375 · 1678375
1 parent f001a37
commit 1678375
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/.changeset/cool-sloths-serve.md b/.changeset/cool-sloths-serve.md
@@ -0,0 +1,5 @@
+---
+"@suid/system": minor
+---
+
+Add `createRenderEffectWithCleaning` function
diff --git a/packages/system/src/createRenderEffectWithCleaning.ts b/packages/system/src/createRenderEffectWithCleaning.ts
@@ -0,0 +1,15 @@
+import { createRenderEffect, onCleanup } from "solid-js";
+
+function createRenderEffectWithCleaning(effect: () => () => void) {
+  let lastCleanup: (() => void) | undefined;
+  onCleanup(() => lastCleanup?.());
+  createRenderEffect<(() => void) | undefined>((prevCleanup) => {
+    if (prevCleanup) {
+      lastCleanup = undefined;
+      prevCleanup?.();
+    }
+    return (lastCleanup = effect());
+  });
+}
+
+export default createRenderEffectWithCleaning;