From b807f73bc3680a2ef25580fa5225dbb7a3d6b811 Mon Sep 17 00:00:00 2001
From: fantasticit <fantasticit@163.com>
Date: Sun, 24 Apr 2022 20:57:43 +0800
Subject: [PATCH] feat: add loading in message box

---
 packages/client/src/components/message/index.tsx | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/packages/client/src/components/message/index.tsx b/packages/client/src/components/message/index.tsx
index 496c8629..0ee9ae08 100644
--- a/packages/client/src/components/message/index.tsx
+++ b/packages/client/src/components/message/index.tsx
@@ -8,6 +8,7 @@ import { DataRender } from 'components/data-render';
 import { Empty } from 'components/empty';
 import { Placeholder } from './placeholder';
 import styles from './index.module.scss';
+import { useUser } from 'data/user';
 
 const { Text } = Typography;
 const PAGE_SIZE = 6;
@@ -82,7 +83,7 @@ const MessagesRender = ({ messageData, loading, error, onClick = null, page = 1,
   );
 };
 
-export const Message = () => {
+const MessageBox = () => {
   const { data: allMsgs, loading: allLoading, error: allError, page: allPage, setPage: allSetPage } = useAllMessages();
   const {
     data: readMsgs,
@@ -206,3 +207,8 @@ export const Message = () => {
     </Dropdown>
   );
 };
+
+export const Message = () => {
+  const { loading, error } = useUser();
+  return <DataRender loading={loading} error={error} normalContent={() => <MessageBox />} />;
+};