「深入浅出」主流前端框架更新批处理方式

377次阅读  |  发布于2年以前

一 背景

大家好， 一提到更新，是前端框架中一个老生常谈的问题，这些知识也是在面试中，面试官比较喜欢问的，那么在不同的技术框架背景下，处理更新的手段各不相同，今天我们来探讨一下，主流的前端框架批量处理的方式，和其内部的实现原理。

通过今天的学习，你将收获这些内容：

1 一次 vue 案例

首先来想一个问题。比如在 vue 中一次更新中。

<template>
   <div>
       姓名： {{ name }}
       年龄： {{ age }}
       <button @click="handleClick" >点击</button>
   </div>
</template>

<script>
export default {
    data(){
        return {
            age:0,
            name:''
        }
    },
    methods:{
        handleClick(){
            this.name = 'alien'
            this.age = 18
        }
    }
}
</script>

如上是一个非常简单的逻辑代码，点击按钮，会触发 name 和 age 的更新。那么首先想一个问题就是：

那么暴露的问题就是，我们在 handleClick 中，同时改变了 name 和 age 属性，那么按照正常情况下，会分别触发 name 和 age 的 set，那么如果不做处理，那么会让渲染 watcher 执行两次，结果就是组件会 update 两次，但是结果是这样的吗？

结果是：vue 底层通过批量处理，只让组件 update 一次。

2 一次 react 案例

上面介绍了在 vue 中更新批处理的案例之后，我们来看一下在 react 中的批量更新处理。把上述案例用 react 来实现一下：

function Index(){
    const [ age , setAge ] = React.useState(0)
    const [ name, setName ] = React.useState('')
    return <div>
       姓名： {name}
       年龄： {age}
       <button onClick={()=>{
          setAge(18)
          setName('alien')
       }}
       >点击</button>
    </div>
}

点击按钮，触发更新，会触发两次 useState 的更新函数。那么 React 的更新流程大致是这样的。

那么按常理来说，Index 组件会执行两次。可事实是只执行一次 render。

3 批量处理意义

通过上面的案例说明在主流框架中，对于更新都采用批处理。一次上下文中的 update 会被合并成一次更新。那么为什么要进行更新批处理呢？

批处理主要是出于对性能方面的考虑，这里拿 react 为例子，看一下批处理前后的对比情况：

例子一：假设没有批量更新：

/ ------ js 层面 ------

/ ------ 浏览器渲染 ------

我们可以看到如果没有批量更新处理，那么会多走很多步骤，包括 render 阶段 ，commit 阶段，dom 的更新等，这些都会造成性能的浪费，接下来看一下有批量更新的情况。

例子二：存在批量更新。

/ ------ js 层面 ------

/ ------ 浏览器渲染 ------

从上面可以直观看到更新批处理的作用了，本质上在 js 的执行上下文上优化了很多步骤，减少性能开销。

二 简述宏任务和微任务

在正式讲批量更新之前，先来温习一下宏任务和微任务，这应该算是前端工程师必须掌握的知识点。

所谓宏任务，我们可以理解成，<script> 标签中主代码执行，一次用户交互（比如触发了一次点击事件引起的回调函数），定时器 setInterval ，延时器 setTimeout 队列， MessageChannel 等。这些宏任务通过 event loop，来实现有条不紊的执行。

例如在浏览器环境下，宏任务的执行并不会影响到浏览器的渲染和响应。我们来做个实验。

function Index(){
    const [ number , setNumber ] = useState(0)
    useEffect(()=>{
        let timer
        function run(){
            timer = setTimeout(() => {
                console.log('----宏任务执行----')
                run()
            }, 0)
        }
        run()
        return () => clearTimeout(timer)
    },[])
    return <div>
        <button onClick={() => setNumber(number + 1 )}  >点击{number}</button>
    </div>
}

如上简单的 demo 中，通过递归调用 run 函数，让 setTimeout 宏任务反复执行。

这种情况下 setTimeout 执行并不影响点击事件的执行和页面的正常渲染。

什么是微任务呢 ？

那么我们再来分析一下微任务，在 js 执行过程中，我们希望一些任务，不阻塞代码执行，又能让该任务在此轮 event loop 执行完毕，那么就引入了一个微任务队列的概念了。

微任务相比宏任务有如下特点：

常见的微任务，有 Promise， queueMicrotask ，浏览器环境下的 MutationObserver ，node 环境下 process.nextTick 等。

我们同样做个实验看一下微任务：

function Index(){
    const [ number , setNumber ] = useState(0)
    useEffect(()=>{
        function run(){
            Promise.resolve().then(()=>{
                run()
            })
        }
        run()
    },[])
    return <div>
        <button onClick={() => setNumber(number + 1 )}  >点击{number}</button>
    </div>
}

三 微任务｜宏任务实现批量更新

讲完了宏任务和微任务，继续来看第一种批量更新的实现，就是基于宏任务 和 微任务 来实现。

先来描述一下这种方式，比如每次更新，我们先并不去立即执行更新任务，而是先把每一个更新任务放入一个待更新队列 updateQueue 里面，然后 js 执行完毕，用一个微任务统一去批量更新队列里面的任务，如果微任务存在兼容性，那么降级成一个宏任务。这里优先采用微任务的原因就是微任务的执行时机要早于下一次宏任务的执行。

典型的案例就是 vue 更新原理，vue.$nextTick原理 ，还有 v18 中 scheduleMicrotask 的更新原理。

以 vue 为例子我们看一下 nextTick 的实现：

runtime-core/src/scheduler.ts

const p = Promise.resolve() 
/* nextTick 实现，用微任务实现的 */
export function nextTick(fn?: () => void): Promise<void> {
  return fn ? p.then(fn) : p
}

