注册
web

我花了一天时间,做了一个图片上传组件,看起来很酷实际上确实很酷

今天,我花了一天的时间做了一个图片上传组件。效果如下:


1721809724352-1721805313079-1721805282575-1721805255491-20240724_151251.gif


可能有人觉得,这个组件很简单,没什么技术含量,其实确实也啥技术含量。但是,我是想借这个组件,来表达一种封装的思想在里面,希望可以帮助到一些朋友。


简单的描述下这个组件的功能:



  • 用户可以点击下面颜色比较绚丽的上传按钮,选择本地图片进行上传,也可以直接点击图片区域进行上传。
  • 上传过程中,会有一个上传中的进度条,上传完成后,会有一个上传成功的提示,如果失败了,会有一个上传失败的提示,而且支持重试。
  • 可以点击图片右上角的删除按钮,删除图片。
  • 并发控制,最多只能同时上传 N 张图片,也就是所谓的限频,这里是 2 张。

是不是看了这么多功能之后,就开始有点头皮发麻了?哈哈,不要怕,这就带你了解下,如何拆解这种功能,而且,学会了这种拆解的办法,后面你遇到更加复杂的,也可以得心应手。


拆解功能,逐步实现


首先,我们思考,我们该使用自底向上的思路,还是自顶向下的思路来拆解这个功能呢?我的建议自顶向下的思路去思考架构,然后自底向上的去实现功能。


架构图


因为我们这个图片上传组件是支持多长图片同时上传的,而且,我们还需要支持上传失败重试的功能,所以,为了让功能更加聚焦,我们把关注点放在 PhotoItem 上,没一个 PhotoItem 就是一个图片上传的单元。他可以独立的上传,独立的删除,独立的重试。


那么,为了让 PhotoItem 这个组件更加简洁,我们把上传逻辑放在hooks useUpload中,这样,PhotoItem 只需要关注自己的展示逻辑即可。


这样做的目的是做到关注点分离,通常来讲,也是符合单一职责原则的。写出来的组件维护性必定大大提升。


代码实现


我们先来看下 useUpload 的代码,因为PhotoItem 依赖他,我们先实现它。


"use client";
export const useUploader = (uploadAction) => {
const [isUploading, setIsUploading] = useState(false);
const [error, setError] = useState(null);
const upload = useCallback(async (file) => {
setIsUploading(true);
setError(null);
try {
return await uploadAction(file);
} catch (err) {
setError(err.message || 'Upload failed');
} finally {
setIsUploading(false);
}
}, [uploadAction]);

const reset = useCallback(() => {
setIsUploading(false);
setError(null);
}, []);

return { upload, isUploading, error, reset };
};

可以看到,我们的 hooks 非常之简单,就是暴露了一个实现图片上传的狗子 upload,然而,他替我们的组件管理了上传中,上传失败,的状态,因此,接下来看,我们的PhotoItem 组件将会有多清晰。


export const PhotoItem = ({
file,
onRemove,
onUploadComplete,
onUploadError,
}
) => {
const { upload, isUploading, error, reset } = useUploader();

const startUpload = useCallback(async () => {
try {
const url = await upload(file);
onUploadComplete(url);
} catch (err) {
onUploadError();
}
}, [file, upload, onUploadComplete, onUploadError]);

useEffect(() => {
startUpload();
}, [queueUpload, startUpload]);

const handleRetry = () => {
reset();
startUpload();
};

return (
<div className="relative w-full h-20">
<img
src={URL.createObjectURL(file)}
/>

{!isUploading && !error(
Uploaded
)}
{isUploading && (
<Progress />
)}
{error && (
<span>Failed</span>
)}
</div>

);
};

OK,到目前为止,还是极其简单的,但是我们貌似忘记了一个很核心的功能,限制并发数。为什么要限制并发数,因为我们自己的服务器或者三方的服务器,可能会有并发数的限制,如果我们不限制并发数,可能会导致一次传多张图片是卡住。


思考,如何限制并发数


我们想一样,是谁触发了上传的呢?是不是 PhotoItem 组件呢?是的,我们可以在 PhotoItem 组件中,去控制并发数,但是,这样做,会导致 PhotoItem 组件的逻辑变得复杂,因为他不仅要关注自己的展示逻辑,还要关注并发数的控制逻辑。这就显的不太合适了。所以,我们应该把他丢出去对吧,截止到目前为止,我们的PhotoUploader 这个组件似乎并没有干任何事情,我们思考下,并发控制的逻辑是否应该是他来呢?


答案是显然的,我们应该把并发控制的逻辑放在 PhotoUploader 组件中,因为他是整个上传组件的入口,他应该关注并发控制,而不是 PhotoItem 组件,而且最本质的原因是,PhotoItem 也不关心是否有其他的 PhotoItem 。


那么,问题来了,并发控制怎么写呢?使用什么数据结构较为合适呢?不卖关子了,我们知道,队列是最合适的数据结构,因为他是先进先出的,我们可以把上传任务放在队列中,然后,每次上传完成,就从队列中取出一个任务,继续上传。


好,我们改造一下,我们的 PhotoItem 组件,让他不要直接执行上传逻辑,而是把他做成一个任务,然后,把任务放在队列中,然后,我们在 PhotoUploader 组件中,去控制并发数。



export const PhotoItem = ({
file,
onRemove,
...
queueUpload // 加一个队列操作器
}
) => {
const { upload, isUploading, error, reset } = useUploader();
...

useEffect(() => {
queueUpload(startUpload); // 修改这里
}, [queueUpload, startUpload]);


const handleRetry = () => {
reset();
queueUpload(startUpload);//修改这里
};

// .... 其他几乎不变


在来看看我们的 PhotoUploader 组件,他是如何控制并发数的。很简单,我们只需要维护一个队列,然后,每次上传完成,就从队列中取出一个任务,继续上传。


  const processQueue = useCallback(() => {
while (activeUploadsRef.current < MAX_CONCURRENT_UPLOADS && uploadQueueRef.current.length > 0) {
const nextUpload = uploadQueueRef.current.shift();
activeUploadsRef.current++;
nextUpload();
}
}, []);

const queueUpload = useCallback((startUpload) => {
if (activeUploadsRef.current < MAX_CONCURRENT_UPLOADS) {
activeUploadsRef.current++;
startUpload();
} else {
uploadQueueRef.current.push(startUpload);
}
}, []);


这里,只给出最最核心的逻辑,实际上就是维护的了一个任务队列,然后,每次上传完成,就判断下队列中是否还有任务,并且是否超过了并发数,如果没有超过,并且队列中还有任务,就继续上传。仅此而已。


总结一下


这个图片上传组件,看似简单,但是,他涉及到了很多的知识点,比如并发控制,上传失败重试,组件拆解,自顶向下的架构设计,自底向上的功能实现。我们在实现这个组件的过程中。有过很多的思考,比如:



  • 如何拆解功能,让组件更加聚焦,做到关注点分离。
  • 控制并发数,使用队列是最合适的数据结构。
  • 如何设计一个 hooks,让组件更加简洁。
  • 以及自顶向下的架构设计,自底向上的功能实现。

只有建立起这些系统性的思维,我们才能在遇到更加复杂的问题时,得心应手。希望这篇文章对你有所帮助。


欢迎关注我老码沉思录,获取我最新的知识分享。


作者:brzhang
来源:juejin.cn/post/7394854112510951443

0 个评论

要回复文章请先登录注册