来来来,手摸手写一个hook

hello,这里是潇晨,今天就带着大家一起来手写一个迷你版的hooks,方便大家理解hook在源码中的运行机制,配有图解,保姆级的教程,只求同学一个小小的👍,🐶。

第一步:引入React和ReactDOM

因为我们要将jsx转变为virtual-dom,这一步分工作就交给babel吧,而jsxbabel进行词法解析之后会形成React.createElement()的调用,而React.createElement()执行之后的返回结果就是jsx对象或者叫virtual-dom

又因为我们要将我们的demo渲染到dom上,所以我们引入ReactDOM

import React from "react";
import ReactDOM from "react-dom";
第二步:我们来写一个小demo

我们定义两个状态countage,在点击的时候触发更新,让它们的值加1。

在源码中useState是保存在一个Dispatcher对象上面的,并且在mountupdate的时候取到的是不同的hooks,所以我们先暂时从Dispatcher上拿到useState,等下在来定义Dispatcher

接下来定义一个schedule函数,每次调用的时候会重新渲染组件。

function App() {
   
  let [count, setCount] = Dispatcher.useState(1);
  let [age, setAge] = Dispatcher.useState(10);
  return (
    <>
      <p>Clicked {
   count} times</p>
      <button onClick={
   () => setCount(() => count + 1)}> Add count</button>
      <p>Age is {
   age}</p>
      <button onClick={
   () => setAge(() => age + 1)}> Add age</button>
    </>
  );
}

function schedule() {
       //每次调用会重新渲染组件
  ReactDOM.render(<App />, document.querySelector("#root"));
}

schedule();
第三步:定义Dispatcher

在看这部分前,先来捋清楚fiberhookupdate的关系,看图:

image-20211129105128673

Dispatcher是什么:Dispatcher在源码中就是一个对象,上面存放着各种各样的hooks,在mountupdate的时候会使用过不同的Dispatcher,来看看在源码中Dispatcher是什么样子:

在调用useState之后,会调用一个resolveDispatcher的函数,这个函数调用之后会返回一个dispatcher对象,这个对象上就有useState等钩子。

image-20211126164214374

那我们来看看这个函数做了啥事情,这个函数比较简单,直接从ReactCurrentDispatcher对象上拿到current,然后返回出来的这个current就是dispatcher,那这个ReactCurrentDispatcher又是个啥?别急,继续在源码中来找一下。

image-20211126164903336

在源码中有这样一段代码,如果是在正式环境中,分为两种情况

  1. 如果满足 current === null || current.memoizedState === null,说明我们处于首次渲染的时候,也就是mount的时候,其中current就是我们fiber节点,memoizedState保存了fiberhook,也就是说在应用首次渲染的时候,current fiber是不存在的,我们还没有创造出任何fiber节点,或者存在某些fiber,但是上面没有构建相应的hook,这个时候就可以认为是处于首次渲染的时候,我们取到的是HooksDispatcherOnMount
  2. 如果不满足 current === null || current.memoizedState === null,就说明我们处于更新阶段,也就是update的时候,我们取到的是HooksDispatcherOnUpdate
if (__DEV__) {
   
    if (current !== null && current.memoizedState !== null) {
   
      ReactCurrentDispatcher.current = HooksDispatcherOnUpdateInDEV;
    } else if (hookTypesDev !== null) {
   
      ReactCurrentDispatcher.current = HooksDispatcherOnMountWithHookTypesInDEV;
    } else {
   
      ReactCurrentDispatcher.current = HooksDispatcherOnMountInDEV;
    }
  } else {
   
    ReactCurrentDispatcher.current =
      current === null || current.memoizedState ===</
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值