ztr/TelegramPWA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Refactoring] Teact: Simplify memo

Browse Source
This commit is contained in:
Alexander Zinchuk 2022-02-25 22:52:26 +02:00
parent bd5f280e29
commit f61db93008
1 changed files with 4 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/lib/teact/teact.ts
View File

@ -3,7 +3,7 @@ import {
fastRaf, fastRafPrimary, onTickEnd, onTickEndPrimary, throttleWithPrimaryRaf, throttleWithRaf, fastRaf, fastRafPrimary, onTickEnd, onTickEndPrimary, throttleWithPrimaryRaf, throttleWithRaf,
} from '../../util/schedulers'; } from '../../util/schedulers';
import { flatten, orderBy } from '../../util/iteratees'; import { flatten, orderBy } from '../../util/iteratees';
import arePropsShallowEqual, { getUnequalProps } from '../../util/arePropsShallowEqual'; import { getUnequalProps } from '../../util/arePropsShallowEqual';
import { handleError } from '../../util/handleError'; import { handleError } from '../../util/handleError';
import { removeAllDelegatedListeners } from './dom-events'; import { removeAllDelegatedListeners } from './dom-events';
@ -676,26 +676,10 @@ export function useRef<T>(initial?: T | null) {
}), []); }), []);
} }
export function memo<T extends FC>(Component: T, areEqual = arePropsShallowEqual, debugKey?: string) { export function memo<T extends FC>(Component: T, debugKey?: string) {
return function TeactMemoWrapper(props: Props) { return function TeactMemoWrapper(props: Props) {
// eslint-disable-next-line react-hooks/rules-of-hooks // eslint-disable-next-line react-hooks/exhaustive-deps
const propsRef = useRef(props); return useMemo(() => createElement(Component, props), Object.values(props), debugKey);
const renderedRef = useRef();
if (!renderedRef.current || (propsRef.current && !areEqual(propsRef.current, props))) {
if (DEBUG && debugKey) {
// eslint-disable-next-line no-console
console.log(
`[Teact.memo] ${Component.name} (${debugKey}): Update is caused by:`,
getUnequalProps(propsRef.current!, props).join(', '),
);
}
propsRef.current = props;
renderedRef.current = createElement(Component, props) as VirtualElementComponent;
}
return renderedRef.current;
} as T; } as T;
} }