【框架实现】深入scheduler调度系统实现机制

在学习了computed和watch的代码之后,我们会发现,computed和watch中都包含调度器scheduler的概念。完整的来说,它应该是调度系统;

调度系统包含两部分实现:

  1. lazy:懒执行;
  2. scheduler:调度器;

懒执行

参考vue3源码packages/reactivity/src/effect.ts中183-185行:

ts 复制代码
// 如果lazy是真,不会直接执行run函数;
// 当我们setter触发时,才会执行watch
if (!options || !options.lazy) {
    _effect.run()
}

在我们的项目vue-mini中的effect.ts修改:

ts 复制代码
export interface ReactiveEffectOptions {
  lazy?: boolean;
  scheduler?: EffectScheduler;
}

export function effect<T = any>(fn: () => T, options?: ReactiveEffectOptions) {
  const _effect = new ReactiveEffect(fn);
  if (!options || !options.lazy) {
    _effect.run();
  }
}

修改完成之后,写一个测试实例;

js 复制代码
 <script>
    const { reactive, effect } = Vue;

    const obj = reactive({
      count: 1,
    });

    effect(
      () => {
        console.log(obj.count);
      },
      {
        lazy: true,
      }
    );
    obj.count = 2;
    console.log("代码结束");
</script>

当lazy是true的时候,只打印出代码结束;

当lazy是false的时候,打印出1和2:

调度器

调度器分为两个部分:

  1. 控制执行顺序;
  2. 控制执行规则;

控制执行顺序

我们上面的测试实例:

js 复制代码
 <script>
    const { reactive, effect } = Vue;

    const obj = reactive({
      count: 1,
    });

    effect(
      () => {
        console.log(obj.count);
      },
    );
    obj.count = 2;
    console.log("代码结束");
</script>

最后打印输出的顺序如下图所示:

如果我们想要输出的顺序变成下面这样呢?

1

代码结束

2

是否能用scheduler做到?我们来创建一个测试实例运行一下!

ts 复制代码
const { reactive, effect } = Vue

const obj = reactive({
  count: 1
})

effect(
  () => {
    console.log(obj.count)
  },
  {
    scheduler() {
      setTimeout(() => {
        console.log(obj.count)
      })
    }
  }
)
obj.count = 2
console.log('代码结束')

打印的顺序就变成了1、代码结束、2了!

这是为什么呢? 会想到我们自己实现的triggerEffect方法,其中有scheduler的判断;当我们在watch中传入scheduler时就不会去执行run方法了,scheduler方法是一个异步任务,只有等所有的同步任务完成后才会触发;也就是说scheduler会影响代码的执行顺序

js 复制代码
// 触发指定依赖
export function triggerEffect(effect: ReactiveEffect) {
  // 判读是否有scheduler
  if (effect.scheduler) {
    effect.scheduler();
  } else {
    effect.run();
  }
}

那现在为我们的项目vue-mini去增加scheduler相关的逻辑; 在shared/index.ts中增加一个extend方法;

ts 复制代码
export const extend = Object.assign;

修改effect.ts方法,增加合并_effect和options的逻辑:

ts 复制代码
export function effect<T = any>(fn: () => T, options?: ReactiveEffectOptions) {
  const _effect = new ReactiveEffect(fn);
  if (options) {
    // 合并_effect和options
    extend(_effect, options);
  }
  if (!options || !options.lazy) {
    _effect.run();
  }
}

运行上面相同的实例,会发现打印的顺序改变了;

ts 复制代码
const { reactive, effect } = Vue

const obj = reactive({
  count: 1
})

effect(
  () => {
    console.log(obj.count)
  },
  {
    scheduler() {
      setTimeout(() => {
        console.log(obj.count)
      })
    }
  }
)
obj.count = 2
console.log('代码结束')

控制执行规则

创建一个新的测试实例:

ts 复制代码
const { reactive, effect } = Vue

const obj = reactive({
  count: 1
})

effect(() => {
  console.log(obj.count)
})
obj.count = 2
obj.count = 3

这次打印出1、2、3;最后obj.count都会被修改成3,那我们能跳过2的流程吗?在vue3的源码中scheduler.ts文件中有一个queuePreFlushCb方法,它是用来控制scheduler的执行规则的,现在我们来看一下它是怎么运行的;

首先要去vue3源码中export queuePreFlushCb方法,重新build一遍;然后修改我们的测试实例,再运行一下;

ts 复制代码
const { reactive, effect, queuePreFlushCb } = Vue

const obj = reactive({
  count: 1
})

effect(
  () => {
    console.log(obj.count)
  },
  {
    scheduler() {
      queuePreFlushCb(() => console.log(obj.count))
    }
  }
)
obj.count = 2
obj.count = 3

queueFlush方法是关键

ts 复制代码
function queueFlush() {
  // 此时isFlushing和isFlushPending都为false
  if (!isFlushing && !isFlushPending) {
    isFlushPending = true
    // resolvedPromise是Promise.resolve(),flushJobs都变成异步微任务;
    currentFlushPromise = resolvedPromise.then(flushJobs)
  }
}

接着我们来实现scheduler的控制执行规则功能; 新建packages/runtime-core/src/scheduler.ts;

ts 复制代码
let isFlushPending = false;
const pendingPreFlushCbs: Function[] = [];
let currentFlushPromise: Promise<void> | null = null;
const resolvedPromise = Promise.resolve() as Promise<any>;

export function queuePreFlushCb(cb: Function) {
  queueCb(cb, pendingPreFlushCbs);
}

export function queueCb(cb: Function, pendingQueue: Function[]) {
  pendingQueue.push(cb);
  queueFlush();
}

export function queueFlush() {
  if (!isFlushPending) {
    isFlushPending = true;
    currentFlushPromise = resolvedPromise.then(flushJobs);
  }
}

function flushJobs() {
  isFlushPending = false;
  flushPostFlushCbs();
}

export function flushPostFlushCbs() {
  if (pendingPreFlushCbs.length) {
    let activePostFlushCbs = [...new Set(pendingPreFlushCbs)];
    pendingPreFlushCbs.length = 0;

    for (
      let postFlushIndex = 0;
      postFlushIndex < activePostFlushCbs.length;
      postFlushIndex++
    ) {
      activePostFlushCbs[postFlushIndex]();
    }
  }
}

记得还有两个地方需要export queuePreFlushCb方法;

相关推荐
来一碗刘肉面19 分钟前
React - ajax 配置代理
前端·react.js·ajax
bin915329 分钟前
DeepSeek 助力 Vue 开发:打造丝滑的二维码生成(QR Code)
前端·javascript·vue.js·ecmascript·deepseek
@LitterFisher33 分钟前
Excell 代码处理
前端·javascript·excel
m0_748230942 小时前
Redis 通用命令
前端·redis·bootstrap
YaHuiLiang2 小时前
一切的根本都是前端“娱乐圈化”
前端·javascript·代码规范
ObjectX前端实验室4 小时前
个人网站开发记录-引流公众号 & 谷歌分析 & 谷歌广告 & GTM
前端·程序员·开源
CL_IN4 小时前
企业数据集成:实现高效调拨出库自动化
java·前端·自动化
浪九天5 小时前
Vue 不同大版本与 Node.js 版本匹配的详细参数
前端·vue.js·node.js
qianmoQ5 小时前
第五章:工程化实践 - 第三节 - Tailwind CSS 大型项目最佳实践
前端·css
尚学教辅学习资料6 小时前
基于SpringBoot+vue+uniapp的智慧旅游小程序+LW示例参考
vue.js·spring boot·uni-app·旅游