Download web/src/components/TaskLogsTable.js from Jack698/velo: direct link, hf CLI and curl.
- Browser
- Download file 14.3 kB
-
https://huggingface.co/spaces/Jack698/velo/resolve/main/web/src/components/TaskLogsTable.js
- Command line
-
hf download hf://spaces/Jack698/velo/web/src/components/TaskLogsTable.js
-
curl -L -o TaskLogsTable.js https://huggingface.co/spaces/Jack698/velo/resolve/main/web/src/components/TaskLogsTable.js
14.3 kB
| /* | |
| Copyright (c) 2025 Tethys Plex | |
| This file is part of Veloera. | |
| This program is free software: you can redistribute it and/or modify | |
| it under the terms of the GNU General Public License as published by | |
| the Free Software Foundation, either version 3 of the License, or | |
| (at your option) any later version. | |
| This program is distributed in the hope that it will be useful, | |
| but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
| GNU General Public License for more details. | |
| You should have received a copy of the GNU General Public License | |
| along with this program. If not, see <https://www.gnu.org/licenses/>. | |
| */ | |
| 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; | |