再看看 react v18 里面的实现。

react-reconciler/src/ReactFiberWorkLoop/ensureRootIsScheduled

function ensureRootIsScheduled(root, currentTime) {
     /* 省去没有必要的逻辑 */
     if (newCallbackPriority === SyncLane) {
        /* 支持微任务 */
        if (supportsMicrotasks) {
            /* 通过微任务处理 */
            scheduleMicrotask(flushSyncCallbacks);
        }
     }
}

接下里看一下 scheduleMicrotask 是如何实现的。

/* 向下兼容 */
var scheduleMicrotask = typeof queueMicrotask === 'function' ? queueMicrotask : typeof Promise !== 'undefined' ? function (callback) {
  return Promise.resolve(null).then(callback).catch(handleErrorInNextTick);
} : scheduleTimeout;

scheduleMicrotask 也是用的 Promise.resolve ，还有一个 setTimeout 向下兼容的情况。

大致实现流程图如下所示：

4.jpeg

接下来模拟一下，这个方式的实现。

class Scheduler {
    constructor(){
        this.callbacks = []
        /* 微任务批量处理 */
        queueMicrotask(()=>{
            this.runTask()
        })
    }
    /* 增加任务 */
    addTask(fn){
        this.callbacks.push(fn)
    }
    runTask(){
        console.log('------合并更新开始------')
        while(this.callbacks.length > 0){
            const cur = this.callbacks.shift()
            cur()
        }
        console.log('------合并更新结束------')
        console.log('------开始更新组件------')
    }
}
function nextTick(cb){
    const scheduler = new Scheduler()
    cb(scheduler.addTask.bind(scheduler))
}

/* 模拟一次更新 */
function mockOnclick(){
   nextTick((add)=>{
       add(function(){
           console.log('第一次更新')
       })
       console.log('----宏任务逻辑----')
       add(function(){
        console.log('第二次更新')
       })
   })
}

mockOnclick()

我们来模拟一下具体实现细节：

看一下打印效果：

3.jpeg

四 可控任务实现批量更新

上述介绍了通过微任务的方式实现了批量更新，还有一种方式，通过拦截把任务变成可控的，典型的就是 React v17 之前的 batchEventUpdate 批量更新。这种情况的更新来源于对事件进行拦截，比如 React 的事件系统。

以 React 的事件批量更新为例子，比如我们的 onClick ，onChange 事件都是被 React 的事件系统处理的。外层用一个统一的处理函数进行拦截。而我们绑定的事件都是在该函数的执行上下文内部被调用的。

那么比如在一次点击事件中触发了多次更新。本质上外层在 React 事件系统处理函数的上下文中，这样的情况下，就可以通过一个开关，证明当前更新是可控的，可以做批量处理。接下来 React 就用一次就可以了。

来看一下 React 的底层实现逻辑：

react-dom/src/events/ReactDOMUpdateBatching.js

export function batchedEventUpdates(fn, a) {
  /* 开启批量更新  */
  const prevExecutionContext = executionContext;
  executionContext |= EventContext;
  try {
    /* 这里执行了的事件处理函数， 比如在一次点击事件中触发setState,那么它将在这个函数内执行 */
    return fn(a);
  } finally {
    /* try 里面 return 不会影响 finally 执行  */
    /* 完成一次事件，批量更新  */
    executionContext = prevExecutionContext;
    if (executionContext === NoContext) {
        /* 立即执行更新。  */
      flushSyncCallbackQueue();
    }
  }
}

在 React 事件执行之前通过 isBatchingEventUpdates=true 打开开关，开启事件批量更新，当该事件结束，再通过 isBatchingEventUpdates = false;关闭开关，然后在 scheduleUpdateOnFiber 中根据这个开关来确定是否进行批量更新。

比如一次点击事件中：

const [ age , setAge ] = React.useState(0)
const [ name, setName ] = React.useState('')
const handleClick=()=>{
    setAge(18)
    setName('alien')
}

我们用一幅流程图来描述一下原理。

5.jpeg

接下来我们模拟一下具体的实现：

<body>  
    <button onclick="handleClick()" >点击</button>
</body>
<script>
  let  batchEventUpdate = false 
  let callbackQueue = []

  function flushSyncCallbackQueue(){
      console.log('-----执行批量更新-------')
      while(callbackQueue.length > 0 ){
          const cur = callbackQueue.shift()
          cur()
      }
      console.log('-----批量更新结束-------')
  }

  function wrapEvent(fn){
     return function (){
         /* 开启批量更新状态 */
        batchEventUpdate = true
        fn()
        /* 立即执行更新任务 */
        flushSyncCallbackQueue()
        /* 关闭批量更新状态 */
        batchEventUpdate = false
     }
  }

  function setState(fn){
      /* 如果在批量更新状态下，那么批量更新 */
      if(batchEventUpdate){
          callbackQueue.push(fn)
      }else{
          /* 如果没有在批量更新条件下，那么直接更新。 */
          fn()
      }
  }

  function handleClick(){
      setState(()=>{
          console.log('---更新1---')
      })
      console.log('上下文执行')
      setState(()=>{
          console.log('---更新2---')
      })
  }
  /* 让 handleClick 变成可控的  */
  handleClick = wrapEvent(handleClick)


</script>

打印结果：

6.jpg

分析一下核心流程：

五 总结

本章节介绍了主流框架实现更新批处理的方式。送人玫瑰，手有余香，希望看完觉的有收获的同学，可以给笔者点赞 ➕ 关注一波 ，以此鼓励我继续创作前端硬文。

参考资料

Copyright© 2013-2020

All Rights Reserved 京ICP备2023019179号-8