San3yuan | 4d0e803 | 2025-04-04 17:21:40 +0800 | [diff] [blame] | 1 | import { useState, useEffect, useCallback } from 'react' |
| 2 | import { data } from 'react-router' |
| 3 | |
阳菜,放晴! | 7e1e3a5 | 2025-06-05 23:00:51 +0800 | [diff] [blame] | 4 | // type RequestFunction<T> = () => Promise<T> |
| 5 | type RequestFunction<T, P = any> = (params?: P) => Promise<T>; |
San3yuan | 4d0e803 | 2025-04-04 17:21:40 +0800 | [diff] [blame] | 6 | |
| 7 | interface UseApiResult<T> { |
| 8 | data: T | null |
| 9 | loading: boolean |
| 10 | error: Error | null |
| 11 | refresh: () => void |
| 12 | } |
| 13 | |
阳菜,放晴! | 7e1e3a5 | 2025-06-05 23:00:51 +0800 | [diff] [blame] | 14 | // export function useApi<T>( |
| 15 | // requestFn: RequestFunction<T>, |
| 16 | // immediate = true |
| 17 | // ): UseApiResult<T> { |
| 18 | export function useApi<T, P = any>( |
| 19 | requestFn: RequestFunction<T, P>, // 接收参数 |
| 20 | immediate = true |
| 21 | ) { |
San3yuan | 4d0e803 | 2025-04-04 17:21:40 +0800 | [diff] [blame] | 22 | const [data, setData] = useState<T | null>(null) |
| 23 | const [loading, setLoading] = useState(false) |
| 24 | const [error, setError] = useState<Error | null>(null) |
| 25 | |
阳菜,放晴! | 7e1e3a5 | 2025-06-05 23:00:51 +0800 | [diff] [blame] | 26 | const execute = useCallback(async (params?: P) => { // 添加参数 |
San3yuan | 4d0e803 | 2025-04-04 17:21:40 +0800 | [diff] [blame] | 27 | try { |
| 28 | setLoading(true) |
阳菜,放晴! | 7e1e3a5 | 2025-06-05 23:00:51 +0800 | [diff] [blame] | 29 | const result = await requestFn(params);//传参 |
San3yuan | 4d0e803 | 2025-04-04 17:21:40 +0800 | [diff] [blame] | 30 | setData(result) |
| 31 | setError(null) |
| 32 | return result // 返回请求结果 |
| 33 | } catch (err) { |
| 34 | setError(err as Error) |
| 35 | throw err |
| 36 | } finally { |
| 37 | setLoading(false) |
| 38 | } |
| 39 | }, [requestFn]) |
| 40 | |
| 41 | useEffect(() => { |
| 42 | if (immediate) { |
| 43 | execute() |
| 44 | } |
| 45 | }, [execute, immediate]) |
| 46 | |
| 47 | return { data, loading, error, refresh: execute } |
| 48 | } |