feature: use fetch hook success handler

daodaoedu · Mar 17, 2024 · cabe95e · cabe95e
1 parent bb2eb89
commit cabe95e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/hooks/useFetch.jsx b/hooks/useFetch.jsx
@@ -1,6 +1,6 @@
 import { useEffect, useReducer, useState } from 'react';
 
-const useFetch = (url, { initialValue } = {}) => {
+const useFetch = (url, { initialValue, onSuccess } = {}) => {
   const [render, refetch] = useReducer((pre) => !pre, true);
   const [data, setData] = useState(initialValue);
   const [isFetching, setIsFetching] = useState(true);
@@ -25,6 +25,10 @@ const useFetch = (url, { initialValue } = {}) => {
     };
   }, [url, render]);
 
+  useEffect(() => {
+    if (onSuccess) onSuccess(data);
+  }, [onSuccess, data]);
+
   return { data, isFetching, isError, refetch };
 };