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 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<any>(null);
const newItemsRef = useRef<any[]>([]);
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
};
}