MjLogsTable.js 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953
  1. import React, { useEffect, useState } from 'react';
  2. import { useTranslation } from 'react-i18next';
  3. import {
  4. API,
  5. copy,
  6. isAdmin,
  7. showError,
  8. showSuccess,
  9. timestamp2string,
  10. } from '../../helpers';
  11. import {
  12. Button,
  13. Card,
  14. Checkbox,
  15. Divider,
  16. Form,
  17. ImagePreview,
  18. Layout,
  19. Modal,
  20. Progress,
  21. Skeleton,
  22. Table,
  23. Tag,
  24. Typography,
  25. } from '@douyinfe/semi-ui';
  26. import { ITEMS_PER_PAGE } from '../../constants';
  27. import {
  28. IconEyeOpened,
  29. IconSearch,
  30. IconSetting,
  31. } from '@douyinfe/semi-icons';
  32. const { Text } = Typography;
  33. const colors = [
  34. 'amber',
  35. 'blue',
  36. 'cyan',
  37. 'green',
  38. 'grey',
  39. 'indigo',
  40. 'light-blue',
  41. 'lime',
  42. 'orange',
  43. 'pink',
  44. 'purple',
  45. 'red',
  46. 'teal',
  47. 'violet',
  48. 'yellow',
  49. ];
  50. // 定义列键值常量
  51. const COLUMN_KEYS = {
  52. SUBMIT_TIME: 'submit_time',
  53. DURATION: 'duration',
  54. CHANNEL: 'channel',
  55. TYPE: 'type',
  56. TASK_ID: 'task_id',
  57. SUBMIT_RESULT: 'submit_result',
  58. TASK_STATUS: 'task_status',
  59. PROGRESS: 'progress',
  60. IMAGE: 'image',
  61. PROMPT: 'prompt',
  62. PROMPT_EN: 'prompt_en',
  63. FAIL_REASON: 'fail_reason',
  64. };
  65. const LogsTable = () => {
  66. const { t } = useTranslation();
  67. const [isModalOpen, setIsModalOpen] = useState(false);
  68. const [modalContent, setModalContent] = useState('');
  69. // 列可见性状态
  70. const [visibleColumns, setVisibleColumns] = useState({});
  71. const [showColumnSelector, setShowColumnSelector] = useState(false);
  72. const isAdminUser = isAdmin();
  73. // 加载保存的列偏好设置
  74. useEffect(() => {
  75. const savedColumns = localStorage.getItem('mj-logs-table-columns');
  76. if (savedColumns) {
  77. try {
  78. const parsed = JSON.parse(savedColumns);
  79. const defaults = getDefaultColumnVisibility();
  80. const merged = { ...defaults, ...parsed };
  81. setVisibleColumns(merged);
  82. } catch (e) {
  83. console.error('Failed to parse saved column preferences', e);
  84. initDefaultColumns();
  85. }
  86. } else {
  87. initDefaultColumns();
  88. }
  89. }, []);
  90. // 获取默认列可见性
  91. const getDefaultColumnVisibility = () => {
  92. return {
  93. [COLUMN_KEYS.SUBMIT_TIME]: true,
  94. [COLUMN_KEYS.DURATION]: true,
  95. [COLUMN_KEYS.CHANNEL]: isAdminUser,
  96. [COLUMN_KEYS.TYPE]: true,
  97. [COLUMN_KEYS.TASK_ID]: true,
  98. [COLUMN_KEYS.SUBMIT_RESULT]: isAdminUser,
  99. [COLUMN_KEYS.TASK_STATUS]: true,
  100. [COLUMN_KEYS.PROGRESS]: true,
  101. [COLUMN_KEYS.IMAGE]: true,
  102. [COLUMN_KEYS.PROMPT]: true,
  103. [COLUMN_KEYS.PROMPT_EN]: true,
  104. [COLUMN_KEYS.FAIL_REASON]: true,
  105. };
  106. };
  107. // 初始化默认列可见性
  108. const initDefaultColumns = () => {
  109. const defaults = getDefaultColumnVisibility();
  110. setVisibleColumns(defaults);
  111. localStorage.setItem('mj-logs-table-columns', JSON.stringify(defaults));
  112. };
  113. // 处理列可见性变化
  114. const handleColumnVisibilityChange = (columnKey, checked) => {
  115. const updatedColumns = { ...visibleColumns, [columnKey]: checked };
  116. setVisibleColumns(updatedColumns);
  117. };
  118. // 处理全选
  119. const handleSelectAll = (checked) => {
  120. const allKeys = Object.keys(COLUMN_KEYS).map((key) => COLUMN_KEYS[key]);
  121. const updatedColumns = {};
  122. allKeys.forEach((key) => {
  123. if ((key === COLUMN_KEYS.CHANNEL || key === COLUMN_KEYS.SUBMIT_RESULT) && !isAdminUser) {
  124. updatedColumns[key] = false;
  125. } else {
  126. updatedColumns[key] = checked;
  127. }
  128. });
  129. setVisibleColumns(updatedColumns);
  130. };
  131. // 更新表格时保存列可见性
  132. useEffect(() => {
  133. if (Object.keys(visibleColumns).length > 0) {
  134. localStorage.setItem('mj-logs-table-columns', JSON.stringify(visibleColumns));
  135. }
  136. }, [visibleColumns]);
  137. function renderType(type) {
  138. switch (type) {
  139. case 'IMAGINE':
  140. return (
  141. <Tag color='blue' size='large' shape='circle'>
  142. {t('绘图')}
  143. </Tag>
  144. );
  145. case 'UPSCALE':
  146. return (
  147. <Tag color='orange' size='large' shape='circle'>
  148. {t('放大')}
  149. </Tag>
  150. );
  151. case 'VARIATION':
  152. return (
  153. <Tag color='purple' size='large' shape='circle'>
  154. {t('变换')}
  155. </Tag>
  156. );
  157. case 'HIGH_VARIATION':
  158. return (
  159. <Tag color='purple' size='large' shape='circle'>
  160. {t('强变换')}
  161. </Tag>
  162. );
  163. case 'LOW_VARIATION':
  164. return (
  165. <Tag color='purple' size='large' shape='circle'>
  166. {t('弱变换')}
  167. </Tag>
  168. );
  169. case 'PAN':
  170. return (
  171. <Tag color='cyan' size='large' shape='circle'>
  172. {t('平移')}
  173. </Tag>
  174. );
  175. case 'DESCRIBE':
  176. return (
  177. <Tag color='yellow' size='large' shape='circle'>
  178. {t('图生文')}
  179. </Tag>
  180. );
  181. case 'BLEND':
  182. return (
  183. <Tag color='lime' size='large' shape='circle'>
  184. {t('图混合')}
  185. </Tag>
  186. );
  187. case 'UPLOAD':
  188. return (
  189. <Tag color='blue' size='large' shape='circle'>
  190. 上传文件
  191. </Tag>
  192. );
  193. case 'SHORTEN':
  194. return (
  195. <Tag color='pink' size='large' shape='circle'>
  196. {t('缩词')}
  197. </Tag>
  198. );
  199. case 'REROLL':
  200. return (
  201. <Tag color='indigo' size='large' shape='circle'>
  202. {t('重绘')}
  203. </Tag>
  204. );
  205. case 'INPAINT':
  206. return (
  207. <Tag color='violet' size='large' shape='circle'>
  208. {t('局部重绘-提交')}
  209. </Tag>
  210. );
  211. case 'ZOOM':
  212. return (
  213. <Tag color='teal' size='large' shape='circle'>
  214. {t('变焦')}
  215. </Tag>
  216. );
  217. case 'CUSTOM_ZOOM':
  218. return (
  219. <Tag color='teal' size='large' shape='circle'>
  220. {t('自定义变焦-提交')}
  221. </Tag>
  222. );
  223. case 'MODAL':
  224. return (
  225. <Tag color='green' size='large' shape='circle'>
  226. {t('窗口处理')}
  227. </Tag>
  228. );
  229. case 'SWAP_FACE':
  230. return (
  231. <Tag color='light-green' size='large' shape='circle'>
  232. {t('换脸')}
  233. </Tag>
  234. );
  235. default:
  236. return (
  237. <Tag color='white' size='large' shape='circle'>
  238. {t('未知')}
  239. </Tag>
  240. );
  241. }
  242. }
  243. function renderCode(code) {
  244. switch (code) {
  245. case 1:
  246. return (
  247. <Tag color='green' size='large' shape='circle'>
  248. {t('已提交')}
  249. </Tag>
  250. );
  251. case 21:
  252. return (
  253. <Tag color='lime' size='large' shape='circle'>
  254. {t('等待中')}
  255. </Tag>
  256. );
  257. case 22:
  258. return (
  259. <Tag color='orange' size='large' shape='circle'>
  260. {t('重复提交')}
  261. </Tag>
  262. );
  263. case 0:
  264. return (
  265. <Tag color='yellow' size='large' shape='circle'>
  266. {t('未提交')}
  267. </Tag>
  268. );
  269. default:
  270. return (
  271. <Tag color='white' size='large' shape='circle'>
  272. {t('未知')}
  273. </Tag>
  274. );
  275. }
  276. }
  277. function renderStatus(type) {
  278. switch (type) {
  279. case 'SUCCESS':
  280. return (
  281. <Tag color='green' size='large' shape='circle'>
  282. {t('成功')}
  283. </Tag>
  284. );
  285. case 'NOT_START':
  286. return (
  287. <Tag color='grey' size='large' shape='circle'>
  288. {t('未启动')}
  289. </Tag>
  290. );
  291. case 'SUBMITTED':
  292. return (
  293. <Tag color='yellow' size='large' shape='circle'>
  294. {t('队列中')}
  295. </Tag>
  296. );
  297. case 'IN_PROGRESS':
  298. return (
  299. <Tag color='blue' size='large' shape='circle'>
  300. {t('执行中')}
  301. </Tag>
  302. );
  303. case 'FAILURE':
  304. return (
  305. <Tag color='red' size='large' shape='circle'>
  306. {t('失败')}
  307. </Tag>
  308. );
  309. case 'MODAL':
  310. return (
  311. <Tag color='yellow' size='large' shape='circle'>
  312. {t('窗口等待')}
  313. </Tag>
  314. );
  315. default:
  316. return (
  317. <Tag color='white' size='large' shape='circle'>
  318. {t('未知')}
  319. </Tag>
  320. );
  321. }
  322. }
  323. const renderTimestamp = (timestampInSeconds) => {
  324. const date = new Date(timestampInSeconds * 1000); // 从秒转换为毫秒
  325. const year = date.getFullYear(); // 获取年份
  326. const month = ('0' + (date.getMonth() + 1)).slice(-2); // 获取月份,从0开始需要+1,并保证两位数
  327. const day = ('0' + date.getDate()).slice(-2); // 获取日期,并保证两位数
  328. const hours = ('0' + date.getHours()).slice(-2); // 获取小时,并保证两位数
  329. const minutes = ('0' + date.getMinutes()).slice(-2); // 获取分钟,并保证两位数
  330. const seconds = ('0' + date.getSeconds()).slice(-2); // 获取秒钟,并保证两位数
  331. return `${year}-${month}-${day} ${hours}:${minutes}:${seconds}`; // 格式化输出
  332. };
  333. // 修改renderDuration函数以包含颜色逻辑
  334. function renderDuration(submit_time, finishTime) {
  335. if (!submit_time || !finishTime) return 'N/A';
  336. const start = new Date(submit_time);
  337. const finish = new Date(finishTime);
  338. const durationMs = finish - start;
  339. const durationSec = (durationMs / 1000).toFixed(1);
  340. const color = durationSec > 60 ? 'red' : 'green';
  341. return (
  342. <Tag color={color} size='large' shape='circle'>
  343. {durationSec} {t('秒')}
  344. </Tag>
  345. );
  346. }
  347. // 定义所有列
  348. const allColumns = [
  349. {
  350. key: COLUMN_KEYS.SUBMIT_TIME,
  351. title: t('提交时间'),
  352. dataIndex: 'submit_time',
  353. render: (text, record, index) => {
  354. return <div>{renderTimestamp(text / 1000)}</div>;
  355. },
  356. },
  357. {
  358. key: COLUMN_KEYS.DURATION,
  359. title: t('花费时间'),
  360. dataIndex: 'finish_time',
  361. render: (finish, record) => {
  362. return renderDuration(record.submit_time, finish);
  363. },
  364. },
  365. {
  366. key: COLUMN_KEYS.CHANNEL,
  367. title: t('渠道'),
  368. dataIndex: 'channel_id',
  369. className: isAdmin() ? 'tableShow' : 'tableHiddle',
  370. render: (text, record, index) => {
  371. return isAdminUser ? (
  372. <div>
  373. <Tag
  374. color={colors[parseInt(text) % colors.length]}
  375. size='large'
  376. shape='circle'
  377. onClick={() => {
  378. copyText(text);
  379. }}
  380. >
  381. {' '}
  382. {text}{' '}
  383. </Tag>
  384. </div>
  385. ) : (
  386. <></>
  387. );
  388. },
  389. },
  390. {
  391. key: COLUMN_KEYS.TYPE,
  392. title: t('类型'),
  393. dataIndex: 'action',
  394. render: (text, record, index) => {
  395. return <div>{renderType(text)}</div>;
  396. },
  397. },
  398. {
  399. key: COLUMN_KEYS.TASK_ID,
  400. title: t('任务ID'),
  401. dataIndex: 'mj_id',
  402. render: (text, record, index) => {
  403. return <div>{text}</div>;
  404. },
  405. },
  406. {
  407. key: COLUMN_KEYS.SUBMIT_RESULT,
  408. title: t('提交结果'),
  409. dataIndex: 'code',
  410. className: isAdmin() ? 'tableShow' : 'tableHiddle',
  411. render: (text, record, index) => {
  412. return isAdminUser ? <div>{renderCode(text)}</div> : <></>;
  413. },
  414. },
  415. {
  416. key: COLUMN_KEYS.TASK_STATUS,
  417. title: t('任务状态'),
  418. dataIndex: 'status',
  419. className: isAdmin() ? 'tableShow' : 'tableHiddle',
  420. render: (text, record, index) => {
  421. return <div>{renderStatus(text)}</div>;
  422. },
  423. },
  424. {
  425. key: COLUMN_KEYS.PROGRESS,
  426. title: t('进度'),
  427. dataIndex: 'progress',
  428. render: (text, record, index) => {
  429. return (
  430. <div>
  431. {
  432. <Progress
  433. stroke={
  434. record.status === 'FAILURE'
  435. ? 'var(--semi-color-warning)'
  436. : null
  437. }
  438. percent={text ? parseInt(text.replace('%', '')) : 0}
  439. showInfo={true}
  440. aria-label='drawing progress'
  441. style={{ minWidth: '160px' }}
  442. />
  443. }
  444. </div>
  445. );
  446. },
  447. },
  448. {
  449. key: COLUMN_KEYS.IMAGE,
  450. title: t('结果图片'),
  451. dataIndex: 'image_url',
  452. render: (text, record, index) => {
  453. if (!text) {
  454. return t('无');
  455. }
  456. return (
  457. <Button
  458. onClick={() => {
  459. setModalImageUrl(text);
  460. setIsModalOpenurl(true);
  461. }}
  462. className="!rounded-full"
  463. >
  464. {t('查看图片')}
  465. </Button>
  466. );
  467. },
  468. },
  469. {
  470. key: COLUMN_KEYS.PROMPT,
  471. title: 'Prompt',
  472. dataIndex: 'prompt',
  473. render: (text, record, index) => {
  474. if (!text) {
  475. return t('无');
  476. }
  477. return (
  478. <Typography.Text
  479. ellipsis={{ showTooltip: true }}
  480. style={{ width: 100 }}
  481. onClick={() => {
  482. setModalContent(text);
  483. setIsModalOpen(true);
  484. }}
  485. >
  486. {text}
  487. </Typography.Text>
  488. );
  489. },
  490. },
  491. {
  492. key: COLUMN_KEYS.PROMPT_EN,
  493. title: 'PromptEn',
  494. dataIndex: 'prompt_en',
  495. render: (text, record, index) => {
  496. if (!text) {
  497. return t('无');
  498. }
  499. return (
  500. <Typography.Text
  501. ellipsis={{ showTooltip: true }}
  502. style={{ width: 100 }}
  503. onClick={() => {
  504. setModalContent(text);
  505. setIsModalOpen(true);
  506. }}
  507. >
  508. {text}
  509. </Typography.Text>
  510. );
  511. },
  512. },
  513. {
  514. key: COLUMN_KEYS.FAIL_REASON,
  515. title: t('失败原因'),
  516. dataIndex: 'fail_reason',
  517. fixed: 'right',
  518. render: (text, record, index) => {
  519. if (!text) {
  520. return t('无');
  521. }
  522. return (
  523. <Typography.Text
  524. ellipsis={{ showTooltip: true }}
  525. style={{ width: 100 }}
  526. onClick={() => {
  527. setModalContent(text);
  528. setIsModalOpen(true);
  529. }}
  530. >
  531. {text}
  532. </Typography.Text>
  533. );
  534. },
  535. },
  536. ];
  537. // 根据可见性设置过滤列
  538. const getVisibleColumns = () => {
  539. return allColumns.filter((column) => visibleColumns[column.key]);
  540. };
  541. const [logs, setLogs] = useState([]);
  542. const [loading, setLoading] = useState(true);
  543. const [activePage, setActivePage] = useState(1);
  544. const [logCount, setLogCount] = useState(ITEMS_PER_PAGE);
  545. const [pageSize, setPageSize] = useState(ITEMS_PER_PAGE);
  546. const [isModalOpenurl, setIsModalOpenurl] = useState(false);
  547. const [showBanner, setShowBanner] = useState(false);
  548. // 定义模态框图片URL的状态和更新函数
  549. const [modalImageUrl, setModalImageUrl] = useState('');
  550. let now = new Date();
  551. // Form 初始值
  552. const formInitValues = {
  553. channel_id: '',
  554. mj_id: '',
  555. dateRange: [
  556. timestamp2string(now.getTime() / 1000 - 2592000),
  557. timestamp2string(now.getTime() / 1000 + 3600)
  558. ],
  559. };
  560. // Form API 引用
  561. const [formApi, setFormApi] = useState(null);
  562. const [stat, setStat] = useState({
  563. quota: 0,
  564. token: 0,
  565. });
  566. // 获取表单值的辅助函数
  567. const getFormValues = () => {
  568. const formValues = formApi ? formApi.getValues() : {};
  569. // 处理时间范围
  570. let start_timestamp = timestamp2string(now.getTime() / 1000 - 2592000);
  571. let end_timestamp = timestamp2string(now.getTime() / 1000 + 3600);
  572. if (formValues.dateRange && Array.isArray(formValues.dateRange) && formValues.dateRange.length === 2) {
  573. start_timestamp = formValues.dateRange[0];
  574. end_timestamp = formValues.dateRange[1];
  575. }
  576. return {
  577. channel_id: formValues.channel_id || '',
  578. mj_id: formValues.mj_id || '',
  579. start_timestamp,
  580. end_timestamp,
  581. };
  582. };
  583. const setLogsFormat = (logs) => {
  584. for (let i = 0; i < logs.length; i++) {
  585. logs[i].timestamp2string = timestamp2string(logs[i].created_at);
  586. logs[i].key = '' + logs[i].id;
  587. }
  588. // data.key = '' + data.id
  589. setLogs(logs);
  590. setLogCount(logs.length + pageSize);
  591. // console.log(logCount);
  592. };
  593. const loadLogs = async (startIdx, pageSize = ITEMS_PER_PAGE) => {
  594. setLoading(true);
  595. let url = '';
  596. const { channel_id, mj_id, start_timestamp, end_timestamp } = getFormValues();
  597. let localStartTimestamp = Date.parse(start_timestamp);
  598. let localEndTimestamp = Date.parse(end_timestamp);
  599. if (isAdminUser) {
  600. url = `/api/mj/?p=${startIdx}&page_size=${pageSize}&channel_id=${channel_id}&mj_id=${mj_id}&start_timestamp=${localStartTimestamp}&end_timestamp=${localEndTimestamp}`;
  601. } else {
  602. url = `/api/mj/self/?p=${startIdx}&page_size=${pageSize}&mj_id=${mj_id}&start_timestamp=${localStartTimestamp}&end_timestamp=${localEndTimestamp}`;
  603. }
  604. const res = await API.get(url);
  605. const { success, message, data } = res.data;
  606. if (success) {
  607. if (startIdx === 0) {
  608. setLogsFormat(data);
  609. } else {
  610. let newLogs = [...logs];
  611. newLogs.splice(startIdx * pageSize, data.length, ...data);
  612. setLogsFormat(newLogs);
  613. }
  614. } else {
  615. showError(message);
  616. }
  617. setLoading(false);
  618. };
  619. const pageData = logs.slice(
  620. (activePage - 1) * pageSize,
  621. activePage * pageSize,
  622. );
  623. const handlePageChange = (page) => {
  624. setActivePage(page);
  625. if (page === Math.ceil(logs.length / pageSize) + 1) {
  626. // In this case we have to load more data and then append them.
  627. loadLogs(page - 1, pageSize).then((r) => { });
  628. }
  629. };
  630. const handlePageSizeChange = async (size) => {
  631. localStorage.setItem('mj-page-size', size + '');
  632. setPageSize(size);
  633. setActivePage(1);
  634. await loadLogs(0, size);
  635. };
  636. const refresh = async () => {
  637. // setLoading(true);
  638. setActivePage(1);
  639. await loadLogs(0, pageSize);
  640. };
  641. const copyText = async (text) => {
  642. if (await copy(text)) {
  643. showSuccess(t('已复制:') + text);
  644. } else {
  645. // setSearchKeyword(text);
  646. Modal.error({ title: t('无法复制到剪贴板,请手动复制'), content: text });
  647. }
  648. };
  649. useEffect(() => {
  650. const localPageSize = parseInt(localStorage.getItem('mj-page-size')) || ITEMS_PER_PAGE;
  651. setPageSize(localPageSize);
  652. loadLogs(0, localPageSize).then();
  653. }, []);
  654. useEffect(() => {
  655. const mjNotifyEnabled = localStorage.getItem('mj_notify_enabled');
  656. if (mjNotifyEnabled !== 'true') {
  657. setShowBanner(true);
  658. }
  659. }, []);
  660. // 列选择器模态框
  661. const renderColumnSelector = () => {
  662. return (
  663. <Modal
  664. title={t('列设置')}
  665. visible={showColumnSelector}
  666. onCancel={() => setShowColumnSelector(false)}
  667. footer={
  668. <div className="flex justify-end">
  669. <Button
  670. theme="light"
  671. onClick={() => initDefaultColumns()}
  672. className="!rounded-full"
  673. >
  674. {t('重置')}
  675. </Button>
  676. <Button
  677. theme="light"
  678. onClick={() => setShowColumnSelector(false)}
  679. className="!rounded-full"
  680. >
  681. {t('取消')}
  682. </Button>
  683. <Button
  684. type='primary'
  685. onClick={() => setShowColumnSelector(false)}
  686. className="!rounded-full"
  687. >
  688. {t('确定')}
  689. </Button>
  690. </div>
  691. }
  692. >
  693. <div style={{ marginBottom: 20 }}>
  694. <Checkbox
  695. checked={Object.values(visibleColumns).every((v) => v === true)}
  696. indeterminate={
  697. Object.values(visibleColumns).some((v) => v === true) &&
  698. !Object.values(visibleColumns).every((v) => v === true)
  699. }
  700. onChange={(e) => handleSelectAll(e.target.checked)}
  701. >
  702. {t('全选')}
  703. </Checkbox>
  704. </div>
  705. <div className="flex flex-wrap max-h-96 overflow-y-auto rounded-lg p-4" style={{ border: '1px solid var(--semi-color-border)' }}>
  706. {allColumns.map((column) => {
  707. // 为非管理员用户跳过管理员专用列
  708. if (
  709. !isAdminUser &&
  710. (column.key === COLUMN_KEYS.CHANNEL ||
  711. column.key === COLUMN_KEYS.SUBMIT_RESULT)
  712. ) {
  713. return null;
  714. }
  715. return (
  716. <div key={column.key} className="w-1/2 mb-4 pr-2">
  717. <Checkbox
  718. checked={!!visibleColumns[column.key]}
  719. onChange={(e) =>
  720. handleColumnVisibilityChange(column.key, e.target.checked)
  721. }
  722. >
  723. {column.title}
  724. </Checkbox>
  725. </div>
  726. );
  727. })}
  728. </div>
  729. </Modal>
  730. );
  731. };
  732. return (
  733. <>
  734. {renderColumnSelector()}
  735. <Layout>
  736. <Card
  737. className="!rounded-2xl mb-4"
  738. title={
  739. <div className="flex flex-col w-full">
  740. <div className="flex flex-col md:flex-row justify-between items-center">
  741. <div className="flex items-center text-orange-500 mb-2 md:mb-0">
  742. <IconEyeOpened className="mr-2" />
  743. {loading ? (
  744. <Skeleton.Title
  745. style={{
  746. width: 300,
  747. marginBottom: 0,
  748. marginTop: 0
  749. }}
  750. />
  751. ) : (
  752. <Text>
  753. {isAdminUser && showBanner
  754. ? t('当前未开启Midjourney回调,部分项目可能无法获得绘图结果,可在运营设置中开启。')
  755. : t('Midjourney 任务记录')}
  756. </Text>
  757. )}
  758. </div>
  759. </div>
  760. <Divider margin="12px" />
  761. {/* 搜索表单区域 */}
  762. <Form
  763. initValues={formInitValues}
  764. getFormApi={(api) => setFormApi(api)}
  765. onSubmit={refresh}
  766. allowEmpty={true}
  767. autoComplete="off"
  768. layout="vertical"
  769. trigger="change"
  770. stopValidateWithError={false}
  771. >
  772. <div className="flex flex-col gap-4">
  773. <div className="grid grid-cols-1 md:grid-cols-2 lg:grid-cols-4 gap-4">
  774. {/* 时间选择器 */}
  775. <div className="col-span-1 lg:col-span-2">
  776. <Form.DatePicker
  777. field='dateRange'
  778. className="w-full"
  779. type='dateTimeRange'
  780. placeholder={[t('开始时间'), t('结束时间')]}
  781. showClear
  782. pure
  783. />
  784. </div>
  785. {/* 任务 ID */}
  786. <Form.Input
  787. field='mj_id'
  788. prefix={<IconSearch />}
  789. placeholder={t('任务 ID')}
  790. className="!rounded-full"
  791. showClear
  792. pure
  793. />
  794. {/* 渠道 ID - 仅管理员可见 */}
  795. {isAdminUser && (
  796. <Form.Input
  797. field='channel_id'
  798. prefix={<IconSearch />}
  799. placeholder={t('渠道 ID')}
  800. className="!rounded-full"
  801. showClear
  802. pure
  803. />
  804. )}
  805. </div>
  806. {/* 操作按钮区域 */}
  807. <div className="flex justify-between items-center pt-2">
  808. <div></div>
  809. <div className="flex gap-2">
  810. <Button
  811. type='primary'
  812. htmlType='submit'
  813. loading={loading}
  814. className="!rounded-full"
  815. >
  816. {t('查询')}
  817. </Button>
  818. <Button
  819. theme='light'
  820. onClick={() => {
  821. if (formApi) {
  822. formApi.reset();
  823. // 重置后立即查询,使用setTimeout确保表单重置完成
  824. setTimeout(() => {
  825. refresh();
  826. }, 100);
  827. }
  828. }}
  829. className="!rounded-full"
  830. >
  831. {t('重置')}
  832. </Button>
  833. <Button
  834. theme='light'
  835. type='tertiary'
  836. icon={<IconSetting />}
  837. onClick={() => setShowColumnSelector(true)}
  838. className="!rounded-full"
  839. >
  840. {t('列设置')}
  841. </Button>
  842. </div>
  843. </div>
  844. </div>
  845. </Form>
  846. </div>
  847. }
  848. shadows='always'
  849. bordered={false}
  850. >
  851. <Table
  852. columns={getVisibleColumns()}
  853. dataSource={pageData}
  854. rowKey='key'
  855. loading={loading}
  856. scroll={{ x: 'max-content' }}
  857. className="rounded-xl overflow-hidden"
  858. size="middle"
  859. pagination={{
  860. formatPageText: (page) =>
  861. t('第 {{start}} - {{end}} 条,共 {{total}} 条', {
  862. start: page.currentStart,
  863. end: page.currentEnd,
  864. total: logCount,
  865. }),
  866. currentPage: activePage,
  867. pageSize: pageSize,
  868. total: logCount,
  869. pageSizeOptions: [10, 20, 50, 100],
  870. showSizeChanger: true,
  871. onPageSizeChange: (size) => {
  872. handlePageSizeChange(size);
  873. },
  874. onPageChange: handlePageChange,
  875. }}
  876. />
  877. </Card>
  878. <Modal
  879. visible={isModalOpen}
  880. onOk={() => setIsModalOpen(false)}
  881. onCancel={() => setIsModalOpen(false)}
  882. closable={null}
  883. bodyStyle={{ height: '400px', overflow: 'auto' }} // 设置模态框内容区域样式
  884. width={800} // 设置模态框宽度
  885. >
  886. <p style={{ whiteSpace: 'pre-line' }}>{modalContent}</p>
  887. </Modal>
  888. <ImagePreview
  889. src={modalImageUrl}
  890. visible={isModalOpenurl}
  891. onVisibleChange={(visible) => setIsModalOpenurl(visible)}
  892. />
  893. </Layout>
  894. </>
  895. );
  896. };
  897. export default LogsTable;