useState与useReducer源码浅析

2023-12-09 00:42:50

useState 源码分析

useState函数的源码位于react-dom/cjs/react-dom.development.js文件中。其基本结构如下：

export function useState(initialState) {
  const hook = getOrCreateHook();
  if (hook !== null) {
    return hook;
  }

  const state = {
    value: typeof initialState === 'function' ? initialState() : initialState,
    next: null
  };

  hook = {
    state: state,
    queue: [],
    next: null
  };

  updateQueue(queue1, action);

  return hook.state.value;
}

函数首先判断当前是否有hook，如果没有，则创建一个新的hook。hook是一个对象，包含了state、queue和next属性。state属性是状态值，queue属性是更新队列，next属性指向下一个hook。

如果存在hook，则直接返回hook。

接下来，函数创建了一个state对象，该对象包含了value和next属性。value属性是状态值，next属性指向下一个state对象。

然后，函数创建了一个hook对象，该对象包含了state、queue和next属性。state属性指向刚创建的state对象，queue属性是更新队列，next属性指向下一个hook。

接下来，函数将当前的更新操作添加到更新队列中。

最后，函数返回hook.state.value，即当前的状态值。

useReducer 源码分析

useReducer函数的源码位于react-dom/cjs/react-dom.development.js文件中。其基本结构如下：

export function useReducer(reducer, initialState, init) {
  const hook = getOrCreateHook();
  if (hook !== null) {
    return hook;
  }

  const state = {
    value: typeof initialState === 'function' ? initialState(init) : initialState,
    next: null
  };

  hook = {
    state: state,
    queue: [],
    next: null
  };

  updateQueue(queue1, action);

  return hook.state.value;
}

如果存在hook，则直接返回hook。

接下来，函数创建了一个state对象，该对象包含了value和next属性。value属性是状态值，next属性指向下一个state对象。

然后，函数创建了一个hook对象，该对象包含了state、queue和next属性。state属性指向刚创建的state对象，queue属性是更新队列，next属性指向下一个hook。

接下来，函数将当前的更新操作添加到更新队列中。

最后，函数返回hook.state.value，即当前的状态值。