web/src/components/TaskLogsTable.js

import React, { useEffect, useState } from 'react';
import { Label } from 'semantic-ui-react';
import { API, copy, isAdmin, showError, showSuccess, timestamp2string } from '../helpers';

import {
    Table,
    Tag,
    Form,
    Button,
    Layout,
    Modal,
    Typography, Progress, Card
} from '@douyinfe/semi-ui';
import { ITEMS_PER_PAGE } from '../constants';

const colors = ['amber', 'blue', 'cyan', 'green', 'grey', 'indigo',
    'light-blue', 'lime', 'orange', 'pink',
    'purple', 'red', 'teal', 'violet', 'yellow'
]


const renderTimestamp = (timestampInSeconds) => {
    const date = new Date(timestampInSeconds * 1000); // 从秒转换为毫秒

    const year = date.getFullYear(); // 获取年份
    const month = ('0' + (date.getMonth() + 1)).slice(-2); // 获取月份，从0开始需要+1，并保证两位数
    const day = ('0' + date.getDate()).slice(-2); // 获取日期，并保证两位数
    const hours = ('0' + date.getHours()).slice(-2); // 获取小时，并保证两位数
    const minutes = ('0' + date.getMinutes()).slice(-2); // 获取分钟，并保证两位数
    const seconds = ('0' + date.getSeconds()).slice(-2); // 获取秒钟，并保证两位数

    return `${year}-${month}-${day} ${hours}:${minutes}:${seconds}`; // 格式化输出
};

function renderDuration(submit_time, finishTime) {
    // 确保startTime和finishTime都是有效的时间戳
    if (!submit_time || !finishTime) return 'N/A';

    // 将时间戳转换为Date对象
    const start = new Date(submit_time);
    const finish = new Date(finishTime);

    // 计算时间差（毫秒）
    const durationMs = finish - start;

    // 将时间差转换为秒，并保留一位小数
    const durationSec = (durationMs / 1000).toFixed(1);

    // 设置颜色：大于60秒则为红色，小于等于60秒则为绿色
    const color = durationSec > 60 ? 'red' : 'green';

    // 返回带有样式的颜色标签
    return (
        <Tag color={color} size="large">
            {durationSec} 秒
        </Tag>
    );
}

