React Hooks原理

原理

import React from 'react';
import ReactDOM from 'react-dom';

interface WorkHook {
    memoriedState: any,
    next: WorkHook | undefined
}

let firstWorkHook: WorkHook = {
    memoriedState: undefined,
    next: undefined
};

let currentWorkHook = firstWorkHook;

function useReducer(reducer: Function, initArg: any, init?: Function) {
    let current: WorkHook = currentWorkHook;
    if (typeof current.memoriedState === 'undefined') {
        let initState = initArg;
        if (typeof init === 'function') {
            initState = init(initArg);
        }
        current.memoriedState = initState;
    }
    current.next = currentWorkHook = current.next ? current.next : { memoriedState: undefined, next: undefined };
    let dispatch = (action: any) => {
        current.memoriedState = reducer(current.memoriedState, action);
        render();
    }
    return [
        current.memoriedState,
        dispatch
    ]
}

function useEffect(callback: Function, dependencies?: any[]) {
    if (typeof dependencies === 'undefined') {
        return callback();
    }
    let current = currentWorkHook;
    let memoriedDependencies = current.memoriedState;
    if (typeof memoriedDependencies === 'undefined') {
        current.memoriedState = dependencies;
        callback();
    } else {
        if (!dependencies.every((item, index) => item === memoriedDependencies[index])) {
            callback();
            current.memoriedState = dependencies;
        }
    }
    current.next = currentWorkHook = current.next ? current.next : { memoriedState: undefined, next: undefined };

}

function useState(initialState: any) {
    return useReducer((state: any, action: any) => action, initialState);
}

const INCREMENT = 'INCREMENT';
const DECREMENT = 'DECREMENT';

let reducer = (number: any, action: any) => {
    switch (action.type) {
        case INCREMENT:
            return number + 1;
        case DECREMENT:
            return number - 1;
        default:
            return number;
    }
}
let init = (number: number) => {
    return number;
}
let Counter = () => {
    let [number, dispatch] = useReducer(reducer, 0, init);
    let [name, setName] = useState('計數(shù)器');
    useEffect(() => {
        alert('名字改變了');
    }, [name]);
    useEffect(() => {
        alert('數(shù)值改變了');
    }, [number]);
    useEffect(() => {
        alert('只出現(xiàn)一次');
    }, []);
    useEffect(() => {
        alert('每次都有我');
    });
    return <>
        <p>{name}:{number}</p>
        <button onClick={() => dispatch({ type: INCREMENT })}>+</button>
        <button onClick={() => dispatch({ type: DECREMENT })}>-</button>
        <button onClick={() => setName('計數(shù)器' + Date.now())}>改名字</button>
    </>
}

function render() {
    currentWorkHook = firstWorkHook;
    ReactDOM.render(<Counter />, document.getElementById('root'));
}
render();

結果

結果.gif
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
【社區(qū)內容提示】社區(qū)部分內容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發(fā)布,文章內容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內容

友情鏈接更多精彩內容