Files
gpustack-ui/src/pages/llmodels/components/view-logs-modal.tsx
T

104 lines
2.7 KiB
TypeScript

import LogsViewer from '@/components/logs-viewer/virtual-log-list';
import useSetChunkRequest from '@/hooks/use-chunk-request';
import { useIntl } from '@umijs/max';
import { Modal } from 'antd';
import React, { useCallback, useEffect, useState } from 'react';
import { MODELS_API } from '../apis';
import { InstanceRealLogStatus } from '../config';
type ViewModalProps = {
open: boolean;
url: string;
id?: number | string;
modelId?: number | string;
tail?: number;
onCancel: () => void;
};
const ViewCodeModal: React.FC<ViewModalProps> = (props) => {
const viewportHeight = window.innerHeight;
const intl = useIntl();
const { setChunkRequest } = useSetChunkRequest();
const { open, url, onCancel, tail } = props || {};
const [modalSize] = useState<any>({
width: '100%',
height: viewportHeight - 86
});
const [enableScorllLoad, setEnableScorllLoad] = useState(true);
const logsViewerRef = React.useRef<any>(null);
const requestRef = React.useRef<any>(null);
const handleCancel = useCallback(() => {
logsViewerRef.current?.abort();
onCancel();
}, [onCancel]);
const updateHandler = (list: any) => {
const data = list?.find((item: any) => item.data?.id === props.id);
// state in InstanceRealLogStatus will not enable scorll load, because it is in the trasisition state
if (data) {
setEnableScorllLoad(
() => !InstanceRealLogStatus.includes(data?.data?.state)
);
}
};
useEffect(() => {
if (!props.id) return;
if (open) {
requestRef.current?.current?.cancel?.();
requestRef.current = setChunkRequest({
url: `${MODELS_API}/${props.modelId}/instances`,
handler: updateHandler
});
} else {
logsViewerRef.current?.abort();
}
return () => {
requestRef.current?.current?.cancel?.();
};
}, [props.id, open]);
return (
<Modal
title={
<span className="flex flex-center">
<span style={{ fontWeight: 'var(--font-weight-bold)' }}>
{' '}
{intl.formatMessage({ id: 'common.button.viewlog' })}
</span>
</span>
}
zIndex={3000}
open={open}
centered={true}
onCancel={handleCancel}
destroyOnClose={true}
closeIcon={true}
maskClosable={false}
keyboard={true}
styles={{
content: {
borderRadius: 0
}
}}
width={modalSize.width}
footer={null}
>
<LogsViewer
ref={logsViewerRef}
height={modalSize.height}
diffHeight={93}
url={url}
tail={tail}
enableScorllLoad={enableScorllLoad}
params={{
follow: true
}}
></LogsViewer>
</Modal>
);
};
export default React.memo(ViewCodeModal);