在react应用开发中,管理异步操作的加载状态和错误信息是常见且重复的任务。本文将深入探讨如何利用自定义hook来抽象和封装这类重复逻辑,例如加载状态、错误提示及其定时清除机制,从而显著提升代码的可复用性、可维护性与组件的整洁度。通过实例演示,我们将学习如何设计和实现一个通用的自定义hook,以简化组件内部的复杂状态管理。
在React组件中,处理数据加载、表单提交或任何异步操作时,我们通常需要维护以下几种状态:
原始代码示例清晰地展示了这种重复模式:
// LOADING ALL VENDORS const [loadingAllVendors, setLoadingAllVendors] = useState(true); const loadAllVendorsErrorTimeout = useRef (null); const [loadAllVendorsError, setLoadAllVendorsError] = useState (null); const handleLoadAllVendorsError = (error: string|null) => { /* ... */ }; const loadAllVendorsErrorTime: number = 6; const timedLoadAllVendorsError = useCallback((error: string, seconds: number) => { /* ... */ }, []); // LOADING ALL MANUFACTURERS const [loadingAllManufacturers, setLoadingAllManufacturers] = useState (true); const loadAllManufacturersErrorTimeout = useRef (null); const [loadAllManufacturersError, setLoadAllManufacturersError] = useState (null); const handleLoadAllManufacturersError = (error: string|null) => { /* ... */ }; const loadAllManufacturersErrorTime: number = 6; const timedLoadAllManufacturersError = useCallback((error: string, seconds: number) => { /* ... */ }, []); // SEARCHING PARTS const [searching, setSearching] = useState (false); const searchErrorTimeout = useRef (null); const [searchError, setSearchError] = useState (null); const handleSearchError = (error: string|null) => { /* ... */ }; const searchErrorTime: number = 6; const timedSearchError = useCallback((error: string, seconds: number) => { /* ... */ }, []);
可以看到,每个异步操作都重复定义了以下六个相似的逻辑块:
这种重复代码不仅增加了维护成本,也使得组件逻辑变得臃肿。
React的自定义Hook是解决这类问题的理想方案。自定义Hook本质上是一个JavaScript函数,其名称以use开头,并且可以在其中调用其他Hook(如useState、useE
ffect、useCallback等)。它允许我们将组件逻辑(特别是状态管理逻辑)从组件中提取出来,实现复用。
通过自定义Hook,我们可以将上述重复的加载状态、错误状态及其定时清除逻辑封装成一个独立的、可复用的单元。
我们将创建一个名为 useAsyncOperationState 的自定义Hook,它将负责管理异步操作的加载状态、错误状态以及错误信息的定时清除。
useAsyncOperationState Hook将提供以下功能:
它还可以接受一个配置对象,用于设置初始加载状态和默认的错误显示时长。
创建一个名为 useAsyncOperationState.ts (或 .js) 的文件:
import { useState, useRef, useCallback, useEffect } from 'react';
// 配置接口,用于定制Hook的行为
interface UseAsyncOperationStateConfig {
initialLoading?: boolean; // 初始加载状态,默认为 false
defaultErrorDisplaySeconds?: number; // 默认错误显示时长,默认为 6 秒
}
// Hook返回值的接口
interface UseAsyncOperationState {
isLoading: boolean;
error: string | null;
setIsLoading: (loading: boolean) => void;
setError: (errorMessage: string | null) => void;
handleError: (errorMessage: string | null) => void;
displayTimedError: (errorMessage: string, seconds?: number) => void;
}
function useAsyncOperationState(config?: UseAsyncOperationStateConfig): UseAsyncOperationState {
const {
initialLoading = false,
defaultErrorDisplaySeconds = 6,
} = config || {};
const [isLoading, setIsLoading] = useState(initialLoading);
const [error, setError] = useState(null);
const errorTimeoutRef = useRef(null);
// 在组件卸载时清除任何未完成的定时器,避免内存泄漏
useEffect(() => {
return () => {
if (errorTimeoutRef.current) {
clearTimeout(errorTimeoutRef.current);
}
};
}, []);
// 通用的错误处理函数,负责日志记录和设置错误状态
const handleError = useCallback((errorMessage: string | null) => {
if (errorMessage) {
console.error(errorMessage); // 在控制台输出错误信息
}
setError(errorMessage); // 更新错误状态
}, []);
// 带有定时清除功能的错误显示函数
const displayTimedError = useCallback((errorMessage: string, seconds?: number) => {
const displayDuration = seconds ?? defaultErrorDisplaySeconds; // 使用传入时长或默认时长
handleError(errorMessage); // 立即设置错误信息
// 清除之前的定时器,确保只有一个定时器在运行
if (errorTimeoutRef.current) {
clearTimeout(errorTimeoutRef.current);
}
// 设置新的定时器,在指定时间后清除错误
errorTimeoutRef.current = setTimeout(() => {
setError(null);
}, displayDuration * 1000);
}, [handleError, defaultErrorDisplaySeconds]); // 依赖项包括 handleError 和 defaultErrorDisplaySeconds
return {
isLoading,
error,
setIsLoading,
setError, // 暴露直接设置 error 的方法
handleError,
displayTimedError,
};
}
export default useAsyncOperationState; 现在,我们可以将组件中重复的逻辑替换为对 useAsyncOperationState Hook的调用。
import React, { useEffect } from 'react';
import useAsyncOperationState from './useAsyncOperationState'; // 假设Hook文件路径
function MyDataComponent() {
// 使用Hook管理加载所有供应商的状态和错误
const {
isLoading: loadingAllVendors,
error: loadAllVendorsError,
setIsLoading: setLoadingAllVendors,
displayTimedError: displayTimedLoadAllVendorsError,
} = useAsyncOperationState({ initialLoading: true, defaultErrorDisplaySeconds: 6 });
// 使用Hook管理加载所有制造商的状态和错误
const {
isLoading: loadingAllManufacturers,
error: loadAllManufacturersError,
setIsLoading: setLoadingAllManufacturers,
displayTimedError: displayTimedLoadAllManufacturersError,
} = useAsyncOperationState({ initialLoading: true, defaultErrorDisplaySeconds: 6 });
// 使用Hook管理部件搜索的状态和错误
const {
isLoading: searching,
error: searchError,
setIsLoading: setSearching,
displayTimedError: displayTimedSearchError,
} = useAsyncOperationState({ initialLoading: false, defaultErrorDisplaySeconds: 6 });
// 示例:加载供应商数据
useEffect(() => {
const fetchVendors = async () => {
setLoadingAllVendors(true); // 设置加载状态
try {
// 模拟API调用
await new Promise(resolve => setTimeout(resolve, 1500));
// 模拟成功
console.log("Vendors loaded successfully.");
// 如果之前有错误,这里可以清除
displayTimedLoadAllVendorsError(null);
} catch (err: any) {
// 模拟错误,并显示5秒
displayTimedLoadAllVendorsError(`Failed to load vendors: ${err.message}`, 5);
} finally {
setLoadingAllVendors(false); // 结束加载状态
}
};
fetchVendors();
}, [setLoadingAllVendors, displayTimedLoadAllVendorsError]); // 依赖项
// 示例:加载制造商数据 (类似逻辑)
useEffect(() => {
const fetchManufacturers = async () => {
setLoadingAllManufacturers(true);
try {
await new Promise(resolve => setTimeout(resolve, 2000));
console.log("Manufacturers loaded successfully.");
displayTimedLoadAllManufacturersError(null);
} catch (err: any) {
displayTimedLoadAllManufacturersError(`Failed to load manufacturers: ${err.message}`);
} finally {
setLoadingAllManufacturers(false);
}
};
fetchManufacturers();
}, [setLoadingAllManufacturers, displayTimedLoadAllManufacturersError]);
// 示例:搜索部件 (类似逻辑)
const handleSearch = async (query: string) => {
setSearching(true);
try {
await new Promise(resolve => setTimeout(resolve, 1000));
if (query === "error") {
throw new Error("Search failed for 'error' query!");
}
console.log(`Searching for "${query}" successful.`);
displayTimedSearchError(null);
} catch (err: any) {
displayTimedSearchError(`Search error: ${err.message}`, 7);
} finally {
setSearching(false);
}
};
return (
数据管理示例
供应商数据
{loadingAllVendors && 正在加载供应商...
}
{loadAllVendorsError && 错误: {loadAllVendorsError}
}
{!loadingAllVendors && !loadAllVendorsError && 供应商数据已加载。
}
制造商数据
{loadingAllManufacturers && 正在加载制造商...
}
{loadAllManufacturersError && 错误: {loadAllManufacturersError}
}
{!loadingAllManufacturers && !loadAllManufacturersError && 制造商数据已加载。
}
部件搜索
handleSearch(e.target.value)} />
{searching && 正在搜索...
}
{searchError && 搜索错误: {searchError}
}
{!searching && !searchError && 等待搜索。
}
);
}
export default MyDataComponent;通过上述改造,原始组件中大量的重复状态和逻辑被抽象到了 useAsyncOperationState Hook中。组件内部现在只关注如何使用这些状态和函数,而不再关心它们的具体实现细节。