mirror of
https://github.com/enricoros/big-AGI.git
synced 2026-05-10 21:50:14 -07:00
useAsyncCall: simplifies some state tracking
This commit is contained in:
@@ -0,0 +1,49 @@
|
||||
import * as React from 'react';
|
||||
|
||||
interface AsyncState<T> {
|
||||
isLoading: boolean;
|
||||
error: Error | null;
|
||||
data: T | null;
|
||||
}
|
||||
|
||||
export function useAsyncCall<T extends (...args: any[]) => Promise<any>>(asyncFunction: T) {
|
||||
const [state, setState] = React.useState<AsyncState<Awaited<ReturnType<T>>>>({
|
||||
isLoading: false,
|
||||
error: null,
|
||||
data: null,
|
||||
});
|
||||
|
||||
const isMounted = React.useRef(true);
|
||||
|
||||
React.useEffect(() => {
|
||||
return () => {
|
||||
isMounted.current = false;
|
||||
};
|
||||
}, []);
|
||||
|
||||
const execute = React.useCallback(
|
||||
async (...args: Parameters<T>) => {
|
||||
setState(prev => ({ ...prev, isLoading: true, error: null }));
|
||||
|
||||
try {
|
||||
const result = await asyncFunction(...args);
|
||||
if (isMounted.current) {
|
||||
setState({ isLoading: false, error: null, data: result });
|
||||
}
|
||||
return result;
|
||||
} catch (err) {
|
||||
if (isMounted.current) {
|
||||
setState(prev => ({
|
||||
...prev,
|
||||
isLoading: false,
|
||||
error: err instanceof Error ? err : new Error('An error occurred'),
|
||||
}));
|
||||
}
|
||||
throw err;
|
||||
}
|
||||
},
|
||||
[asyncFunction],
|
||||
);
|
||||
|
||||
return [state.isLoading, execute, state.error, state.data] as const;
|
||||
}
|
||||
Reference in New Issue
Block a user