面试之《react hooks在源码中是怎么实现的?》

发布于:2025-02-27 ⋅ 阅读:(10) ⋅ 点赞:(0)

要深入理解 React Hooks 在源码中的实现,可以从以下几个关键方面来剖析:

核心数据结构

在 React 内部,使用链表来管理每个函数组件的 Hooks。每个 Hook 对应一个节点,这些节点通过 next 指针相连。以下是简化后的 Hook 节点结构:

// 简化的 Hook 节点结构
function Hook() {
    this.memoizedState = null; // 存储当前 Hook 的状态
    this.baseState = null;
    this.baseUpdate = null;
    this.queue = null; // 存储更新队列
    this.next = null; // 指向下一个 Hook 节点
}

对于每个函数组件,React 会维护一个 fiber 对象,fiber 是 React 内部用于协调渲染的核心数据结构,其中 memoizedState 属性指向该组件 Hooks 链表的头部。

useState 的实现原理

useState 是最常用的 Hook 之一,下面是简化的 useState 实现逻辑:

// 全局变量,用于记录当前正在处理的 fiber
let currentlyRenderingFiber = null;
// 全局变量,用于记录当前正在处理的 Hook
let workInProgressHook = null;

function useState(initialState) {
    // 获取当前的 Hook 节点
    let hook;
    if (workInProgressHook === null) {
        // 如果是首次渲染,创建新的 Hook 节点
        hook = {
            memoizedState: initialState,
            queue: {
                pending: null
            },
            next: null
        };
        if (currentlyRenderingFiber.memoizedState === null) {
            // 如果是该组件的第一个 Hook,将其作为链表头部
            currentlyRenderingFiber.memoizedState = hook;
        } else {
            // 否则,将新 Hook 节点添加到链表末尾
            let lastHook = currentlyRenderingFiber.memoizedState;
            while (lastHook.next !== null) {
                lastHook = lastHook.next;
            }
            lastHook.next = hook;
        }
    } else {
        // 如果不是首次渲染,获取当前 Hook 节点
        hook = workInProgressHook;
        workInProgressHook = workInProgressHook.next;
    }

    // 处理更新队列
    let baseState = hook.memoizedState;
    let firstUpdate = hook.queue.pending;
    if (firstUpdate!== null) {
        let update = firstUpdate;
        do {
            const action = update.action;
            baseState = action(baseState);
            update = update.next;
        } while (update!== null && update!== firstUpdate);
        hook.queue.pending = null;
    }
    hook.memoizedState = baseState;

    // 返回状态和更新函数
    const setState = (action) => {
        const update = {
            action,
            next: null
        };
        if (hook.queue.pending === null) {
            update.next = update;
        } else {
            update.next = hook.queue.pending.next;
            hook.queue.pending.next = update;
        }
        hook.queue.pending = update;
        // 触发重新渲染
        scheduleUpdate();
    };

    return [baseState, setState];
}

上述代码中,首次调用 useState 时会创建一个新的 Hook 节点并添加到 Hooks 链表中,后续调用则直接获取对应的 Hook 节点。setState 函数会将更新添加到更新队列中,并触发重新渲染。

useEffect 的实现原理

useEffect 用于处理副作用,下面是简化的 useEffect 实现:

function useEffect(create, deps) {
    // 获取当前的 Hook 节点
    let hook;
    if (workInProgressHook === null) {
        // 如果是首次渲染,创建新的 Hook 节点
        hook = {
            memoizedState: null,
            next: null
        };
        if (currentlyRenderingFiber.memoizedState === null) {
            currentlyRenderingFiber.memoizedState = hook;
        } else {
            let lastHook = currentlyRenderingFiber.memoizedState;
            while (lastHook.next !== null) {
                lastHook = lastHook.next;
            }
            lastHook.next = hook;
        }
    } else {
        // 如果不是首次渲染,获取当前 Hook 节点
        hook = workInProgressHook;
        workInProgressHook = workInProgressHook.next;
    }

    // 获取上一次的依赖项
    const prevDeps = hook.memoizedState? hook.memoizedState[1] : null;
    let hasChanged = true;
    if (prevDeps!== null) {
        hasChanged = false;
        for (let i = 0; i < deps.length; i++) {
            if (prevDeps[i]!== deps[i]) {
                hasChanged = true;
                break;
            }
        }
    }

    if (hasChanged) {
        // 如果依赖项发生变化,添加副作用到待执行队列
        hook.memoizedState = [
            () => {
                const cleanUp = create();
                return cleanUp;
            },
            deps
        ];
        currentlyRenderingFiber.effects.push(hook.memoizedState[0]);
    }
}

在 useEffect 中,会比较前后两次的依赖项数组。如果依赖项发生变化,会将副作用函数添加到 fiber 的 effects 数组中,在渲染完成后执行这些副作用。

调用顺序的保证

React 严格依赖于 Hooks 的调用顺序。在每次渲染时,workInProgressHook 会依次指向 Hooks 链表中的每个节点。如果 Hooks 的调用顺序发生改变,会导致 workInProgressHook 无法正确获取对应的 Hook 节点,从而引发错误。

重新渲染和状态更新

当调用 setState 等更新函数时,会触发 scheduleUpdate 函数,该函数会标记当前 fiber 需要重新渲染。React 会重新执行函数组件,再次按顺序调用 Hooks,更新状态和副作用。
以上代码只是简化的实现,实际的 React 源码要复杂得多,包含了大量的错误处理、性能优化和兼容性处理等逻辑。但通过这些简化代码,可以理解 React Hooks 核心的实现原理。