响应系统
// 用一个全局变量存储被注册的副作用函数
let activeEffect;
// effect 栈(避免嵌套副作用函数发生错乱)
const effectStack = [];
// 存储副作用函数的桶
const bucket = new WeakMap();
const data = {
foo: 1,
bar: 2,
};
const obj = new Proxy(data, {
//拦截读取操作
get(target, key) {
//将副作用函数 activeEffect 添加到存储副作用函数的桶中
track(target, key);
// 返回属性值
return target[key];
},
//拦截设置操作
set(target, key, newVal) {
//设置属性值
target[key] = newVal;
ƒ
// 把副作用函数从桶里取出并执行
trigger(target, key);
},
});
//定义一个任务队列
const jobQueue = new Set()
// 使用 Promise.resolve()创建一个 promise 实例,我们用它将一个任务添加到微任务队列
const p = Promise.resolve() //一个标志代表是否正在刷新队列
let isFlushing = false
function flushJob() { // 如果队列正在刷新, 则什么都不做
if (isFlushing) return // 设置为 true, 代表正在刷新
isFlushing = true
// 在微任务队列中刷新 jobQueue 队列
p.then(() => {
jobQueue.forEach(job => job())
}).finally(() => { //结束后重置 isFlushing
isFlushing = false
})
}
// effect(() => {
// obj.foo + obj.bar
// }, {
// scheduler(fn) {
// /// 每次调度时, 将副作用函数添加到 jobQueue 队列中
// jobQueue.add(fn)
// // 调用 flushJob 刷新队列
// flushJob()
// }
// })
//在get 拦截函数内调用 track 函数追踪变化
function track(target, key) {
// 没有 activeEffect,直接 return
if (!activeEffect) return;
let depsMap = bucket.get(target);
if (!depsMap) {
bucket.set(target, (depsMap = new Map()));
}
let deps = depsMap.get(key);
if (!deps) {
depsMap.set(key, (deps = new Set()));
}
// 把当前激活的副作用函数添加到依赖集合deps中
deps.add(activeEffect);
// 将其添加到activeEffect.deps数组中
activeEffect.deps.push(deps);
}
//在set 拦截函数内调用 trigger 函数触发变化
function trigger(target, key) {
const depsMap = bucket.get(target);
if (!depsMap) return;
const effects = depsMap.get(key);
const effectsToRun = new Set();
effects && effects.forEach(effect => {
// 如果trigger触发执行的副作用函数与当前正在执行的副作用函数相同,则不触发执行(避免无限递归)
if (effect !== activeEffect) {
effectsToRun.add(effect);
}
});
effectsToRun && effectsToRun.forEach((effectFn) => {
if (effectFn.options.scheduler) {
effectFn.options.scheduler(effectFn); //新增
} else {
effectFn()
}
});
}
function effect(fn, options = {}) {
const effectFn = () => {
// debugger
// 调用cleanup 函数清除依赖
cleanup(effectFn);
// 当 effectFn 执行时,将其设置为当前激活的副作用函数
activeEffect = effectFn;
// 在调用副作用函数之前,将当前副作用函数压入 effectStack 栈中
effectStack.push(effectFn);
const res = fn();
// 当 effectFn 执行完毕,将其从 effectStack 中移除
effectStack.pop();
activeEffect = effectStack[effectStack.length - 1];
return res
};
// 将options挂载到effectFn上
effectFn.options = options;
// activeEffect.deps 用来存储所有与该副作用函数相关联的依赖集合 effectFn.deps[]
effectFn.deps = [];
// 只有非lazy的时候执行副作用函数
if (!options.lazy) {
effectFn();
}
return effectFn
}
// 函数清除依赖
function cleanup(effectFn) {
//遍历 effectFn.deps 数组
for (let i = 0; i < effectFn.deps.length; i++) { // deps 是依赖集合
const deps = effectFn.deps[i]; //将 effectFn 从依赖集合中移除
deps.delete(effectFn);
} // 最后需要重置 effectFn.deps 数组 effectFn.deps.length=0; }
}
// 计算属性
function computed(getter) {
let value
let dirty = true
// debugger
// 把getter作为副作用函数,创建一个lazy的effect
const effectFn = effect(getter, {
lazy: true,
scheduler: () => {
dirty = true
// 当计算属性依赖发生变化时,手动调用trigger函数触发响应
trigger(obj, 'value')
}
})
const obj = {
// 当读取value时才执行effectFn
get value() {
if (dirty) {
value = effectFn();
dirty = false
}
// 当读取value时,手动调用 track 函数进行追踪
track(obj, 'value')
return value
}
}
return obj
}
// watch 函数
function watch(source, cb) {
let getter
if (typeof source === 'function') {
getter = source
} else {
getter = () => traverse(source)
}
//定义旧值与新值
let oldValue, newValue
// 使用 effect 注册副作用函数时, 开启 lazy 选项, 并把返回值存储到 effectFn 中以便后续手动调用
const effectFn = effect(
() => getter(), {
lazy: true,
scheduler() {
// 在scheduler 中重新执行副作用函数, 得到的是新值 newValue = effectFn()// 将旧值和新值作为回调函数的参数
cb(newValue, oldValue) // 更新旧值, 不然下一次会得到错误的旧值
oldValue = newValue
}
}
)
// 手动调用副作用函数,拿到的值就是旧值
oldValue = effectFn()
}
const sumRes = computed(() => obj.foo + obj.bar);
effect(() => {
console.log(sumRes.value);
})
obj.foo++
Last updated