diff --git a/src/pages/cluster-management/hooks/use-add-worker-message.ts b/src/pages/cluster-management/hooks/use-add-worker-message.ts index c6e9e6b9..524a37fb 100644 --- a/src/pages/cluster-management/hooks/use-add-worker-message.ts +++ b/src/pages/cluster-management/hooks/use-add-worker-message.ts @@ -1,9 +1,7 @@ import useSetChunkRequest from '@/hooks/use-chunk-request'; import useUpdateChunkedList from '@/hooks/use-update-chunk-list'; -import { useIntl } from '@umijs/max'; -import { message } from 'antd'; import _ from 'lodash'; -import { useRef } from 'react'; +import { useRef, useState } from 'react'; import { WORKERS_API } from '../../resources/apis'; export default function useAddWorkerMessage(params: { @@ -11,21 +9,14 @@ export default function useAddWorkerMessage(params: { }) { const chunkRequestRef = useRef(null); const newItemsRef = useRef([]); - const intl = useIntl(); - const [messageApi, contextHolder] = message.useMessage(); + const [addedCount, setAddedCount] = useState(0); - const showAddWorkerMessage = _.debounce(() => { + const showAddWorkerMessage = () => { if (newItemsRef.current.length > 0) { const count = newItemsRef.current.length; - messageApi.success( - intl.formatMessage( - { id: 'clusters.addworker.message.success' }, - { count } - ) - ); - newItemsRef.current = []; + setAddedCount(count); } - }, 300); + }; const { setChunkRequest } = useSetChunkRequest(); const { updateChunkedList } = useUpdateChunkedList({ @@ -58,7 +49,7 @@ export default function useAddWorkerMessage(params: { }; return { - contextHolder, + addedCount, createModelsChunkRequest }; }