源码:
// 为了可以储存多个值
let state = [];
// 存储改变值的方法
let setters = [];
// 根据这个下标查找对应的值和改变值的方法
let stateIndex = 0;
// 函数主体
function useState(initialState) {
// 1. 因为useState可以调用多次,单独一个变量无法存储多个值,所以使用数组进行存储
/*
* 每次改变值都会重新渲染,每次渲染都会重新调用useState方法,
* 这个判断是为了组件渲染前是否有存储过值,如果有就取出来,如果没有就取初始传递的值
*/
state[stateIndex] = state[stateIndex] ? state[stateIndex] : initialState;
// 2. 将改变值的方法存储到数组中
setters.push(createSetter(stateIndex))
// 3. 根据下标在存储值的数组查出值,根据下标在存储改变方法的数组查出方法
let value = state[stateIndex]
let setter = setters[stateIndex]
// 4. useState是将值和改变值的方法存储到数组中,所以在组件中每次使用useState方法,就会将值和方法存起来,每次存起来存储的数组下标就会+1
stateIndex++;
// 5. useState方法最终返回一个数组,数组有两个参数,一个是值,一个是改变值的方法
return [value, setter]
}
/*
这里解释一下为什么要以闭包的形式将改变值的方法存储到数组中;
闭包有个特性就是当前代码执行完毕,闭包中的参数不会被垃圾回收机制所回收
所以我们通过闭包这种形式将下标存储到闭包中;
setters存储改变值的方法,当我们在组件中想要改变值的时候,只需要将新值传到方法中,
也就是createSetter返回的方法,createSetter这个方法将新值,根据之前闭包保存下来的下标,
去更新state中对应的值;
*/
function createSetter(index) {
return function (newState) {
state[index] = newState;
render()
}
}
// 这里的render方法用于重新渲染组件
function render(){
// 这里将stateIndex重置为0,是为了每次重新渲染下标都要从0开始
// 如果不清0,那么下标就会一直增加下去,就会出现问题
stateIndex = 0
effectIndex = 0
ReactDOM.render(<StateYM/>, document.getElementById('root'))
}