useState and useSafeSetState

README.md

Example

import { useSafeSetState } from './custom';

// ...
const [state, safeSetState] = useSafeSetState({ name: '', email: '' })
// ...

custom.js

import { useReducer, useEffect, useRef } from 'react'

function useSetState(initialState) {
  return useReducer(
    (state, newState) => ({...state, ...newState}),
    initialState,
  )
}

function useSafeSetState(initialState) {
  const [state, setState] = useSetState(initialState)

  const mountedRef = useRef(false)
  useEffect(() => {
    mountedRef.current = true
    return () => (mountedRef.current = false)
  }, [])
  const safeSetState = (...args) => mountedRef.current && setState(...args)

  return [state, safeSetState]
}