A simple hook to implement interval polling of any thunk (ReactJS)

useCustomPollingExample.js

import React, { useState } from 'react';
import usePolling from './usePolling';
import ...

const useCustomPolling = () => {
  const [state, setState] = useState()

  const cb = async (cancelPolling) => {
    // do some logics here... whatever...
    
    if (!isDone) return; // Continue polling
    
    // do some more logics here... 
    
    cancelPolling(); // Stop polling
  };

  const cancelPolling = usePolling(cb, 5000);

  return { state, cancelPolling };
};

usePolling.js

import { useEffect, useRef } from 'react';

const usePolling = (callback, delay) => {
  const savedCallback = useRef();
  const intervalRef = useRef();
  const cleanup = () => clearInterval(intervalRef.current);

  useEffect(() => {
    savedCallback.current = callback;
  }, [callback]);

  useEffect(() => {
    const tick = () => savedCallback.current(cleanup);

    if (delay != null) {
      intervalRef.current = setInterval(tick, delay);
    }

    tick();

    return cleanup;
  }, [delay]);

  return cleanup;
};

export default usePolling;