Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat(FR-252): Introduce session status info modal in session detail panel #3190

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
@@ -0,0 +1,219 @@
import { useSuspendedBackendaiClient } from '../../hooks';
import { useCurrentUserRole } from '../../hooks/backendai';
import BAIModal from '../BAIModal';
import DoubleTag from '../DoubleTag';
import Flex from '../Flex';
import { statusTagColor } from './SessionStatusTag';
import { SessionStatusDetailModalFragment$key } from './__generated__/SessionStatusDetailModalFragment.graphql';
import { CheckCircleOutlined, CloseCircleOutlined } from '@ant-design/icons';
import {
Descriptions,
ModalProps,
Tag,
Timeline,
Typography,
theme,
} from 'antd';
import graphql from 'babel-plugin-relay/macro';
import dayjs from 'dayjs';
import _ from 'lodash';
import { Fragment } from 'react';
import { useTranslation } from 'react-i18next';
import { useFragment } from 'react-relay';

export const statusInfoTagColor = {
'idle-timeout': 'green',
'user-requested': 'green',
scheduled: 'green',
'self-terminated': 'green',
'failed-to-start': 'red',
'creation-failed': 'red',
'no-available-instances': 'red',
};

type Predicates = {
name: string;
msg: string;
};

type ErrorCollection = {
name: string;
repr: string;
src: string;
agent_id?: string;
traceback?: string;
};

type StatusData = {
kernel?: {
exit_code: number | string;
};
session?: {
status: string;
};
scheduler?: {
failed_predicates: Array<Predicates>;
passed_predicates: Array<Predicates>;
retries: number;
last_try: string;
msg?: string;
};
error?: {
name: string;
repr: string;
src: string;
collection: Array<ErrorCollection>;
};
};

interface SessionStatusDetailModalProps extends ModalProps {
sessionFrgmt: SessionStatusDetailModalFragment$key;
}

const SessionStatusDetailModal: React.FC<SessionStatusDetailModalProps> = ({
sessionFrgmt,
...modalProps
}) => {
const { t } = useTranslation();
const { token } = theme.useToken();
const userRole = useCurrentUserRole();
const baiClient = useSuspendedBackendaiClient();

const session = useFragment(
graphql`
fragment SessionStatusDetailModalFragment on ComputeSessionNode {
id
status
status_info
status_data
starts_at
}
`,
sessionFrgmt,
);
console.log(session);
const statusData: StatusData = JSON.parse(session.status_data || '{}');
console.log('#', statusData);

return (
<BAIModal
title={
<>
{t('session.StatusInfo')}
<DoubleTag
values={[
{
label: session.status ?? '',
color: session.status
? _.get(statusTagColor, session.status)
: undefined,
style: { fontWeight: 'normal' },
},
{
label: session.status_info ?? '',
color: session.status_info
? _.get(statusInfoTagColor, session.status_info)
: undefined,
style: { borderStyle: 'dashed', fontWeight: 'normal' },
},
]}
/>
</>
}
footer={null}
width={450}
{...modalProps}
>
<Descriptions
size="small"
// labelStyle={{ width: '40%' }}
title={t('session.StatusDetail')}
layout="vertical"
column={2}
>
{statusData?.kernel ? (
<Descriptions.Item label={t('session.KernelExitCode')} span={1}>
{statusData.kernel.exit_code}
</Descriptions.Item>
) : null}
{statusData?.session ? (
<Descriptions.Item label={t('session.SessionStatus')} span={1}>
{statusData.session?.status}
</Descriptions.Item>
) : null}
{statusData?.scheduler ? (
<>
<Descriptions.Item label={t('session.TotalRetries')} span={1}>
{statusData.scheduler?.retries}
</Descriptions.Item>
<Descriptions.Item label={t('session.LastTry')} span={1}>
{dayjs(statusData.scheduler?.last_try).format('lll')}
</Descriptions.Item>
<Descriptions.Item label={t('session.Message')} span={2}>
{statusData.scheduler?.msg}
</Descriptions.Item>
<Descriptions.Item>
<Descriptions title={t('session.PredicateChecks')} column={1}>
<Descriptions.Item>
<Timeline
items={_.concat(
_.map(statusData.scheduler?.failed_predicates, (p) => {
return {
dot: <CloseCircleOutlined />,
color: token.colorError,
children: (
<Flex direction="column" align="stretch">
<Typography.Text>{p.name}</Typography.Text>
<Typography.Text type="secondary">
{p.msg}
</Typography.Text>
</Flex>
),
};
}),
_.map(statusData.scheduler?.passed_predicates, (p) => {
return {
dot: <CheckCircleOutlined />,
color: token.colorSuccess,
children: <Typography.Text>{p.name}</Typography.Text>,
};
}),
)}
/>
</Descriptions.Item>
</Descriptions>
</Descriptions.Item>
</>
) : null}
{statusData?.error
? _.map(statusData?.error?.collection ?? statusData, (collection) => {
return (
<Fragment key={collection.name}>
<Descriptions.Item label={t('dialog.error.Error')} span={2}>
<Tag color="red">{collection.name}</Tag>
</Descriptions.Item>
{(userRole === 'superadmin' ||
!baiClient._config.hideAgents) &&
collection?.agent_id && (
<Descriptions.Item label={t('session.AgentId')}>
{collection?.agent_id}
</Descriptions.Item>
)}
<Descriptions.Item label={t('session.Message')} span={2}>
{collection.repr}
</Descriptions.Item>
{collection?.traceback && (
<Descriptions.Item label={t('session.Traceback')} span={2}>
<pre>{collection?.traceback}</pre>
</Descriptions.Item>
)}
</Fragment>
);
})
: null}
</Descriptions>
</BAIModal>
);
};

