fix: add worker message

This commit is contained in:
jialin
2025-12-17 14:59:07 +08:00
parent bd46fcdc75
commit 5e48d75ead
@@ -1,9 +1,7 @@
import useSetChunkRequest from '@/hooks/use-chunk-request'; import useSetChunkRequest from '@/hooks/use-chunk-request';
import useUpdateChunkedList from '@/hooks/use-update-chunk-list'; import useUpdateChunkedList from '@/hooks/use-update-chunk-list';
import { useIntl } from '@umijs/max';
import { message } from 'antd';
import _ from 'lodash'; import _ from 'lodash';
import { useRef } from 'react'; import { useRef, useState } from 'react';
import { WORKERS_API } from '../../resources/apis'; import { WORKERS_API } from '../../resources/apis';
export default function useAddWorkerMessage(params: { export default function useAddWorkerMessage(params: {
@@ -11,21 +9,14 @@ export default function useAddWorkerMessage(params: {
}) { }) {
const chunkRequestRef = useRef<any>(null); const chunkRequestRef = useRef<any>(null);
const newItemsRef = useRef<any[]>([]); const newItemsRef = useRef<any[]>([]);
const intl = useIntl(); const [addedCount, setAddedCount] = useState(0);
const [messageApi, contextHolder] = message.useMessage();
const showAddWorkerMessage = _.debounce(() => { const showAddWorkerMessage = () => {
if (newItemsRef.current.length > 0) { if (newItemsRef.current.length > 0) {
const count = newItemsRef.current.length; const count = newItemsRef.current.length;
messageApi.success( setAddedCount(count);
intl.formatMessage(
{ id: 'clusters.addworker.message.success' },
{ count }
)
);
newItemsRef.current = [];
} }
}, 300); };
const { setChunkRequest } = useSetChunkRequest(); const { setChunkRequest } = useSetChunkRequest();
const { updateChunkedList } = useUpdateChunkedList({ const { updateChunkedList } = useUpdateChunkedList({
@@ -58,7 +49,7 @@ export default function useAddWorkerMessage(params: {
}; };
return { return {
contextHolder, addedCount,
createModelsChunkRequest createModelsChunkRequest
}; };
} }