open.source/manifold
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Make useStateCheckEquality more flexible

Browse Source
This commit is contained in:
Marshall Polaris 2022-08-09 23:53:27 -07:00
parent 1298f2ceb0
commit 0e4b317322
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
web/hooks/use-state-check-equality.ts
View File

@ -1,5 +1,5 @@
import { isEqual } from 'lodash' import { isEqual } from 'lodash'
import { useMemo, useRef, useState } from 'react' import { SetStateAction, useMemo, useRef, useState } from 'react'
export const useStateCheckEquality = <T>(initialState: T) => { export const useStateCheckEquality = <T>(initialState: T) => {
const [state, setState] = useState(initialState) const [state, setState] = useState(initialState)
@ -8,8 +8,9 @@ export const useStateCheckEquality = <T>(initialState: T) => {
stateRef.current = state stateRef.current = state
const checkSetState = useMemo( const checkSetState = useMemo(
() => (newState: T) => { () => (next: SetStateAction<T>) => {
const state = stateRef.current const state = stateRef.current
const newState = next instanceof Function ? next(state) : next
if (!isEqual(state, newState)) { if (!isEqual(state, newState)) {
setState(newState) setState(newState)
} }