react-use/src/useSetState.ts
2021-04-06 11:38:36 +03:00

17 lines
474 B
TypeScript

import { useCallback, useState } from 'react';
const useSetState = <T extends object>(
initialState: T = {} as T
): [T, (patch: Partial<T> | ((prevState: T) => Partial<T>)) => void] => {
const [state, set] = useState<T>(initialState);
const setState = useCallback((patch) => {
set((prevState) =>
Object.assign({}, prevState, patch instanceof Function ? patch(prevState) : patch)
);
}, []);
return [state, setState];
};
export default useSetState;