React 18 state 状态更新函数

参考文章

把一系列 state 更新加入队列

设置组件 state 会把一次重新渲染加入队列。但有时可能会希望在下次渲染加入队列之前对 state 的值执行多次操作。为此,了解 React 如何批量更新 state 会很有帮助。

React 会对 state 更新进行批处理

在下面的示例中,可能会认为点击 "+3" 按钮会使计数器递增三次,因为它调用了 setNumber(number + 1) 三次:

js 复制代码
import { useState } from 'react';

export default function Counter() {
  const [number, setNumber] = useState(0);

  return (
    <>
      <h1>{number}</h1>
      <button onClick={() => {
        setNumber(number + 1);
        setNumber(number + 1);
        setNumber(number + 1);
      }}>+3</button>
    </>
  )
}

但是,每一次渲染的 state 值都是固定的,因此无论调用多少次 setNumber(1),在第一次渲染的事件处理函数内部的 number 值总是 0

js 复制代码
setNumber(0 + 1);
setNumber(0 + 1);
setNumber(0 + 1);

但是这里还有另外一个影响因素需要讨论。React 会等到事件处理函数中的 所有 代码都运行完毕再处理 state 更新。 这就是为什么重新渲染只会发生在所有这些 setNumber() 调用 之后 的原因。

这让你可以更新多个 state 变量------甚至来自多个组件的 state 变量------而不会触发太多的 重新渲染。但这也意味着只有在事件处理函数及其中任何代码执行完成 之后 ,UI 才会更新。这种特性也就是 批处理,它会使 React 应用运行得更快。它还会帮你避免处理只更新了一部分 state 变量的令人困惑的"半成品"渲染。

React 不会跨 多个 需要刻意触发的事件(如点击)进行批处理------每次点击都是单独处理的。请放心,React 只会在一般来说安全的情况下才进行批处理。这可以确保,例如,如果第一次点击按钮会禁用表单,那么第二次点击就不会再次提交它。

在下次渲染前多次更新同一个 state

这是一个不常见的用例,但是如果想在下次渲染之前多次更新同一个 state,可以像 setNumber(n => n + 1) 这样传入一个根据队列中的前一个 state 计算下一个 state 的 函数 ,而不是像 setNumber(number + 1) 这样传入 下一个 state 值。这是一种告诉 React "用 state 值做某事"而不是仅仅替换它的方法。

现在尝试递增计数器:

js 复制代码
import { useState } from 'react';

export default function Counter() {
  const [number, setNumber] = useState(0);

  return (
    <>
      <h1>{number}</h1>
      <button onClick={() => {
        setNumber(n => n + 1);
        setNumber(n => n + 1);
        setNumber(n => n + 1);
      }}>+3</button>
    </>
  )
}

在这里,n => n + 1 被称为 更新函数。当将它传递给一个 state 设置函数时:

  1. React 会将此函数加入队列,以便在事件处理函数中的所有其他代码运行后进行处理。
  2. 在下一次渲染期间,React 会遍历队列并给你更新之后的最终 state。
js 复制代码
setNumber(n => n + 1);
setNumber(n => n + 1);
setNumber(n => n + 1);

下面是 React 在执行事件处理函数时处理这几行代码的过程:

  1. setNumber(n => n + 1)n => n + 1 是一个函数。React 将它加入队列。
  2. setNumber(n => n + 1)n => n + 1 是一个函数。React 将它加入队列。
  3. setNumber(n => n + 1)n => n + 1 是一个函数。React 将它加入队列。

当在下次渲染期间调用 useState 时,React 会遍历队列。之前的 number state 的值是 0,所以这就是 React 作为参数 n 传递给第一个更新函数的值。然后 React 会获取上一个更新函数的返回值,并将其作为 n 传递给下一个更新函数,以此类推:

更新队列 n 返回值
n => n + 1 0 0 + 1 = 1
n => n + 1 1 1 + 1 = 2
n => n + 1 2 2 + 1 = 3

React 会保存 3 为最终结果并从 useState 中返回。

这就是为什么在上面的示例中点击"+3"正确地将值增加"+3"

如果在替换 state 后更新 state 会发生什么

这个事件处理函数会怎么样?你认为 number 在下一次渲染中的值是什么?

js 复制代码
import { useState } from 'react';

export default function Counter() {
  const [number, setNumber] = useState(0);

  return (
    <>
      <h1>{number}</h1>
      <button onClick={() => {
        setNumber(number + 5);
        setNumber(n => n + 1);
      }}>增加数字</button>
    </>
  )
}

这是事件处理函数告诉 React 要做的事情:

  1. setNumber(number + 5)number0,所以 setNumber(0 + 5)。React 将 "替换为 5" 添加到其队列中。
  2. setNumber(n => n + 1)n => n + 1 是一个更新函数。 React 将 该函数 添加到其队列中。

在下一次渲染期间,React 会遍历 state 队列:

更新队列 n 返回值
"替换为 5" 0(未使用) 5
n => n + 1 5 5 + 1 = 6

React 会保存 6 为最终结果并从 useState 中返回。

注意:setState(x) 实际上会像 setState(n => x) 一样运行,只是没有使用 n

如果在更新 state 后替换 state 会发生什么

让我们再看一个例子。你认为 number 在下一次渲染中的值是什么?

js 复制代码
import { useState } from 'react';

export default function Counter() {
  const [number, setNumber] = useState(0);

  return (
    <>
      <h1>{number}</h1>
      <button onClick={() => {
        setNumber(number + 5);
        setNumber(n => n + 1);
        setNumber(42);
      }}>增加数字</button>
    </>
  )
}

以下是 React 在执行事件处理函数时处理这几行代码的过程:

  1. setNumber(number + 5)number0,所以 setNumber(0 + 5)。React 将 "替换为 5" 添加到其队列中。
  2. setNumber(n => n + 1)n => n + 1 是一个更新函数。React 将该函数添加到其队列中。
  3. setNumber(42):React 将 "替换为 42" 添加到其队列中。

在下一次渲染期间,React 会遍历 state 队列:

更新队列 n 返回值
"替换为 5" 0(未使用) 5
n => n + 1 5 5 + 1 = 6
"替换为 42" 6(未使用) 42

然后 React 会保存 42 为最终结果并从 useState 中返回。

总而言之,以下是可以考虑传递给 setNumber state 设置函数的内容:

  • 一个更新函数 (例如:n => n + 1)会被添加到队列中。
  • 任何其他的值 (例如:数字 5)会导致"替换为 5"被添加到队列中,已经在队列中的内容会被忽略。

事件处理函数执行完成后,React 将触发重新渲染。在重新渲染期间,React 将处理队列。更新函数会在渲染期间执行,因此 更新函数必须是 纯函数 并且只 返回 结果。不要尝试从它们内部设置 state 或者执行其他副作用。在严格模式下,React 会执行每个更新函数两次(但是丢弃第二个结果)以便帮助你发现错误。

命名惯例

通常可以通过相应 state 变量的第一个字母来命名更新函数的参数:

js 复制代码
setEnabled(e => !e);
setLastName(ln => ln.reverse());
setFriendCount(fc => fc * 2);

如果你喜欢更冗长的代码,另一个常见的惯例是重复使用完整的 state 变量名称,如 setEnabled(enabled => !enabled),或使用前缀,如 setEnabled(prevEnabled => !prevEnabled)

摘要

  • 设置 state 不会更改现有渲染中的变量,但会请求一次新的渲染。
  • React 会在事件处理函数执行完成之后处理 state 更新。这被称为批处理。
  • 要在一个事件中多次更新某些 state,可以使用 setNumber(n => n + 1) 更新函数。