我打破了 React Hook 必须按顺序、不能在条件语句中调用的枷锁

export default function App() {

if (Math.random() > 0.5) {

useState(10000, ‘key1’);

}

const [value, setValue] = useState(0, “key2”);

return (

<button onClick={() => setValue(value + 1)}>+

{value}

);

}

复制代码

这样，通过 _hooks['key'] 来查找，就无所谓前序的 Hook 出现的任何意外情况了。

也就是说，原本的存储方式是：

_hooks: [

{ value: ‘first’, update: function },

{ value: ‘second’, update: function },

]

复制代码

改造后：

_hooks: [

key1: { value: ‘first’, update: function },

key2: { value: ‘second’, update: function },

]

复制代码

注意，数组本身就支持对象的 key 值特性，不需要改造 _hooks 的结构。

改造源码

---

来试着改造一下 Preact 源码，它的 Hook 包的位置在 hooks/src/index.js 下，找到 useState 方法：

export function useState(initialState) {

currentHook = 1;

return useReducer(invokeOrReturn, initialState, undefined);

}

复制代码

它的底层调用了 useReducer，所以新增加一个 key 参数透传下去：

• export function useState(initialState, key) {

currentHook = 1;

• return useReducer(invokeOrReturn, initialState, undefined, key);

}

复制代码

useReducer 原本是通过全局索引去获取 Hook state：

// 全局索引

let currentIndex

export function useReducer(reducer, initialState, init) {

const hookState = getHookState(currentIndex++, 2);

hookState._reducer = reducer;

return hookState._value;

}

复制代码

改造成兼容版本，有 key 的时候优先传入 key 值：

// 全局索引

let currentIndex

• export function useReducer(reducer, initialState, init, key) {

• const hookState = getHookState(key || currentIndex++, 2);

hookState._reducer = reducer;

return hookState._value;

}

复制代码

最后改造一下 getHookState 方法：

function getHookState(index, type) {

const hooks =

currentComponent.__hooks ||

(currentComponent.__hooks = {

_list: [],

_pendingEffects: [],

});

// 传入 key 值是 string 或 symbol 都可以

• if (typeof index !== ‘number’) {

• if (!hooks._list[index]) {

•  hooks._list[index] = {};
  

• }

• } else {

if (index >= hooks._list.length) {

hooks._list.push({});

}

}

// 这里天然支持 key 值取用的方式

return hooks._list[index];

}

复制代码

这里设计成传入 key 值的时候，初始化就不往数组里 push 新状态，而是直接通过下标写入即可，原本的取状态的写法 hooks._list[index] 本身就支持通过 key 从数组上取值，不用改动。

至此，改造就完成了。

来试试新用法：

export default function App() {

if (Math.random() > 0.5) {

useState(10000, ‘key1’);

}

const [value, setValue] = useState(0, ‘key2’);

return (

<button onClick={() => setValue(value + 1)}>+

{value}

);

}

复制代码

自动编译

---

事实上 React 团队也考虑过给每次调用加一个 key 值的设计，在 Dan Abramov 的 为什么顺序调用对 React Hooks 很重要？ 中已经详细解释过这个提案。

多重的缺陷导致这个提案被否决了，尤其是在遇到自定义 Hook 的时候，比如你提取了一个 useFormInput：

const valueKey = Symbol();

function useFormInput() {

const [value, setValue] = useState(valueKey);

return {

value,

onChange(e) {

setValue(e.target.value);

},

};

}

复制代码

然后在组件中多次调用它：

function Form() {

// 使用 Symbol

const name = useFormInput();

// 又一次使用了同一个 Symbol

const surname = useFormInput();

// …

return (

<>

<input {…name} />

<input {…surname} />

{/* … */}

</>

)

}

复制代码

此时这个通过 key 寻找 Hook state 的方式就会发生冲突。

但我的想法是，能不能借助 babel 插件的编译能力，实现编译期自动为每一次 Hook 调用都注入一个 key， 伪代码如下：

traverse(node) {

if (isReactHookInvoking(node)) {

addFunctionParameter(node, getUniqKey(node))

}

}

复制代码

生成这样的代码：

function Form() {

• const name = useFormInput(‘key_1’);

• const surname = useFormInput(‘key_2’);

// …

return (

<>

<input {…name} />

<input {…surname} />

{/* … */}

</>

)

}

• function useFormInput(key) {

• const [value, setValue] = useState(key);

return {

value,

onChange(e) {

setValue(e.target.value);

},

};

}

复制代码

key 的生成策略可以是随机值，也可以是注入一个 Symbol，这个无所谓，保证运行时期不会改变即可。也许有一些我没有考虑周到的地方，对此有任何想法的同学都欢迎加我微信 sshsunlight 讨论，当然单纯的交个朋友也没问题，大佬或者萌新都欢迎。

总结

---

本文只是一篇探索性质的文章：

• 介绍 Hook 实现的大概原理以及限制

• 探索出修改源码机制绕过限制的方法

其实本意是帮助大家更好的理解 Hook。

我并不希望 React 取消掉这些限制，我觉得这也是设计的取舍。

如果任何子函数，任何条件表达式中都可以调用 Hook，代码也会变得更加难以理解和维护。

如果你真的希望更加灵活的使用类似的 Hook 能力，Vue3 底层响应式收集依赖的原理就可以完美的绕过这些限制，但更加灵活的同时也一定会无法避免的增加更多维护风险。

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数前端工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年Web前端开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上前端开发知识点，真正体系化！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！

如果你觉得这些内容对你有帮助，可以扫码获取！！（备注：前端）

最后

正值金三银四招聘旺季，很多小伙伴都询问我有没有前端方面的面试题，特地整理出来赠送给大家！

资料领取方式：点击这里前往免费获取

够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。**

[外链图片转存中…(img-jIqXPWds-1713835984505)]

[外链图片转存中…(img-aDKf55Rj-1713835984506)]

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上前端开发知识点，真正体系化！

[外链图片转存中…(img-Ynp8e87Y-1713835984506)]

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！

如果你觉得这些内容对你有帮助，可以扫码获取！！（备注：前端）

[外链图片转存中…(img-rNP8comz-1713835984506)]

最后

正值金三银四招聘旺季，很多小伙伴都询问我有没有前端方面的面试题，特地整理出来赠送给大家！

资料领取方式：点击这里前往免费获取