fix: add worker message
This commit is contained in:
@@ -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
|
||||
};
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user