const LogsTable = () => {
    const [isModalOpen, setIsModalOpen] = useState(false);
    const [modalContent, setModalContent] = useState('');
    const isAdminUser = isAdmin();
    const columns = [
        {
            title: "提交时间",
            dataIndex: 'submit_time',
            render: (text, record, index) => {
                return (
                    <div>
                        {text ? renderTimestamp(text) : "-"}
                    </div>
                );
            },
        },
        {
            title: "结束时间",
            dataIndex: 'finish_time',
            render: (text, record, index) => {
                return (
                    <div>
                        {text ? renderTimestamp(text) : "-"}
                    </div>
                );
            },
        },
        {
            title: '进度',
            dataIndex: 'progress',
            width: 50,
            render: (text, record, index) => {
                return (
                    <div>
                        {
                            // 转换例如100%为数字100，如果text未定义，返回0
                            isNaN(text.replace('%', '')) ? text : <Progress width={42} type="circle" showInfo={true} percent={Number(text.replace('%', '') || 0)} aria-label="drawing progress" />
                        }
                    </div>
                );
            },
        },
        {
            title: '花费时间',
            dataIndex: 'finish_time', // 以finish_time作为dataIndex
            key: 'finish_time',
            render: (finish, record) => {
                // 假设record.start_time是存在的，并且finish是完成时间的时间戳
                return <>
                    {
                        finish ? renderDuration(record.submit_time, finish) : "-"
                    }
                </>
            },
        },
        {
            title: "渠道",
            dataIndex: 'channel_id',
            className: isAdminUser ? 'tableShow' : 'tableHiddle',
            render: (text, record, index) => {
                return (
                    <div>
                        <Tag
                            color={colors[parseInt(text) % colors.length]}
                            size='large'
                            onClick={() => {
                                copyText(text); // 假设copyText是用于文本复制的函数
                            }}
                        >
                            {' '}
                            {text}{' '}
                        </Tag>
                    </div>
                );
            },
        },
        {
            title: "平台",
            dataIndex: 'platform',
            render: (text, record, index) => {
                return (
                    <div>
                        {renderPlatform(text)}
                    </div>
                );
            },
        },
        {
            title: '类型',
            dataIndex: 'action',
            render: (text, record, index) => {
                return (
                    <div>
                        {renderType(text)}
                    </div>
                );
            },
        },
        {
            title: '任务ID（点击查看详情）',
            dataIndex: 'task_id',
            render: (text, record, index) => {
                return (<Typography.Text
                    ellipsis={{ showTooltip: true }}
                    //style={{width: 100}}
                    onClick={() => {
                        setModalContent(JSON.stringify(record, null, 2));
                        setIsModalOpen(true);
                    }}
                >
                    <div>
                        {text}
                    </div>
                </Typography.Text>);
            },
        },
        {
            title: '任务状态',
            dataIndex: 'status',
            render: (text, record, index) => {
                return (
                    <div>
                        {renderStatus(text)}
                    </div>
                );
            },
        },

        {
            title: '失败原因',
            dataIndex: 'fail_reason',
            render: (text, record, index) => {
                // 如果text未定义，返回替代文本，例如空字符串''或其他
                if (!text) {
                    return '无';
                }

                return (
                    <Typography.Text
                        ellipsis={{ showTooltip: true }}
                        style={{ width: 100 }}
                        onClick={() => {
                            setModalContent(text);
                            setIsModalOpen(true);
                        }}
                    >
                        {text}
                    </Typography.Text>
                );
            }
        }
    ];

    const [logs, setLogs] = useState([]);
    const [loading, setLoading] = useState(true);
    const [activePage, setActivePage] = useState(1);
    const [logCount, setLogCount] = useState(ITEMS_PER_PAGE);
    const [logType] = useState(0);

    let now = new Date();
    // 初始化start_timestamp为前一天
    let zeroNow = new Date(now.getFullYear(), now.getMonth(), now.getDate());
    const [inputs, setInputs] = useState({
        channel_id: '',
        task_id: '',
        start_timestamp: timestamp2string(zeroNow.getTime() /1000),
        end_timestamp: '',
    });
    const { channel_id, task_id, start_timestamp, end_timestamp } = inputs;

    const handleInputChange = (value, name) => {
        setInputs((inputs) => ({ ...inputs, [name]: value }));
    };


    const setLogsFormat = (logs) => {
        for (let i = 0; i < logs.length; i++) {
            logs[i].timestamp2string = timestamp2string(logs[i].created_at);
            logs[i].key = '' + logs[i].id;
        }
        // data.key = '' + data.id
        setLogs(logs);
        setLogCount(logs.length + ITEMS_PER_PAGE);
        // console.log(logCount);
    }

    const loadLogs = async (startIdx) => {
        setLoading(true);

        let url = '';
        let localStartTimestamp = parseInt(Date.parse(start_timestamp) / 1000);
        let localEndTimestamp = parseInt(Date.parse(end_timestamp) / 1000 );
        if (isAdminUser) {
            url = `/api/task/?p=${startIdx}&channel_id=${channel_id}&task_id=${task_id}&start_timestamp=${localStartTimestamp}&end_timestamp=${localEndTimestamp}`;
        } else {
            url = `/api/task/self?p=${startIdx}&task_id=${task_id}&start_timestamp=${localStartTimestamp}&end_timestamp=${localEndTimestamp}`;
        }
        const res = await API.get(url);
        let { success, message, data } = res.data;
        if (success) {
            if (startIdx === 0) {
                setLogsFormat(data);
            } else {
                let newLogs = [...logs];
                newLogs.splice(startIdx * ITEMS_PER_PAGE, data.length, ...data);
                setLogsFormat(newLogs);
            }
        } else {
            showError(message);
        }
        setLoading(false);
    };

    const pageData = logs.slice((activePage - 1) * ITEMS_PER_PAGE, activePage * ITEMS_PER_PAGE);

    const handlePageChange = page => {
        setActivePage(page);
        if (page === Math.ceil(logs.length / ITEMS_PER_PAGE) + 1) {
            // In this case we have to load more data and then append them.
            loadLogs(page - 1).then(r => {
            });
        }
    };

    const refresh = async () => {
        // setLoading(true);
        setActivePage(1);
        await loadLogs(0);
    };

    const copyText = async (text) => {
        if (await copy(text)) {
            showSuccess('已复制：' + text);
        } else {
            // setSearchKeyword(text);
            Modal.error({ title: "无法复制到剪贴板，请手动复制", content: text });
        }
    }

    useEffect(() => {
        refresh().then();
    }, [logType]);

    const renderType = (type) => {
        switch (type) {
            case 'MUSIC':
                return <Label basic color='grey'> 生成音乐 </Label>;
            case 'LYRICS':
                return <Label basic color='pink'> 生成歌词 </Label>;

            default:
                return <Label basic color='black'> 未知 </Label>;
        }
    }

    const renderPlatform = (type) => {
        switch (type) {
            case "suno":
                return <Label basic color='green'> Suno </Label>;
            default:
                return <Label basic color='black'> 未知 </Label>;
        }
    }

    const renderStatus = (type) => {
        switch (type) {
            case 'SUCCESS':
                return <Label basic color='green'> 成功 </Label>;
            case 'NOT_START':
                return <Label basic color='black'> 未启动 </Label>;
            case 'SUBMITTED':
                return <Label basic color='yellow'> 队列中 </Label>;
            case 'IN_PROGRESS':
                return <Label basic color='blue'> 执行中 </Label>;
            case 'FAILURE':
                return <Label basic color='red'> 失败 </Label>;
            case 'QUEUED':
                return <Label basic color='red'> 排队中 </Label>;
            case 'UNKNOWN':
                return <Label basic color='red'> 未知 </Label>;
            case '':
                return <Label basic color='black'> 正在提交 </Label>;
            default:
                return <Label basic color='black'> 未知 </Label>;
        }
    }

    return (
        <>

            <Layout>
                <Form layout='horizontal' labelPosition='inset'>
                    <>
                        {isAdminUser && <Form.Input field="channel_id" label='渠道 ID' style={{ width: '236px', marginBottom: '10px' }} value={channel_id}
                                                    placeholder={'可选值'} name='channel_id'
                                                    onChange={value => handleInputChange(value, 'channel_id')} />
                        }
                        <Form.Input field="task_id" label={"任务 ID"} style={{ width: '236px', marginBottom: '10px' }} value={task_id}
                            placeholder={"可选值"}
                            name='task_id'
                            onChange={value => handleInputChange(value, 'task_id')} />

                        <Form.DatePicker field="start_timestamp" label={"起始时间"} style={{ width: '236px', marginBottom: '10px' }}
                            initValue={start_timestamp}
                            value={start_timestamp} type='dateTime'
                            name='start_timestamp'
                            onChange={value => handleInputChange(value, 'start_timestamp')} />
                        <Form.DatePicker field="end_timestamp" fluid label={"结束时间"} style={{ width: '236px', marginBottom: '10px' }}
                            initValue={end_timestamp}
                            value={end_timestamp} type='dateTime'
                            name='end_timestamp'
                            onChange={value => handleInputChange(value, 'end_timestamp')} />
                        <Button label={"查询"} type="primary" htmlType="submit" className="btn-margin-right"
                            onClick={refresh}>查询</Button>
                    </>
                </Form>
                <Card>
                    <Table columns={columns} dataSource={pageData} pagination={{
                        currentPage: activePage,
                        pageSize: ITEMS_PER_PAGE,
                        total: logCount,
                        pageSizeOpts: [10, 20, 50, 100],
                        onPageChange: handlePageChange,
                    }} loading={loading} />
                </Card>
                <Modal
                    visible={isModalOpen}
                    onOk={() => setIsModalOpen(false)}
                    onCancel={() => setIsModalOpen(false)}
                    closable={null}
                    bodyStyle={{ height: '400px', overflow: 'auto' }} // 设置模态框内容区域样式
                    width={800} // 设置模态框宽度
                >
                    <p style={{ whiteSpace: 'pre-line' }}>{modalContent}</p>
                </Modal>
            </Layout>
        </>
    );
};

export default LogsTable;