vue3-yuanma/packages/observer/src/computed.ts
2019-05-29 18:44:50 +08:00

56 lines
1.5 KiB
TypeScript

import { effect } from './index'
import { ReactiveEffect, activeReactiveEffectStack } from './effect'
import { knownValues } from './value'
export interface ComputedValue<T> {
readonly value: T
readonly effect: ReactiveEffect
}
export function computed<T, C = null>(
getter: (this: C, ctx: C) => T,
context?: C
): ComputedValue<T> {
let dirty: boolean = true
let value: any = undefined
const runner = effect(() => getter.call(context, context), {
lazy: true,
scheduler: () => {
dirty = true
}
})
// mark effect as computed so that it gets priority during trigger
runner.computed = true
const computedValue = {
// expose effect so computed can be stopped
effect: runner,
get value() {
if (dirty) {
value = runner()
dirty = false
}
// When computed effects are accessed in a parent effect, the parent
// should track all the dependencies the computed property has tracked.
// This should also apply for chained computed properties.
trackChildRun(runner)
return value
}
}
knownValues.add(computedValue)
return computedValue
}
function trackChildRun(childRunner: ReactiveEffect) {
const parentRunner =
activeReactiveEffectStack[activeReactiveEffectStack.length - 1]
if (parentRunner) {
for (let i = 0; i < childRunner.deps.length; i++) {
const dep = childRunner.deps[i]
if (!dep.has(parentRunner)) {
dep.add(parentRunner)
parentRunner.deps.push(dep)
}
}
}
}