@rc-component/tree/es/useUnmount.js

tree ui component for react

import * as React from 'react';
import { useLayoutEffect } from '@rc-component/util';

/**
 * Trigger only when component unmount
 */
function useUnmount(triggerStart, triggerEnd) {
  const [firstMount, setFirstMount] = React.useState(false);
  useLayoutEffect(() => {
    if (firstMount) {
      triggerStart();
      return () => {
        triggerEnd();
      };
    }
  }, [firstMount]);
  useLayoutEffect(() => {
    setFirstMount(true);
    return () => {
      setFirstMount(false);
    };
  }, []);
}
export default useUnmount;