refactor(system): inline startTrackEffects, endTrackEffects to effect scope

Author: johnsoncodehk

src/effect.ts

@@ -1,3 +1,4 @@
+import { activeEffectScope } from './effectScope.js';
 import { Dependency, DirtyLevels, IEffect, Link, Subscriber, System } from './system.js';
 
 export function effect(fn: () => void) {
@@ -28,9 +29,11 @@ export class Effect<T = any> implements IEffect {
 			Dependency.linkSubscriber(this, System.activeSub!);
 			return;
 		}
-		const activeEffectScopeTrackId = System.activeEffectScopeTrackId;
-		if (activeEffectScopeTrackId !== 0) {
-			Dependency.linkSubscriber(this, System.activeEffectScope!);
+		if (activeEffectScope !== undefined) {
+			const subsTail = this.subsTail;
+			if (subsTail === undefined || subsTail.trackId !== activeEffectScope.trackId) {
+				Dependency.linkSubscriber(this, activeEffectScope);
+			}
 		}
 	}
 

src/effectScope.ts

@@ -1,4 +1,6 @@
-import { DirtyLevels, IEffect, Link, Subscriber } from './system.js';
+import { DirtyLevels, IEffect, Link, Subscriber, System } from './system.js';
+
+export let activeEffectScope: IEffect | undefined = undefined;
 
 export function effectScope() {
 	return new EffectScope();
@@ -14,7 +16,7 @@ export class EffectScope implements IEffect {
 	// Subscriber
 	deps: Link | undefined = undefined;
 	depsTail: Link | undefined = undefined;
-	trackId = 0;
+	readonly trackId = ++System.lastTrackId;
 	dirtyLevel = DirtyLevels.None;
 	canPropagate = false;
 
@@ -26,11 +28,28 @@ export class EffectScope implements IEffect {
 	}
 
 	run<T>(fn: () => T) {
-		const prevSub = Subscriber.startTrackEffects(this);
+		const prevSub = activeEffectScope;
+		activeEffectScope = this;
+		this.depsTail = undefined;
+		this.dirtyLevel = DirtyLevels.None;
 		try {
 			return fn();
 		} finally {
-			Subscriber.endTrackEffects(this, prevSub);
+			if (prevSub !== undefined) {
+				activeEffectScope = prevSub;
+			} else {
+				activeEffectScope = undefined;
+			}
+			const depsTail = this.depsTail as Link | undefined;
+			if (depsTail !== undefined) {
+				if (depsTail.nextDep !== undefined) {
+					Subscriber.clearTrack(depsTail.nextDep);
+					depsTail.nextDep = undefined;
+				}
+			} else if (this.deps !== undefined) {
+				Subscriber.clearTrack(this.deps);
+				this.deps = undefined;
+			}
 		}
 	}
 

src/system.ts

@@ -41,9 +41,7 @@ export const enum DirtyLevels {
 
 export namespace System {
 	export let activeSub: IComputed | IEffect | undefined = undefined;
-	export let activeEffectScope: IEffect | undefined = undefined;
 	export let activeTrackId = 0;
-	export let activeEffectScopeTrackId = 0;
 	export let batchDepth = 0;
 	export let lastTrackId = 0;
 	export let queuedEffects: IEffect | undefined = undefined;
@@ -431,43 +429,6 @@ export namespace Subscriber {
 		sub.trackId = -sub.trackId;
 	}
 
-	export function startTrackEffects(sub: IEffect) {
-		const newVersion = system.lastTrackId + 1;
-		const prevSub = system.activeEffectScope;
-
-		system.activeEffectScope = sub;
-		system.activeEffectScopeTrackId = newVersion;
-		system.lastTrackId = newVersion;
-
-		sub.depsTail = undefined;
-		sub.trackId = newVersion;
-		sub.dirtyLevel = DirtyLevels.None;
-
-		return prevSub;
-	}
-
-	export function endTrackEffects(sub: IEffect, prevSub: IEffect | undefined) {
-		if (prevSub !== undefined) {
-			system.activeEffectScope = prevSub;
-			system.activeEffectScopeTrackId = prevSub.trackId;
-		} else {
-			system.activeEffectScope = undefined;
-			system.activeEffectScopeTrackId = 0;
-		}
-
-		const depsTail = sub.depsTail;
-		if (depsTail !== undefined) {
-			if (depsTail.nextDep !== undefined) {
-				clearTrack(depsTail.nextDep);
-				depsTail.nextDep = undefined;
-			}
-		} else if (sub.deps !== undefined) {
-			clearTrack(sub.deps);
-			sub.deps = undefined;
-		}
-		sub.trackId = -sub.trackId;
-	}
-
 	export function clearTrack(link: Link) {
 		do {
 			const dep = link.dep;