From 180935c6cf65ed70f102245d4aadc7c29019ed01 Mon Sep 17 00:00:00 2001
From: Tristan Yang <yanggc888@163.com>
Date: Thu, 23 Feb 2023 22:17:41 +0800
Subject: [PATCH] refactor: avator src

---
 src/common/component/Avatar.tsx | 15 +++++----------
 1 file changed, 5 insertions(+), 10 deletions(-)

diff --git a/src/common/component/Avatar.tsx b/src/common/component/Avatar.tsx
index c25bc7fb..46a33c93 100644
--- a/src/common/component/Avatar.tsx
+++ b/src/common/component/Avatar.tsx
@@ -1,4 +1,4 @@
-import { FC, ImgHTMLAttributes, useEffect, useState } from "react";
+import { FC, ImgHTMLAttributes, useState } from "react";
 import { getInitials } from "../utils";
 
 interface Props extends ImgHTMLAttributes<HTMLImageElement> {
@@ -29,17 +29,12 @@ const Avatar: FC<Props> = ({
   height,
   ...rest
 }) => {
-  const [avatarSrc, setAvatarSrc] = useState("");
-  useEffect(() => {
-    if (src) {
-      setAvatarSrc(src);
-    }
-  }, [src]);
+  const [error, setError] = useState(false);
   const handleError = () => {
-    setAvatarSrc("");
+    setError(true);
   };
-  if (avatarSrc) {
-    return <img width={width} height={height} src={avatarSrc} onError={handleError} {...rest} />;
+  if (!error && src) {
+    return <img width={width} height={height} src={src} onError={handleError} {...rest} />;
   }
   return (
     <div