export default SessionStatusDetailModal;
68 changes: 46 additions & 22 deletions react/src/components/ComputeSessionNodeItems/SessionStatusTag.tsx
Original file line number Diff line number Diff line change
@@ -1,20 +1,24 @@
import Flex from '../Flex';
import SessionStatusDetailModal, {
statusInfoTagColor,
} from './SessionStatusDetailModal';
import {
SessionStatusTagFragment$data,
SessionStatusTagFragment$key,
} from './__generated__/SessionStatusTagFragment.graphql';
import { LoadingOutlined } from '@ant-design/icons';
import { Tag, theme } from 'antd';
import { LoadingOutlined, QuestionCircleOutlined } from '@ant-design/icons';
import { Tag, Tooltip, theme } from 'antd';
import graphql from 'babel-plugin-relay/macro';
import _ from 'lodash';
import React from 'react';
import React, { Suspense, useState } from 'react';
import { useTranslation } from 'react-i18next';
import { useFragment } from 'react-relay';

interface SessionStatusTagProps {
sessionFrgmt?: SessionStatusTagFragment$key | null;
showInfo?: boolean;
}
const statusTagColor = {
export const statusTagColor = {
//prepare
RESTARTING: 'blue',
PREPARING: 'blue',
Expand All @@ -41,31 +45,30 @@ const isTransitional = (session: SessionStatusTagFragment$data) => {
].includes(session?.status || '');
};

const statusInfoTagColor = {
// 'idle-timeout': undefined,
// 'user-requested': undefined,
// scheduled: undefined,
// 'self-terminated': undefined,
'no-available-instances': 'red',
'failed-to-start': 'red',
'creation-failed': 'red',
};
const SessionStatusTag: React.FC<SessionStatusTagProps> = ({
sessionFrgmt,
showInfo,
}) => {
const { t } = useTranslation();
const { token } = theme.useToken();
const [openStatusDetailModal, setOpenStatusDetailModal] =
useState<boolean>(false);

const session = useFragment(
graphql`
fragment SessionStatusTagFragment on ComputeSessionNode {
id
name
status
status_info
status_data

...SessionStatusDetailModalFragment
}
`,
sessionFrgmt,
);
const { token } = theme.useToken();

const hasDetail = session?.status_data && session?.status_data !== '{}';

return session ? (
_.isEmpty(session.status_info) || !showInfo ? (
Expand All @@ -85,14 +88,28 @@ const SessionStatusTag: React.FC<SessionStatusTagProps> = ({
</Tag>
) : (
<Flex>
<Tag
style={{ margin: 0, zIndex: 1 }}
color={
session.status ? _.get(statusTagColor, session.status) : undefined
}
<Tooltip
title={hasDetail ? t('button.ClickForMoreDetails') : undefined}
>
{session.status}
</Tag>
<Tag
color={
session.status ? _.get(statusTagColor, session.status) : undefined
}
icon={hasDetail ? <QuestionCircleOutlined /> : undefined}
style={{
margin: 0,
zIndex: 1,
cursor: hasDetail ? 'pointer' : 'auto',
}}
onClick={() => {
if (hasDetail) {
setOpenStatusDetailModal(true);
}
}}
>
{session.status}
</Tag>
</Tooltip>
<Tag
style={{
margin: 0,
Expand All @@ -112,6 +129,13 @@ const SessionStatusTag: React.FC<SessionStatusTagProps> = ({
>
{session.status_info}
</Tag>
<Suspense fallback={null}>
<SessionStatusDetailModal
sessionFrgmt={session}
open={openStatusDetailModal}
onCancel={() => setOpenStatusDetailModal(false)}
/>
</Suspense>
</Flex>
)
) : null;
Expand Down
12 changes: 7 additions & 5 deletions react/src/components/DoubleTag.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ import React from 'react';
export type DoubleTagObjectValue = {
label: string;
color?: string;
style?: React.CSSProperties; // style 속성 추가
};

const DoubleTag: React.FC<{
Expand Down Expand Up @@ -36,11 +37,12 @@ const DoubleTag: React.FC<{
!_.isEmpty(objValue.label) ? (
<Tag
key={idx}
style={
_.last(objectValues) === objValue
? undefined
: { margin: 0, marginRight: -1 }
}
style={{
...(_.last(objectValues) === objValue
? {}
: { margin: 0, marginRight: -1 }),
...objValue.style,
}}
color={objValue.color}
>
{!_.isUndefined(highlightKeyword) ? (
Expand Down
Loading