fastlion-amis/src/hooks/use-set-state.ts

一种MIS页面生成工具

import {useCallback, useState} from 'react';
import isFunction from 'lodash/isFunction';

const useSetState = <T extends object>(
  initialState: T = {} as T
): [T, (patch: Partial<T> | ((prevState: T) => Partial<T>)) => void] => {
  const [state, setState] = useState<T>(initialState);

  const setMergeState = useCallback(patch => {
    setState(prevState => ({
      ...prevState,
      ...(isFunction(patch) ? patch(prevState) : patch)
    }));
  }, []);

  return [state, setMergeState];
};

export default useSetState;