🚀 feat: update shouldUpdate

ant-design · Mar 22, 2024 · 145b20c · 145b20c
1 parent ca771bb
commit 145b20c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/src/ChatList/ShouldUpdateItem.tsx b/src/ChatList/ShouldUpdateItem.tsx
@@ -21,7 +21,12 @@ class ShouldUpdateItem extends Component<
       return nextProps.shouldUpdate(this.props, nextProps);
     }
     try {
-      return !isEqual(this.props, nextProps);
+      return (
+        !isEqual(this.props.content, nextProps?.content) ||
+        !isEqual(this.props.loading, nextProps?.loading) ||
+        !isEqual(this.props.chatItemRenderConfig, nextProps?.chatItemRenderConfig) ||
+        !isEqual(this.props.meta, nextProps?.meta)
+      );
     } catch (error) {
       return true;
     }

diff --git a/src/ProChat/components/ChatList/Extras/User.tsx b/src/ProChat/components/ChatList/Extras/User.tsx
@@ -9,14 +9,18 @@ export const UserMessageExtra: RenderMessageExtra = memo(({ extra, ...rest }) =>
   const hasTranslate = !!extra?.translate;
 
   const [messageItemExtraRender] = useStore((s) => [s.messageItemExtraRender]);
+
+  const dom = messageItemExtraRender?.({ extra, ...rest }, 'user');
+
+  if (!dom) return;
   return (
     <Flexbox gap={8} style={{ marginTop: hasTranslate ? 8 : 0 }}>
       {extra?.translate && (
         <div>
           <Divider style={{ margin: '12px 0' }} />
         </div>
       )}
-      {messageItemExtraRender?.({ extra, ...rest }, 'user')}
+      {dom}
     </Flexbox>
   );
 });