ChannelsTable.js 67 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197
  1. import React, { useEffect, useState, useMemo, useRef } from 'react';
  2. import {
  3. API,
  4. showError,
  5. showInfo,
  6. showSuccess,
  7. timestamp2string,
  8. renderGroup,
  9. renderQuota,
  10. getChannelIcon,
  11. renderQuotaWithAmount
  12. } from '../../helpers/index.js';
  13. import { CHANNEL_OPTIONS, ITEMS_PER_PAGE, MODEL_TABLE_PAGE_SIZE } from '../../constants/index.js';
  14. import {
  15. Button,
  16. Divider,
  17. Dropdown,
  18. Empty,
  19. Input,
  20. InputNumber,
  21. Modal,
  22. Space,
  23. SplitButtonGroup,
  24. Switch,
  25. Table,
  26. Tag,
  27. Tooltip,
  28. Typography,
  29. Checkbox,
  30. Card,
  31. Form,
  32. Tabs,
  33. TabPane,
  34. Select
  35. } from '@douyinfe/semi-ui';
  36. import {
  37. IllustrationNoResult,
  38. IllustrationNoResultDark
  39. } from '@douyinfe/semi-illustrations';
  40. import EditChannel from '../../pages/Channel/EditChannel.js';
  41. import {
  42. IconTreeTriangleDown,
  43. IconSearch,
  44. IconMore,
  45. IconList
  46. } from '@douyinfe/semi-icons';
  47. import { loadChannelModels, isMobile, copy } from '../../helpers';
  48. import EditTagModal from '../../pages/Channel/EditTagModal.js';
  49. import { useTranslation } from 'react-i18next';
  50. import { useTableCompactMode } from '../../hooks/useTableCompactMode';
  51. const ChannelsTable = () => {
  52. const { t } = useTranslation();
  53. let type2label = undefined;
  54. const renderType = (type, multiKey = false) => {
  55. if (!type2label) {
  56. type2label = new Map();
  57. for (let i = 0; i < CHANNEL_OPTIONS.length; i++) {
  58. type2label[CHANNEL_OPTIONS[i].value] = CHANNEL_OPTIONS[i];
  59. }
  60. type2label[0] = { value: 0, label: t('未知类型'), color: 'grey' };
  61. }
  62. let icon = getChannelIcon(type);
  63. if (multiKey) {
  64. icon = (
  65. <div className="flex items-center gap-1">
  66. <IconList className="text-blue-500" />
  67. {icon}
  68. </div>
  69. );
  70. }
  71. return (
  72. <Tag
  73. size='large'
  74. color={type2label[type]?.color}
  75. shape='circle'
  76. prefixIcon={icon}
  77. >
  78. {type2label[type]?.label}
  79. </Tag>
  80. );
  81. };
  82. const renderTagType = () => {
  83. return (
  84. <Tag
  85. color='light-blue'
  86. size='large'
  87. shape='circle'
  88. type='light'
  89. >
  90. {t('标签聚合')}
  91. </Tag>
  92. );
  93. };
  94. const renderStatus = (status, channelInfo = undefined) => {
  95. if (channelInfo) {
  96. if (channelInfo.is_multi_key) {
  97. let keySize = channelInfo.multi_key_size;
  98. let enabledKeySize = keySize;
  99. if (channelInfo.multi_key_status_list) {
  100. // multi_key_status_list is a map, key is key, value is status
  101. // get multi_key_status_list length
  102. enabledKeySize = keySize - Object.keys(channelInfo.multi_key_status_list).length;
  103. }
  104. return renderMultiKeyStatus(status, keySize, enabledKeySize);
  105. }
  106. }
  107. switch (status) {
  108. case 1:
  109. return (
  110. <Tag size='large' color='green' shape='circle'>
  111. {t('已启用')}
  112. </Tag>
  113. );
  114. case 2:
  115. return (
  116. <Tag size='large' color='red' shape='circle'>
  117. {t('已禁用')}
  118. </Tag>
  119. );
  120. case 3:
  121. return (
  122. <Tag size='large' color='yellow' shape='circle'>
  123. {t('自动禁用')}
  124. </Tag>
  125. );
  126. default:
  127. return (
  128. <Tag size='large' color='grey' shape='circle'>
  129. {t('未知状态')}
  130. </Tag>
  131. );
  132. }
  133. };
  134. const renderMultiKeyStatus = (status, keySize, enabledKeySize) => {
  135. switch (status) {
  136. case 1:
  137. return (
  138. <Tag size='large' color='green' shape='circle'>
  139. {t('已启用')} {enabledKeySize}/{keySize}
  140. </Tag>
  141. );
  142. case 2:
  143. return (
  144. <Tag size='large' color='red' shape='circle'>
  145. {t('已禁用')} {enabledKeySize}/{keySize}
  146. </Tag>
  147. );
  148. case 3:
  149. return (
  150. <Tag size='large' color='yellow' shape='circle'>
  151. {t('自动禁用')} {enabledKeySize}/{keySize}
  152. </Tag>
  153. );
  154. default:
  155. return (
  156. <Tag size='large' color='grey' shape='circle'>
  157. {t('未知状态')} {enabledKeySize}/{keySize}
  158. </Tag>
  159. );
  160. }
  161. }
  162. const renderResponseTime = (responseTime) => {
  163. let time = responseTime / 1000;
  164. time = time.toFixed(2) + t(' 秒');
  165. if (responseTime === 0) {
  166. return (
  167. <Tag size='large' color='grey' shape='circle'>
  168. {t('未测试')}
  169. </Tag>
  170. );
  171. } else if (responseTime <= 1000) {
  172. return (
  173. <Tag size='large' color='green' shape='circle'>
  174. {time}
  175. </Tag>
  176. );
  177. } else if (responseTime <= 3000) {
  178. return (
  179. <Tag size='large' color='lime' shape='circle'>
  180. {time}
  181. </Tag>
  182. );
  183. } else if (responseTime <= 5000) {
  184. return (
  185. <Tag size='large' color='yellow' shape='circle'>
  186. {time}
  187. </Tag>
  188. );
  189. } else {
  190. return (
  191. <Tag size='large' color='red' shape='circle'>
  192. {time}
  193. </Tag>
  194. );
  195. }
  196. };
  197. // Define column keys for selection
  198. const COLUMN_KEYS = {
  199. ID: 'id',
  200. NAME: 'name',
  201. GROUP: 'group',
  202. TYPE: 'type',
  203. STATUS: 'status',
  204. RESPONSE_TIME: 'response_time',
  205. BALANCE: 'balance',
  206. PRIORITY: 'priority',
  207. WEIGHT: 'weight',
  208. OPERATE: 'operate',
  209. };
  210. // State for column visibility
  211. const [visibleColumns, setVisibleColumns] = useState({});
  212. const [showColumnSelector, setShowColumnSelector] = useState(false);
  213. // 状态筛选 all / enabled / disabled
  214. const [statusFilter, setStatusFilter] = useState(
  215. localStorage.getItem('channel-status-filter') || 'all'
  216. );
  217. // Load saved column preferences from localStorage
  218. useEffect(() => {
  219. const savedColumns = localStorage.getItem('channels-table-columns');
  220. if (savedColumns) {
  221. try {
  222. const parsed = JSON.parse(savedColumns);
  223. // Make sure all columns are accounted for
  224. const defaults = getDefaultColumnVisibility();
  225. const merged = { ...defaults, ...parsed };
  226. setVisibleColumns(merged);
  227. } catch (e) {
  228. console.error('Failed to parse saved column preferences', e);
  229. initDefaultColumns();
  230. }
  231. } else {
  232. initDefaultColumns();
  233. }
  234. }, []);
  235. // Update table when column visibility changes
  236. useEffect(() => {
  237. if (Object.keys(visibleColumns).length > 0) {
  238. // Save to localStorage
  239. localStorage.setItem(
  240. 'channels-table-columns',
  241. JSON.stringify(visibleColumns),
  242. );
  243. }
  244. }, [visibleColumns]);
  245. // Get default column visibility
  246. const getDefaultColumnVisibility = () => {
  247. return {
  248. [COLUMN_KEYS.ID]: true,
  249. [COLUMN_KEYS.NAME]: true,
  250. [COLUMN_KEYS.GROUP]: true,
  251. [COLUMN_KEYS.TYPE]: true,
  252. [COLUMN_KEYS.STATUS]: true,
  253. [COLUMN_KEYS.RESPONSE_TIME]: true,
  254. [COLUMN_KEYS.BALANCE]: true,
  255. [COLUMN_KEYS.PRIORITY]: true,
  256. [COLUMN_KEYS.WEIGHT]: true,
  257. [COLUMN_KEYS.OPERATE]: true,
  258. };
  259. };
  260. // Initialize default column visibility
  261. const initDefaultColumns = () => {
  262. const defaults = getDefaultColumnVisibility();
  263. setVisibleColumns(defaults);
  264. };
  265. // Handle column visibility change
  266. const handleColumnVisibilityChange = (columnKey, checked) => {
  267. const updatedColumns = { ...visibleColumns, [columnKey]: checked };
  268. setVisibleColumns(updatedColumns);
  269. };
  270. // Handle "Select All" checkbox
  271. const handleSelectAll = (checked) => {
  272. const allKeys = Object.keys(COLUMN_KEYS).map((key) => COLUMN_KEYS[key]);
  273. const updatedColumns = {};
  274. allKeys.forEach((key) => {
  275. updatedColumns[key] = checked;
  276. });
  277. setVisibleColumns(updatedColumns);
  278. };
  279. // Define all columns with keys
  280. const allColumns = [
  281. {
  282. key: COLUMN_KEYS.ID,
  283. title: t('ID'),
  284. dataIndex: 'id',
  285. },
  286. {
  287. key: COLUMN_KEYS.NAME,
  288. title: t('名称'),
  289. dataIndex: 'name',
  290. },
  291. {
  292. key: COLUMN_KEYS.GROUP,
  293. title: t('分组'),
  294. dataIndex: 'group',
  295. render: (text, record, index) => (
  296. <div>
  297. <Space spacing={2}>
  298. {text
  299. ?.split(',')
  300. .sort((a, b) => {
  301. if (a === 'default') return -1;
  302. if (b === 'default') return 1;
  303. return a.localeCompare(b);
  304. })
  305. .map((item, index) => renderGroup(item))}
  306. </Space>
  307. </div>
  308. ),
  309. },
  310. {
  311. key: COLUMN_KEYS.TYPE,
  312. title: t('类型'),
  313. dataIndex: 'type',
  314. render: (text, record, index) => {
  315. if (record.children === undefined) {
  316. if (record.channel_info) {
  317. if (record.channel_info.is_multi_key) {
  318. return <>{renderType(text, record.channel_info)}</>;
  319. }
  320. }
  321. return <>{renderType(text)}</>;
  322. } else {
  323. return <>{renderTagType()}</>;
  324. }
  325. },
  326. },
  327. {
  328. key: COLUMN_KEYS.STATUS,
  329. title: t('状态'),
  330. dataIndex: 'status',
  331. render: (text, record, index) => {
  332. if (text === 3) {
  333. if (record.other_info === '') {
  334. record.other_info = '{}';
  335. }
  336. let otherInfo = JSON.parse(record.other_info);
  337. let reason = otherInfo['status_reason'];
  338. let time = otherInfo['status_time'];
  339. return (
  340. <div>
  341. <Tooltip
  342. content={t('原因:') + reason + t(',时间:') + timestamp2string(time)}
  343. >
  344. {renderStatus(text, record.channel_info)}
  345. </Tooltip>
  346. </div>
  347. );
  348. } else {
  349. return renderStatus(text, record.channel_info);
  350. }
  351. },
  352. },
  353. {
  354. key: COLUMN_KEYS.RESPONSE_TIME,
  355. title: t('响应时间'),
  356. dataIndex: 'response_time',
  357. render: (text, record, index) => (
  358. <div>{renderResponseTime(text)}</div>
  359. ),
  360. },
  361. {
  362. key: COLUMN_KEYS.BALANCE,
  363. title: t('已用/剩余'),
  364. dataIndex: 'expired_time',
  365. render: (text, record, index) => {
  366. if (record.children === undefined) {
  367. return (
  368. <div>
  369. <Space spacing={1}>
  370. <Tooltip content={t('已用额度')}>
  371. <Tag color='white' type='ghost' size='large' shape='circle'>
  372. {renderQuota(record.used_quota)}
  373. </Tag>
  374. </Tooltip>
  375. <Tooltip content={t('剩余额度$') + record.balance + t(',点击更新')}>
  376. <Tag
  377. color='white'
  378. type='ghost'
  379. size='large'
  380. shape='circle'
  381. onClick={() => updateChannelBalance(record)}
  382. >
  383. {renderQuotaWithAmount(record.balance)}
  384. </Tag>
  385. </Tooltip>
  386. </Space>
  387. </div>
  388. );
  389. } else {
  390. return (
  391. <Tooltip content={t('已用额度')}>
  392. <Tag color='white' type='ghost' size='large' shape='circle'>
  393. {renderQuota(record.used_quota)}
  394. </Tag>
  395. </Tooltip>
  396. );
  397. }
  398. },
  399. },
  400. {
  401. key: COLUMN_KEYS.PRIORITY,
  402. title: t('优先级'),
  403. dataIndex: 'priority',
  404. render: (text, record, index) => {
  405. if (record.children === undefined) {
  406. return (
  407. <div>
  408. <InputNumber
  409. style={{ width: 70 }}
  410. name='priority'
  411. onBlur={(e) => {
  412. manageChannel(record.id, 'priority', record, e.target.value);
  413. }}
  414. keepFocus={true}
  415. innerButtons
  416. defaultValue={record.priority}
  417. min={-999}
  418. size="small"
  419. />
  420. </div>
  421. );
  422. } else {
  423. return (
  424. <InputNumber
  425. style={{ width: 70 }}
  426. name='priority'
  427. keepFocus={true}
  428. onBlur={(e) => {
  429. Modal.warning({
  430. title: t('修改子渠道优先级'),
  431. content: t('确定要修改所有子渠道优先级为 ') + e.target.value + t(' 吗?'),
  432. onOk: () => {
  433. if (e.target.value === '') {
  434. return;
  435. }
  436. submitTagEdit('priority', {
  437. tag: record.key,
  438. priority: e.target.value,
  439. });
  440. },
  441. });
  442. }}
  443. innerButtons
  444. defaultValue={record.priority}
  445. min={-999}
  446. size="small"
  447. />
  448. );
  449. }
  450. },
  451. },
  452. {
  453. key: COLUMN_KEYS.WEIGHT,
  454. title: t('权重'),
  455. dataIndex: 'weight',
  456. render: (text, record, index) => {
  457. if (record.children === undefined) {
  458. return (
  459. <div>
  460. <InputNumber
  461. style={{ width: 70 }}
  462. name='weight'
  463. onBlur={(e) => {
  464. manageChannel(record.id, 'weight', record, e.target.value);
  465. }}
  466. keepFocus={true}
  467. innerButtons
  468. defaultValue={record.weight}
  469. min={0}
  470. size="small"
  471. />
  472. </div>
  473. );
  474. } else {
  475. return (
  476. <InputNumber
  477. style={{ width: 70 }}
  478. name='weight'
  479. keepFocus={true}
  480. onBlur={(e) => {
  481. Modal.warning({
  482. title: t('修改子渠道权重'),
  483. content: t('确定要修改所有子渠道权重为 ') + e.target.value + t(' 吗?'),
  484. onOk: () => {
  485. if (e.target.value === '') {
  486. return;
  487. }
  488. submitTagEdit('weight', {
  489. tag: record.key,
  490. weight: e.target.value,
  491. });
  492. },
  493. });
  494. }}
  495. innerButtons
  496. defaultValue={record.weight}
  497. min={-999}
  498. size="small"
  499. />
  500. );
  501. }
  502. },
  503. },
  504. {
  505. key: COLUMN_KEYS.OPERATE,
  506. title: '',
  507. dataIndex: 'operate',
  508. fixed: 'right',
  509. render: (text, record, index) => {
  510. if (record.children === undefined) {
  511. // 创建更多操作的下拉菜单项
  512. const moreMenuItems = [
  513. {
  514. node: 'item',
  515. name: t('删除'),
  516. type: 'danger',
  517. onClick: () => {
  518. Modal.confirm({
  519. title: t('确定是否要删除此渠道?'),
  520. content: t('此修改将不可逆'),
  521. onOk: () => {
  522. manageChannel(record.id, 'delete', record).then(() => {
  523. removeRecord(record);
  524. });
  525. },
  526. });
  527. },
  528. },
  529. {
  530. node: 'item',
  531. name: t('复制'),
  532. type: 'primary',
  533. onClick: () => {
  534. Modal.confirm({
  535. title: t('确定是否要复制此渠道?'),
  536. content: t('复制渠道的所有信息'),
  537. onOk: () => copySelectedChannel(record),
  538. });
  539. },
  540. },
  541. ];
  542. return (
  543. <Space wrap>
  544. <SplitButtonGroup
  545. className="overflow-hidden"
  546. aria-label={t('测试单个渠道操作项目组')}
  547. >
  548. <Button
  549. theme='light'
  550. size="small"
  551. onClick={() => testChannel(record, '')}
  552. >
  553. {t('测试')}
  554. </Button>
  555. <Button
  556. theme='light'
  557. size="small"
  558. icon={<IconTreeTriangleDown />}
  559. onClick={() => {
  560. setCurrentTestChannel(record);
  561. setShowModelTestModal(true);
  562. }}
  563. />
  564. </SplitButtonGroup>
  565. {record.status === 1 ? (
  566. <Button
  567. theme='light'
  568. type='warning'
  569. size="small"
  570. onClick={() => manageChannel(record.id, 'disable', record)}
  571. >
  572. {t('禁用')}
  573. </Button>
  574. ) : (
  575. <Button
  576. theme='light'
  577. type='secondary'
  578. size="small"
  579. onClick={() => manageChannel(record.id, 'enable', record)}
  580. >
  581. {t('启用')}
  582. </Button>
  583. )}
  584. <Button
  585. theme='light'
  586. type='tertiary'
  587. size="small"
  588. onClick={() => {
  589. setEditingChannel(record);
  590. setShowEdit(true);
  591. }}
  592. >
  593. {t('编辑')}
  594. </Button>
  595. <Dropdown
  596. trigger='click'
  597. position='bottomRight'
  598. menu={moreMenuItems}
  599. >
  600. <Button
  601. icon={<IconMore />}
  602. theme='light'
  603. type='tertiary'
  604. size="small"
  605. />
  606. </Dropdown>
  607. </Space>
  608. );
  609. } else {
  610. // 标签操作按钮
  611. return (
  612. <Space wrap>
  613. <Button
  614. theme='light'
  615. type='secondary'
  616. size="small"
  617. onClick={() => manageTag(record.key, 'enable')}
  618. >
  619. {t('启用全部')}
  620. </Button>
  621. <Button
  622. theme='light'
  623. type='warning'
  624. size="small"
  625. onClick={() => manageTag(record.key, 'disable')}
  626. >
  627. {t('禁用全部')}
  628. </Button>
  629. <Button
  630. theme='light'
  631. type='tertiary'
  632. size="small"
  633. onClick={() => {
  634. setShowEditTag(true);
  635. setEditingTag(record.key);
  636. }}
  637. >
  638. {t('编辑')}
  639. </Button>
  640. </Space>
  641. );
  642. }
  643. },
  644. },
  645. ];
  646. const [channels, setChannels] = useState([]);
  647. const [loading, setLoading] = useState(true);
  648. const [activePage, setActivePage] = useState(1);
  649. const [idSort, setIdSort] = useState(false);
  650. const [searching, setSearching] = useState(false);
  651. const [pageSize, setPageSize] = useState(ITEMS_PER_PAGE);
  652. const [channelCount, setChannelCount] = useState(pageSize);
  653. const [groupOptions, setGroupOptions] = useState([]);
  654. const [showEdit, setShowEdit] = useState(false);
  655. const [enableBatchDelete, setEnableBatchDelete] = useState(false);
  656. const [editingChannel, setEditingChannel] = useState({
  657. id: undefined,
  658. });
  659. const [showEditTag, setShowEditTag] = useState(false);
  660. const [editingTag, setEditingTag] = useState('');
  661. const [selectedChannels, setSelectedChannels] = useState([]);
  662. const [enableTagMode, setEnableTagMode] = useState(false);
  663. const [showBatchSetTag, setShowBatchSetTag] = useState(false);
  664. const [batchSetTagValue, setBatchSetTagValue] = useState('');
  665. const [showModelTestModal, setShowModelTestModal] = useState(false);
  666. const [currentTestChannel, setCurrentTestChannel] = useState(null);
  667. const [modelSearchKeyword, setModelSearchKeyword] = useState('');
  668. const [modelTestResults, setModelTestResults] = useState({});
  669. const [testingModels, setTestingModels] = useState(new Set());
  670. const [selectedModelKeys, setSelectedModelKeys] = useState([]);
  671. const [isBatchTesting, setIsBatchTesting] = useState(false);
  672. const [testQueue, setTestQueue] = useState([]);
  673. const [isProcessingQueue, setIsProcessingQueue] = useState(false);
  674. const [modelTablePage, setModelTablePage] = useState(1);
  675. const [activeTypeKey, setActiveTypeKey] = useState('all');
  676. const [typeCounts, setTypeCounts] = useState({});
  677. const requestCounter = useRef(0);
  678. const [formApi, setFormApi] = useState(null);
  679. const [compactMode, setCompactMode] = useTableCompactMode('channels');
  680. const formInitValues = {
  681. searchKeyword: '',
  682. searchGroup: '',
  683. searchModel: '',
  684. };
  685. const allSelectingRef = useRef(false);
  686. // Filter columns based on visibility settings
  687. const getVisibleColumns = () => {
  688. return allColumns.filter((column) => visibleColumns[column.key]);
  689. };
  690. // Column selector modal
  691. const renderColumnSelector = () => {
  692. return (
  693. <Modal
  694. title={t('列设置')}
  695. visible={showColumnSelector}
  696. onCancel={() => setShowColumnSelector(false)}
  697. footer={
  698. <div className="flex justify-end">
  699. <Button
  700. theme="light"
  701. onClick={() => initDefaultColumns()}
  702. >
  703. {t('重置')}
  704. </Button>
  705. <Button
  706. theme="light"
  707. onClick={() => setShowColumnSelector(false)}
  708. >
  709. {t('取消')}
  710. </Button>
  711. <Button
  712. type='primary'
  713. onClick={() => setShowColumnSelector(false)}
  714. >
  715. {t('确定')}
  716. </Button>
  717. </div>
  718. }
  719. >
  720. <div style={{ marginBottom: 20 }}>
  721. <Checkbox
  722. checked={Object.values(visibleColumns).every((v) => v === true)}
  723. indeterminate={
  724. Object.values(visibleColumns).some((v) => v === true) &&
  725. !Object.values(visibleColumns).every((v) => v === true)
  726. }
  727. onChange={(e) => handleSelectAll(e.target.checked)}
  728. >
  729. {t('全选')}
  730. </Checkbox>
  731. </div>
  732. <div
  733. className="flex flex-wrap max-h-96 overflow-y-auto rounded-lg p-4"
  734. style={{ border: '1px solid var(--semi-color-border)' }}
  735. >
  736. {allColumns.map((column) => {
  737. // Skip columns without title
  738. if (!column.title) {
  739. return null;
  740. }
  741. return (
  742. <div
  743. key={column.key}
  744. className="w-1/2 mb-4 pr-2"
  745. >
  746. <Checkbox
  747. checked={!!visibleColumns[column.key]}
  748. onChange={(e) =>
  749. handleColumnVisibilityChange(column.key, e.target.checked)
  750. }
  751. >
  752. {column.title}
  753. </Checkbox>
  754. </div>
  755. );
  756. })}
  757. </div>
  758. </Modal>
  759. );
  760. };
  761. const removeRecord = (record) => {
  762. let newDataSource = [...channels];
  763. if (record.id != null) {
  764. let idx = newDataSource.findIndex((data) => {
  765. if (data.children !== undefined) {
  766. for (let i = 0; i < data.children.length; i++) {
  767. if (data.children[i].id === record.id) {
  768. data.children.splice(i, 1);
  769. return false;
  770. }
  771. }
  772. } else {
  773. return data.id === record.id;
  774. }
  775. });
  776. if (idx > -1) {
  777. newDataSource.splice(idx, 1);
  778. setChannels(newDataSource);
  779. }
  780. }
  781. };
  782. const setChannelFormat = (channels, enableTagMode) => {
  783. let channelDates = [];
  784. let channelTags = {};
  785. for (let i = 0; i < channels.length; i++) {
  786. channels[i].key = '' + channels[i].id;
  787. if (!enableTagMode) {
  788. channelDates.push(channels[i]);
  789. } else {
  790. let tag = channels[i].tag ? channels[i].tag : '';
  791. // find from channelTags
  792. let tagIndex = channelTags[tag];
  793. let tagChannelDates = undefined;
  794. if (tagIndex === undefined) {
  795. // not found, create a new tag
  796. channelTags[tag] = 1;
  797. tagChannelDates = {
  798. key: tag,
  799. id: tag,
  800. tag: tag,
  801. name: '标签:' + tag,
  802. group: '',
  803. used_quota: 0,
  804. response_time: 0,
  805. priority: -1,
  806. weight: -1,
  807. };
  808. tagChannelDates.children = [];
  809. channelDates.push(tagChannelDates);
  810. } else {
  811. // found, add to the tag
  812. tagChannelDates = channelDates.find((item) => item.key === tag);
  813. }
  814. if (tagChannelDates.priority === -1) {
  815. tagChannelDates.priority = channels[i].priority;
  816. } else {
  817. if (tagChannelDates.priority !== channels[i].priority) {
  818. tagChannelDates.priority = '';
  819. }
  820. }
  821. if (tagChannelDates.weight === -1) {
  822. tagChannelDates.weight = channels[i].weight;
  823. } else {
  824. if (tagChannelDates.weight !== channels[i].weight) {
  825. tagChannelDates.weight = '';
  826. }
  827. }
  828. if (tagChannelDates.group === '') {
  829. tagChannelDates.group = channels[i].group;
  830. } else {
  831. let channelGroupsStr = channels[i].group;
  832. channelGroupsStr.split(',').forEach((item, index) => {
  833. if (tagChannelDates.group.indexOf(item) === -1) {
  834. // join
  835. tagChannelDates.group += ',' + item;
  836. }
  837. });
  838. }
  839. tagChannelDates.children.push(channels[i]);
  840. if (channels[i].status === 1) {
  841. tagChannelDates.status = 1;
  842. }
  843. tagChannelDates.used_quota += channels[i].used_quota;
  844. tagChannelDates.response_time += channels[i].response_time;
  845. tagChannelDates.response_time = tagChannelDates.response_time / 2;
  846. }
  847. }
  848. setChannels(channelDates);
  849. };
  850. const loadChannels = async (
  851. page,
  852. pageSize,
  853. idSort,
  854. enableTagMode,
  855. typeKey = activeTypeKey,
  856. statusF,
  857. ) => {
  858. if (statusF === undefined) statusF = statusFilter;
  859. const { searchKeyword, searchGroup, searchModel } = getFormValues();
  860. if (searchKeyword !== '' || searchGroup !== '' || searchModel !== '') {
  861. setLoading(true);
  862. await searchChannels(enableTagMode, typeKey, statusF, page, pageSize, idSort);
  863. setLoading(false);
  864. return;
  865. }
  866. const reqId = ++requestCounter.current; // 记录当前请求序号
  867. setLoading(true);
  868. const typeParam = (typeKey !== 'all') ? `&type=${typeKey}` : '';
  869. const statusParam = statusF !== 'all' ? `&status=${statusF}` : '';
  870. const res = await API.get(
  871. `/api/channel/?p=${page}&page_size=${pageSize}&id_sort=${idSort}&tag_mode=${enableTagMode}${typeParam}${statusParam}`,
  872. );
  873. if (res === undefined || reqId !== requestCounter.current) {
  874. return;
  875. }
  876. const { success, message, data } = res.data;
  877. if (success) {
  878. const { items, total, type_counts } = data;
  879. if (type_counts) {
  880. const sumAll = Object.values(type_counts).reduce((acc, v) => acc + v, 0);
  881. setTypeCounts({ ...type_counts, all: sumAll });
  882. }
  883. setChannelFormat(items, enableTagMode);
  884. setChannelCount(total);
  885. } else {
  886. showError(message);
  887. }
  888. setLoading(false);
  889. };
  890. const copySelectedChannel = async (record) => {
  891. const channelToCopy = { ...record };
  892. channelToCopy.name += t('_复制');
  893. channelToCopy.created_time = null;
  894. channelToCopy.balance = 0;
  895. channelToCopy.used_quota = 0;
  896. delete channelToCopy.test_time;
  897. delete channelToCopy.response_time;
  898. if (!channelToCopy) {
  899. showError(t('渠道未找到,请刷新页面后重试。'));
  900. return;
  901. }
  902. try {
  903. const newChannel = { ...channelToCopy, id: undefined };
  904. const response = await API.post('/api/channel/', newChannel);
  905. if (response.data.success) {
  906. showSuccess(t('渠道复制成功'));
  907. await refresh();
  908. } else {
  909. showError(response.data.message);
  910. }
  911. } catch (error) {
  912. showError(t('渠道复制失败: ') + error.message);
  913. }
  914. };
  915. const refresh = async () => {
  916. const { searchKeyword, searchGroup, searchModel } = getFormValues();
  917. if (searchKeyword === '' && searchGroup === '' && searchModel === '') {
  918. await loadChannels(activePage, pageSize, idSort, enableTagMode);
  919. } else {
  920. await searchChannels(enableTagMode, activeTypeKey, statusFilter, activePage, pageSize, idSort);
  921. }
  922. };
  923. useEffect(() => {
  924. // console.log('default effect')
  925. const localIdSort = localStorage.getItem('id-sort') === 'true';
  926. const localPageSize =
  927. parseInt(localStorage.getItem('page-size')) || ITEMS_PER_PAGE;
  928. const localEnableTagMode = localStorage.getItem('enable-tag-mode') === 'true';
  929. const localEnableBatchDelete = localStorage.getItem('enable-batch-delete') === 'true';
  930. setIdSort(localIdSort);
  931. setPageSize(localPageSize);
  932. setEnableTagMode(localEnableTagMode);
  933. setEnableBatchDelete(localEnableBatchDelete);
  934. loadChannels(1, localPageSize, localIdSort, localEnableTagMode)
  935. .then()
  936. .catch((reason) => {
  937. showError(reason);
  938. });
  939. fetchGroups().then();
  940. loadChannelModels().then();
  941. }, []);
  942. const manageChannel = async (id, action, record, value) => {
  943. let data = { id };
  944. let res;
  945. switch (action) {
  946. case 'delete':
  947. res = await API.delete(`/api/channel/${id}/`);
  948. break;
  949. case 'enable':
  950. data.status = 1;
  951. res = await API.put('/api/channel/', data);
  952. break;
  953. case 'disable':
  954. data.status = 2;
  955. res = await API.put('/api/channel/', data);
  956. break;
  957. case 'priority':
  958. if (value === '') {
  959. return;
  960. }
  961. data.priority = parseInt(value);
  962. res = await API.put('/api/channel/', data);
  963. break;
  964. case 'weight':
  965. if (value === '') {
  966. return;
  967. }
  968. data.weight = parseInt(value);
  969. if (data.weight < 0) {
  970. data.weight = 0;
  971. }
  972. res = await API.put('/api/channel/', data);
  973. break;
  974. }
  975. const { success, message } = res.data;
  976. if (success) {
  977. showSuccess(t('操作成功完成!'));
  978. let channel = res.data.data;
  979. let newChannels = [...channels];
  980. if (action === 'delete') {
  981. } else {
  982. record.status = channel.status;
  983. }
  984. setChannels(newChannels);
  985. } else {
  986. showError(message);
  987. }
  988. };
  989. const manageTag = async (tag, action) => {
  990. console.log(tag, action);
  991. let res;
  992. switch (action) {
  993. case 'enable':
  994. res = await API.post('/api/channel/tag/enabled', {
  995. tag: tag,
  996. });
  997. break;
  998. case 'disable':
  999. res = await API.post('/api/channel/tag/disabled', {
  1000. tag: tag,
  1001. });
  1002. break;
  1003. }
  1004. const { success, message } = res.data;
  1005. if (success) {
  1006. showSuccess('操作成功完成!');
  1007. let newChannels = [...channels];
  1008. for (let i = 0; i < newChannels.length; i++) {
  1009. if (newChannels[i].tag === tag) {
  1010. let status = action === 'enable' ? 1 : 2;
  1011. newChannels[i]?.children?.forEach((channel) => {
  1012. channel.status = status;
  1013. });
  1014. newChannels[i].status = status;
  1015. }
  1016. }
  1017. setChannels(newChannels);
  1018. } else {
  1019. showError(message);
  1020. }
  1021. };
  1022. // 获取表单值的辅助函数
  1023. const getFormValues = () => {
  1024. const formValues = formApi ? formApi.getValues() : {};
  1025. return {
  1026. searchKeyword: formValues.searchKeyword || '',
  1027. searchGroup: formValues.searchGroup || '',
  1028. searchModel: formValues.searchModel || '',
  1029. };
  1030. };
  1031. const searchChannels = async (
  1032. enableTagMode,
  1033. typeKey = activeTypeKey,
  1034. statusF = statusFilter,
  1035. page = 1,
  1036. pageSz = pageSize,
  1037. sortFlag = idSort,
  1038. ) => {
  1039. const { searchKeyword, searchGroup, searchModel } = getFormValues();
  1040. setSearching(true);
  1041. try {
  1042. if (searchKeyword === '' && searchGroup === '' && searchModel === '') {
  1043. await loadChannels(page, pageSz, sortFlag, enableTagMode, typeKey, statusF);
  1044. return;
  1045. }
  1046. const typeParam = (typeKey !== 'all') ? `&type=${typeKey}` : '';
  1047. const statusParam = statusF !== 'all' ? `&status=${statusF}` : '';
  1048. const res = await API.get(
  1049. `/api/channel/search?keyword=${searchKeyword}&group=${searchGroup}&model=${searchModel}&id_sort=${sortFlag}&tag_mode=${enableTagMode}&p=${page}&page_size=${pageSz}${typeParam}${statusParam}`,
  1050. );
  1051. const { success, message, data } = res.data;
  1052. if (success) {
  1053. const { items = [], total = 0, type_counts = {} } = data;
  1054. const sumAll = Object.values(type_counts).reduce((acc, v) => acc + v, 0);
  1055. setTypeCounts({ ...type_counts, all: sumAll });
  1056. setChannelFormat(items, enableTagMode);
  1057. setChannelCount(total);
  1058. setActivePage(page);
  1059. } else {
  1060. showError(message);
  1061. }
  1062. } finally {
  1063. setSearching(false);
  1064. }
  1065. };
  1066. const updateChannelProperty = (channelId, updateFn) => {
  1067. // Create a new copy of channels array
  1068. const newChannels = [...channels];
  1069. let updated = false;
  1070. // Find and update the correct channel
  1071. newChannels.forEach((channel) => {
  1072. if (channel.children !== undefined) {
  1073. // If this is a tag group, search in its children
  1074. channel.children.forEach((child) => {
  1075. if (child.id === channelId) {
  1076. updateFn(child);
  1077. updated = true;
  1078. }
  1079. });
  1080. } else if (channel.id === channelId) {
  1081. // Direct channel match
  1082. updateFn(channel);
  1083. updated = true;
  1084. }
  1085. });
  1086. // Only update state if we actually modified a channel
  1087. if (updated) {
  1088. setChannels(newChannels);
  1089. }
  1090. };
  1091. const processTestQueue = async () => {
  1092. if (!isProcessingQueue || testQueue.length === 0) return;
  1093. const { channel, model, indexInFiltered } = testQueue[0];
  1094. // 自动翻页到正在测试的模型所在页
  1095. if (currentTestChannel && currentTestChannel.id === channel.id) {
  1096. let pageNo;
  1097. if (indexInFiltered !== undefined) {
  1098. pageNo = Math.floor(indexInFiltered / MODEL_TABLE_PAGE_SIZE) + 1;
  1099. } else {
  1100. const filteredModelsList = currentTestChannel.models
  1101. .split(',')
  1102. .filter((m) => m.toLowerCase().includes(modelSearchKeyword.toLowerCase()));
  1103. const modelIdx = filteredModelsList.indexOf(model);
  1104. pageNo = modelIdx !== -1 ? Math.floor(modelIdx / MODEL_TABLE_PAGE_SIZE) + 1 : 1;
  1105. }
  1106. setModelTablePage(pageNo);
  1107. }
  1108. try {
  1109. setTestingModels(prev => new Set([...prev, model]));
  1110. const res = await API.get(`/api/channel/test/${channel.id}?model=${model}`);
  1111. const { success, message, time } = res.data;
  1112. setModelTestResults(prev => ({
  1113. ...prev,
  1114. [`${channel.id}-${model}`]: { success, time }
  1115. }));
  1116. if (success) {
  1117. updateChannelProperty(channel.id, (ch) => {
  1118. ch.response_time = time * 1000;
  1119. ch.test_time = Date.now() / 1000;
  1120. });
  1121. if (!model) {
  1122. showInfo(
  1123. t('通道 ${name} 测试成功,耗时 ${time.toFixed(2)} 秒。')
  1124. .replace('${name}', channel.name)
  1125. .replace('${time.toFixed(2)}', time.toFixed(2)),
  1126. );
  1127. }
  1128. } else {
  1129. showError(message);
  1130. }
  1131. } catch (error) {
  1132. showError(error.message);
  1133. } finally {
  1134. setTestingModels(prev => {
  1135. const newSet = new Set(prev);
  1136. newSet.delete(model);
  1137. return newSet;
  1138. });
  1139. }
  1140. // 移除已处理的测试
  1141. setTestQueue(prev => prev.slice(1));
  1142. };
  1143. // 监听队列变化
  1144. useEffect(() => {
  1145. if (testQueue.length > 0 && isProcessingQueue) {
  1146. processTestQueue();
  1147. } else if (testQueue.length === 0 && isProcessingQueue) {
  1148. setIsProcessingQueue(false);
  1149. setIsBatchTesting(false);
  1150. }
  1151. }, [testQueue, isProcessingQueue]);
  1152. const testChannel = async (record, model) => {
  1153. setTestQueue(prev => [...prev, { channel: record, model }]);
  1154. if (!isProcessingQueue) {
  1155. setIsProcessingQueue(true);
  1156. }
  1157. };
  1158. const batchTestModels = async () => {
  1159. if (!currentTestChannel) return;
  1160. setIsBatchTesting(true);
  1161. // 重置分页到第一页
  1162. setModelTablePage(1);
  1163. const filteredModels = currentTestChannel.models
  1164. .split(',')
  1165. .filter((model) =>
  1166. model.toLowerCase().includes(modelSearchKeyword.toLowerCase()),
  1167. );
  1168. setTestQueue(
  1169. filteredModels.map((model, idx) => ({
  1170. channel: currentTestChannel,
  1171. model,
  1172. indexInFiltered: idx, // 记录在过滤列表中的顺序
  1173. })),
  1174. );
  1175. setIsProcessingQueue(true);
  1176. };
  1177. const handleCloseModal = () => {
  1178. if (isBatchTesting) {
  1179. // 清空测试队列来停止测试
  1180. setTestQueue([]);
  1181. setIsProcessingQueue(false);
  1182. setIsBatchTesting(false);
  1183. showSuccess(t('已停止测试'));
  1184. } else {
  1185. setShowModelTestModal(false);
  1186. setModelSearchKeyword('');
  1187. setSelectedModelKeys([]);
  1188. setModelTablePage(1);
  1189. }
  1190. };
  1191. const channelTypeCounts = useMemo(() => {
  1192. if (Object.keys(typeCounts).length > 0) return typeCounts;
  1193. // fallback 本地计算
  1194. const counts = { all: channels.length };
  1195. channels.forEach((channel) => {
  1196. const collect = (ch) => {
  1197. const type = ch.type;
  1198. counts[type] = (counts[type] || 0) + 1;
  1199. };
  1200. if (channel.children !== undefined) {
  1201. channel.children.forEach(collect);
  1202. } else {
  1203. collect(channel);
  1204. }
  1205. });
  1206. return counts;
  1207. }, [typeCounts, channels]);
  1208. const availableTypeKeys = useMemo(() => {
  1209. const keys = ['all'];
  1210. Object.entries(channelTypeCounts).forEach(([k, v]) => {
  1211. if (k !== 'all' && v > 0) keys.push(String(k));
  1212. });
  1213. return keys;
  1214. }, [channelTypeCounts]);
  1215. const renderTypeTabs = () => {
  1216. if (enableTagMode) return null;
  1217. return (
  1218. <Tabs
  1219. activeKey={activeTypeKey}
  1220. type="card"
  1221. collapsible
  1222. onChange={(key) => {
  1223. setActiveTypeKey(key);
  1224. setActivePage(1);
  1225. loadChannels(1, pageSize, idSort, enableTagMode, key);
  1226. }}
  1227. className="mb-4"
  1228. >
  1229. <TabPane
  1230. itemKey="all"
  1231. tab={
  1232. <span className="flex items-center gap-2">
  1233. {t('全部')}
  1234. <Tag color={activeTypeKey === 'all' ? 'red' : 'grey'} size='small' shape='circle'>
  1235. {channelTypeCounts['all'] || 0}
  1236. </Tag>
  1237. </span>
  1238. }
  1239. />
  1240. {CHANNEL_OPTIONS.filter((opt) => availableTypeKeys.includes(String(opt.value))).map((option) => {
  1241. const key = String(option.value);
  1242. const count = channelTypeCounts[option.value] || 0;
  1243. return (
  1244. <TabPane
  1245. key={key}
  1246. itemKey={key}
  1247. tab={
  1248. <span className="flex items-center gap-2">
  1249. {getChannelIcon(option.value)}
  1250. {option.label}
  1251. <Tag color={activeTypeKey === key ? 'red' : 'grey'} size='small' shape='circle'>
  1252. {count}
  1253. </Tag>
  1254. </span>
  1255. }
  1256. />
  1257. );
  1258. })}
  1259. </Tabs>
  1260. );
  1261. };
  1262. let pageData = channels;
  1263. const handlePageChange = (page) => {
  1264. const { searchKeyword, searchGroup, searchModel } = getFormValues();
  1265. setActivePage(page);
  1266. if (searchKeyword === '' && searchGroup === '' && searchModel === '') {
  1267. loadChannels(page, pageSize, idSort, enableTagMode).then(() => { });
  1268. } else {
  1269. searchChannels(enableTagMode, activeTypeKey, statusFilter, page, pageSize, idSort);
  1270. }
  1271. };
  1272. const handlePageSizeChange = async (size) => {
  1273. localStorage.setItem('page-size', size + '');
  1274. setPageSize(size);
  1275. setActivePage(1);
  1276. const { searchKeyword, searchGroup, searchModel } = getFormValues();
  1277. if (searchKeyword === '' && searchGroup === '' && searchModel === '') {
  1278. loadChannels(1, size, idSort, enableTagMode)
  1279. .then()
  1280. .catch((reason) => {
  1281. showError(reason);
  1282. });
  1283. } else {
  1284. searchChannels(enableTagMode, activeTypeKey, statusFilter, 1, size, idSort);
  1285. }
  1286. };
  1287. const fetchGroups = async () => {
  1288. try {
  1289. let res = await API.get(`/api/group/`);
  1290. if (res === undefined) {
  1291. return;
  1292. }
  1293. setGroupOptions(
  1294. res.data.data.map((group) => ({
  1295. label: group,
  1296. value: group,
  1297. })),
  1298. );
  1299. } catch (error) {
  1300. showError(error.message);
  1301. }
  1302. };
  1303. const submitTagEdit = async (type, data) => {
  1304. switch (type) {
  1305. case 'priority':
  1306. if (data.priority === undefined || data.priority === '') {
  1307. showInfo('优先级必须是整数!');
  1308. return;
  1309. }
  1310. data.priority = parseInt(data.priority);
  1311. break;
  1312. case 'weight':
  1313. if (
  1314. data.weight === undefined ||
  1315. data.weight < 0 ||
  1316. data.weight === ''
  1317. ) {
  1318. showInfo('权重必须是非负整数!');
  1319. return;
  1320. }
  1321. data.weight = parseInt(data.weight);
  1322. break;
  1323. }
  1324. try {
  1325. const res = await API.put('/api/channel/tag', data);
  1326. if (res?.data?.success) {
  1327. showSuccess('更新成功!');
  1328. await refresh();
  1329. }
  1330. } catch (error) {
  1331. showError(error);
  1332. }
  1333. };
  1334. const closeEdit = () => {
  1335. setShowEdit(false);
  1336. };
  1337. const handleRow = (record, index) => {
  1338. if (record.status !== 1) {
  1339. return {
  1340. style: {
  1341. background: 'var(--semi-color-disabled-border)',
  1342. },
  1343. };
  1344. } else {
  1345. return {};
  1346. }
  1347. };
  1348. const batchSetChannelTag = async () => {
  1349. if (selectedChannels.length === 0) {
  1350. showError(t('请先选择要设置标签的渠道!'));
  1351. return;
  1352. }
  1353. if (batchSetTagValue === '') {
  1354. showError(t('标签不能为空!'));
  1355. return;
  1356. }
  1357. let ids = selectedChannels.map((channel) => channel.id);
  1358. const res = await API.post('/api/channel/batch/tag', {
  1359. ids: ids,
  1360. tag: batchSetTagValue === '' ? null : batchSetTagValue,
  1361. });
  1362. if (res.data.success) {
  1363. showSuccess(
  1364. t('已为 ${count} 个渠道设置标签!').replace('${count}', res.data.data),
  1365. );
  1366. await refresh();
  1367. setShowBatchSetTag(false);
  1368. } else {
  1369. showError(res.data.message);
  1370. }
  1371. };
  1372. const testAllChannels = async () => {
  1373. const res = await API.get(`/api/channel/test`);
  1374. const { success, message } = res.data;
  1375. if (success) {
  1376. showInfo(t('已成功开始测试所有已启用通道,请刷新页面查看结果。'));
  1377. } else {
  1378. showError(message);
  1379. }
  1380. };
  1381. const deleteAllDisabledChannels = async () => {
  1382. const res = await API.delete(`/api/channel/disabled`);
  1383. const { success, message, data } = res.data;
  1384. if (success) {
  1385. showSuccess(
  1386. t('已删除所有禁用渠道,共计 ${data} 个').replace('${data}', data),
  1387. );
  1388. await refresh();
  1389. } else {
  1390. showError(message);
  1391. }
  1392. };
  1393. const updateAllChannelsBalance = async () => {
  1394. const res = await API.get(`/api/channel/update_balance`);
  1395. const { success, message } = res.data;
  1396. if (success) {
  1397. showInfo(t('已更新完毕所有已启用通道余额!'));
  1398. } else {
  1399. showError(message);
  1400. }
  1401. };
  1402. const updateChannelBalance = async (record) => {
  1403. const res = await API.get(`/api/channel/update_balance/${record.id}/`);
  1404. const { success, message, balance } = res.data;
  1405. if (success) {
  1406. updateChannelProperty(record.id, (channel) => {
  1407. channel.balance = balance;
  1408. channel.balance_updated_time = Date.now() / 1000;
  1409. });
  1410. showInfo(
  1411. t('通道 ${name} 余额更新成功!').replace('${name}', record.name),
  1412. );
  1413. } else {
  1414. showError(message);
  1415. }
  1416. };
  1417. const batchDeleteChannels = async () => {
  1418. if (selectedChannels.length === 0) {
  1419. showError(t('请先选择要删除的通道!'));
  1420. return;
  1421. }
  1422. setLoading(true);
  1423. let ids = [];
  1424. selectedChannels.forEach((channel) => {
  1425. ids.push(channel.id);
  1426. });
  1427. const res = await API.post(`/api/channel/batch`, { ids: ids });
  1428. const { success, message, data } = res.data;
  1429. if (success) {
  1430. showSuccess(t('已删除 ${data} 个通道!').replace('${data}', data));
  1431. await refresh();
  1432. } else {
  1433. showError(message);
  1434. }
  1435. setLoading(false);
  1436. };
  1437. const fixChannelsAbilities = async () => {
  1438. const res = await API.post(`/api/channel/fix`);
  1439. const { success, message, data } = res.data;
  1440. if (success) {
  1441. showSuccess(t('已修复 ${success} 个通道,失败 ${fails} 个通道。').replace('${success}', data.success).replace('${fails}', data.fails));
  1442. await refresh();
  1443. } else {
  1444. showError(message);
  1445. }
  1446. };
  1447. const renderHeader = () => (
  1448. <div className="flex flex-col w-full">
  1449. {renderTypeTabs()}
  1450. <div className="flex flex-col md:flex-row justify-between gap-4">
  1451. <div className="flex flex-wrap md:flex-nowrap items-center gap-2 w-full md:w-auto order-2 md:order-1">
  1452. <Button
  1453. size='small'
  1454. disabled={!enableBatchDelete}
  1455. theme='light'
  1456. type='danger'
  1457. className="w-full md:w-auto"
  1458. onClick={() => {
  1459. Modal.confirm({
  1460. title: t('确定是否要删除所选通道?'),
  1461. content: t('此修改将不可逆'),
  1462. onOk: () => batchDeleteChannels(),
  1463. });
  1464. }}
  1465. >
  1466. {t('删除所选通道')}
  1467. </Button>
  1468. <Button
  1469. size='small'
  1470. disabled={!enableBatchDelete}
  1471. theme='light'
  1472. type='primary'
  1473. onClick={() => setShowBatchSetTag(true)}
  1474. className="w-full md:w-auto"
  1475. >
  1476. {t('批量设置标签')}
  1477. </Button>
  1478. <Dropdown
  1479. size='small'
  1480. trigger='click'
  1481. render={
  1482. <Dropdown.Menu>
  1483. <Dropdown.Item>
  1484. <Button
  1485. size='small'
  1486. theme='light'
  1487. type='warning'
  1488. className="w-full"
  1489. onClick={() => {
  1490. Modal.confirm({
  1491. title: t('确定?'),
  1492. content: t('确定要测试所有通道吗?'),
  1493. onOk: () => testAllChannels(),
  1494. size: 'small',
  1495. centered: true,
  1496. });
  1497. }}
  1498. >
  1499. {t('测试所有通道')}
  1500. </Button>
  1501. </Dropdown.Item>
  1502. <Dropdown.Item>
  1503. <Button
  1504. size='small'
  1505. theme='light'
  1506. type='secondary'
  1507. className="w-full"
  1508. onClick={() => {
  1509. Modal.confirm({
  1510. title: t('确定?'),
  1511. content: t('确定要更新所有已启用通道余额吗?'),
  1512. onOk: () => updateAllChannelsBalance(),
  1513. size: 'sm',
  1514. centered: true,
  1515. });
  1516. }}
  1517. >
  1518. {t('更新所有已启用通道余额')}
  1519. </Button>
  1520. </Dropdown.Item>
  1521. <Dropdown.Item>
  1522. <Button
  1523. size='small'
  1524. theme='light'
  1525. type='danger'
  1526. className="w-full"
  1527. onClick={() => {
  1528. Modal.confirm({
  1529. title: t('确定是否要删除禁用通道?'),
  1530. content: t('此修改将不可逆'),
  1531. onOk: () => deleteAllDisabledChannels(),
  1532. size: 'sm',
  1533. centered: true,
  1534. });
  1535. }}
  1536. >
  1537. {t('删除禁用通道')}
  1538. </Button>
  1539. </Dropdown.Item>
  1540. <Dropdown.Item>
  1541. <Button
  1542. size='small'
  1543. theme='light'
  1544. type='tertiary'
  1545. className="w-full"
  1546. onClick={() => {
  1547. Modal.confirm({
  1548. title: t('确定是否要修复数据库一致性?'),
  1549. content: t('进行该操作时,可能导致渠道访问错误,请仅在数据库出现问题时使用'),
  1550. onOk: () => fixChannelsAbilities(),
  1551. size: 'sm',
  1552. centered: true,
  1553. });
  1554. }}
  1555. >
  1556. {t('修复数据库一致性')}
  1557. </Button>
  1558. </Dropdown.Item>
  1559. </Dropdown.Menu>
  1560. }
  1561. >
  1562. <Button size='small' theme='light' type='tertiary' className="w-full md:w-auto">
  1563. {t('批量操作')}
  1564. </Button>
  1565. </Dropdown>
  1566. <Button
  1567. size='small'
  1568. theme='light'
  1569. type='secondary'
  1570. className="w-full md:w-auto"
  1571. onClick={() => setCompactMode(!compactMode)}
  1572. >
  1573. {compactMode ? t('自适应列表') : t('紧凑列表')}
  1574. </Button>
  1575. </div>
  1576. <div className="flex flex-col md:flex-row items-start md:items-center gap-4 w-full md:w-auto order-1 md:order-2">
  1577. <div className="flex items-center justify-between w-full md:w-auto">
  1578. <Typography.Text strong className="mr-2">
  1579. {t('使用ID排序')}
  1580. </Typography.Text>
  1581. <Switch
  1582. size='small'
  1583. checked={idSort}
  1584. onChange={(v) => {
  1585. localStorage.setItem('id-sort', v + '');
  1586. setIdSort(v);
  1587. const { searchKeyword, searchGroup, searchModel } = getFormValues();
  1588. if (searchKeyword === '' && searchGroup === '' && searchModel === '') {
  1589. loadChannels(activePage, pageSize, v, enableTagMode);
  1590. } else {
  1591. searchChannels(enableTagMode, activeTypeKey, statusFilter, activePage, pageSize, v);
  1592. }
  1593. }}
  1594. />
  1595. </div>
  1596. <div className="flex items-center justify-between w-full md:w-auto">
  1597. <Typography.Text strong className="mr-2">
  1598. {t('开启批量操作')}
  1599. </Typography.Text>
  1600. <Switch
  1601. size='small'
  1602. checked={enableBatchDelete}
  1603. onChange={(v) => {
  1604. localStorage.setItem('enable-batch-delete', v + '');
  1605. setEnableBatchDelete(v);
  1606. }}
  1607. />
  1608. </div>
  1609. <div className="flex items-center justify-between w-full md:w-auto">
  1610. <Typography.Text strong className="mr-2">
  1611. {t('标签聚合模式')}
  1612. </Typography.Text>
  1613. <Switch
  1614. size='small'
  1615. checked={enableTagMode}
  1616. onChange={(v) => {
  1617. localStorage.setItem('enable-tag-mode', v + '');
  1618. setEnableTagMode(v);
  1619. setActivePage(1);
  1620. loadChannels(1, pageSize, idSort, v);
  1621. }}
  1622. />
  1623. </div>
  1624. {/* 状态筛选器 */}
  1625. <div className="flex items-center justify-between w-full md:w-auto">
  1626. <Typography.Text strong className="mr-2">
  1627. {t('状态筛选')}
  1628. </Typography.Text>
  1629. <Select
  1630. size='small'
  1631. value={statusFilter}
  1632. onChange={(v) => {
  1633. localStorage.setItem('channel-status-filter', v);
  1634. setStatusFilter(v);
  1635. setActivePage(1);
  1636. loadChannels(1, pageSize, idSort, enableTagMode, activeTypeKey, v);
  1637. }}
  1638. >
  1639. <Select.Option value="all">{t('全部')}</Select.Option>
  1640. <Select.Option value="enabled">{t('已启用')}</Select.Option>
  1641. <Select.Option value="disabled">{t('已禁用')}</Select.Option>
  1642. </Select>
  1643. </div>
  1644. </div>
  1645. </div>
  1646. <Divider margin="12px" />
  1647. <div className="flex flex-col md:flex-row justify-between items-center gap-4 w-full">
  1648. <div className="flex gap-2 w-full md:w-auto order-2 md:order-1">
  1649. <Button
  1650. size='small'
  1651. theme='light'
  1652. type='primary'
  1653. className="w-full md:w-auto"
  1654. onClick={() => {
  1655. setEditingChannel({
  1656. id: undefined,
  1657. });
  1658. setShowEdit(true);
  1659. }}
  1660. >
  1661. {t('添加渠道')}
  1662. </Button>
  1663. <Button
  1664. size='small'
  1665. theme='light'
  1666. type='primary'
  1667. className="w-full md:w-auto"
  1668. onClick={refresh}
  1669. >
  1670. {t('刷新')}
  1671. </Button>
  1672. <Button
  1673. size='small'
  1674. theme='light'
  1675. type='tertiary'
  1676. onClick={() => setShowColumnSelector(true)}
  1677. className="w-full md:w-auto"
  1678. >
  1679. {t('列设置')}
  1680. </Button>
  1681. </div>
  1682. <div className="flex flex-col md:flex-row items-center gap-4 w-full md:w-auto order-1 md:order-2">
  1683. <Form
  1684. initValues={formInitValues}
  1685. getFormApi={(api) => setFormApi(api)}
  1686. onSubmit={() => searchChannels(enableTagMode)}
  1687. allowEmpty={true}
  1688. autoComplete="off"
  1689. layout="horizontal"
  1690. trigger="change"
  1691. stopValidateWithError={false}
  1692. className="flex flex-col md:flex-row items-center gap-4 w-full"
  1693. >
  1694. <div className="relative w-full md:w-64">
  1695. <Form.Input
  1696. size='small'
  1697. field="searchKeyword"
  1698. prefix={<IconSearch />}
  1699. placeholder={t('渠道ID,名称,密钥,API地址')}
  1700. showClear
  1701. pure
  1702. />
  1703. </div>
  1704. <div className="w-full md:w-48">
  1705. <Form.Input
  1706. size='small'
  1707. field="searchModel"
  1708. prefix={<IconSearch />}
  1709. placeholder={t('模型关键字')}
  1710. showClear
  1711. pure
  1712. />
  1713. </div>
  1714. <div className="w-full md:w-32">
  1715. <Form.Select
  1716. size='small'
  1717. field="searchGroup"
  1718. placeholder={t('选择分组')}
  1719. optionList={[
  1720. { label: t('选择分组'), value: null },
  1721. ...groupOptions,
  1722. ]}
  1723. className="w-full"
  1724. showClear
  1725. pure
  1726. onChange={() => {
  1727. // 延迟执行搜索,让表单值先更新
  1728. setTimeout(() => {
  1729. searchChannels(enableTagMode);
  1730. }, 0);
  1731. }}
  1732. />
  1733. </div>
  1734. <Button
  1735. size='small'
  1736. type="primary"
  1737. htmlType="submit"
  1738. loading={loading || searching}
  1739. className="w-full md:w-auto"
  1740. >
  1741. {t('查询')}
  1742. </Button>
  1743. <Button
  1744. size='small'
  1745. theme='light'
  1746. onClick={() => {
  1747. if (formApi) {
  1748. formApi.reset();
  1749. // 重置后立即查询,使用setTimeout确保表单重置完成
  1750. setTimeout(() => {
  1751. refresh();
  1752. }, 100);
  1753. }
  1754. }}
  1755. className="w-full md:w-auto"
  1756. >
  1757. {t('重置')}
  1758. </Button>
  1759. </Form>
  1760. </div>
  1761. </div>
  1762. </div>
  1763. );
  1764. return (
  1765. <>
  1766. {renderColumnSelector()}
  1767. <EditTagModal
  1768. visible={showEditTag}
  1769. tag={editingTag}
  1770. handleClose={() => setShowEditTag(false)}
  1771. refresh={refresh}
  1772. />
  1773. <EditChannel
  1774. refresh={refresh}
  1775. visible={showEdit}
  1776. handleClose={closeEdit}
  1777. editingChannel={editingChannel}
  1778. />
  1779. <Card
  1780. className="!rounded-2xl"
  1781. title={renderHeader()}
  1782. shadows='always'
  1783. bordered={false}
  1784. >
  1785. <Table
  1786. columns={compactMode ? getVisibleColumns().map(({ fixed, ...rest }) => rest) : getVisibleColumns()}
  1787. dataSource={pageData}
  1788. scroll={compactMode ? undefined : { x: 'max-content' }}
  1789. pagination={{
  1790. currentPage: activePage,
  1791. pageSize: pageSize,
  1792. total: channelCount,
  1793. pageSizeOpts: [10, 20, 50, 100],
  1794. showSizeChanger: true,
  1795. formatPageText: (page) => t('第 {{start}} - {{end}} 条,共 {{total}} 条', {
  1796. start: page.currentStart,
  1797. end: page.currentEnd,
  1798. total: channelCount,
  1799. }),
  1800. onPageSizeChange: (size) => {
  1801. handlePageSizeChange(size);
  1802. },
  1803. onPageChange: handlePageChange,
  1804. }}
  1805. expandAllRows={false}
  1806. onRow={handleRow}
  1807. rowSelection={
  1808. enableBatchDelete
  1809. ? {
  1810. onChange: (selectedRowKeys, selectedRows) => {
  1811. setSelectedChannels(selectedRows);
  1812. },
  1813. }
  1814. : null
  1815. }
  1816. empty={
  1817. <Empty
  1818. image={<IllustrationNoResult style={{ width: 150, height: 150 }} />}
  1819. darkModeImage={<IllustrationNoResultDark style={{ width: 150, height: 150 }} />}
  1820. description={t('搜索无结果')}
  1821. style={{ padding: 30 }}
  1822. />
  1823. }
  1824. className="rounded-xl overflow-hidden"
  1825. size="middle"
  1826. loading={loading || searching}
  1827. />
  1828. </Card>
  1829. {/* 批量设置标签模态框 */}
  1830. <Modal
  1831. title={t('批量设置标签')}
  1832. visible={showBatchSetTag}
  1833. onOk={batchSetChannelTag}
  1834. onCancel={() => setShowBatchSetTag(false)}
  1835. maskClosable={false}
  1836. centered={true}
  1837. size="small"
  1838. className="!rounded-lg"
  1839. >
  1840. <div className="mb-5">
  1841. <Typography.Text>{t('请输入要设置的标签名称')}</Typography.Text>
  1842. </div>
  1843. <Input
  1844. placeholder={t('请输入标签名称')}
  1845. value={batchSetTagValue}
  1846. onChange={(v) => setBatchSetTagValue(v)}
  1847. size='large'
  1848. />
  1849. <div className="mt-4">
  1850. <Typography.Text type='secondary'>
  1851. {t('已选择 ${count} 个渠道').replace('${count}', selectedChannels.length)}
  1852. </Typography.Text>
  1853. </div>
  1854. </Modal>
  1855. {/* 模型测试弹窗 */}
  1856. <Modal
  1857. title={
  1858. currentTestChannel && (
  1859. <div className="flex flex-col gap-2 w-full">
  1860. <div className="flex items-center gap-2">
  1861. <Typography.Text strong className="!text-[var(--semi-color-text-0)] !text-base">
  1862. {currentTestChannel.name} {t('渠道的模型测试')}
  1863. </Typography.Text>
  1864. <Typography.Text type="tertiary" className="!text-xs flex items-center">
  1865. {t('共')} {currentTestChannel.models.split(',').length} {t('个模型')}
  1866. </Typography.Text>
  1867. </div>
  1868. {/* 搜索与操作按钮 */}
  1869. <div className="flex items-center justify-end gap-2 w-full">
  1870. <Input
  1871. placeholder={t('搜索模型...')}
  1872. value={modelSearchKeyword}
  1873. onChange={(v) => {
  1874. setModelSearchKeyword(v);
  1875. setModelTablePage(1);
  1876. }}
  1877. className="!w-full"
  1878. prefix={<IconSearch />}
  1879. showClear
  1880. />
  1881. <Button
  1882. theme='light'
  1883. onClick={() => {
  1884. if (selectedModelKeys.length === 0) {
  1885. showError(t('请先选择模型!'));
  1886. return;
  1887. }
  1888. copy(selectedModelKeys.join(',')).then((ok) => {
  1889. if (ok) {
  1890. showSuccess(t('已复制 ${count} 个模型').replace('${count}', selectedModelKeys.length));
  1891. } else {
  1892. showError(t('复制失败,请手动复制'));
  1893. }
  1894. });
  1895. }}
  1896. >
  1897. {t('复制已选')}
  1898. </Button>
  1899. <Button
  1900. theme='light'
  1901. type='primary'
  1902. onClick={() => {
  1903. if (!currentTestChannel) return;
  1904. const successKeys = currentTestChannel.models
  1905. .split(',')
  1906. .filter((m) => m.toLowerCase().includes(modelSearchKeyword.toLowerCase()))
  1907. .filter((m) => {
  1908. const result = modelTestResults[`${currentTestChannel.id}-${m}`];
  1909. return result && result.success;
  1910. });
  1911. if (successKeys.length === 0) {
  1912. showInfo(t('暂无成功模型'));
  1913. }
  1914. setSelectedModelKeys(successKeys);
  1915. }}
  1916. >
  1917. {t('选择成功')}
  1918. </Button>
  1919. </div>
  1920. </div>
  1921. )
  1922. }
  1923. visible={showModelTestModal && currentTestChannel !== null}
  1924. onCancel={handleCloseModal}
  1925. footer={
  1926. <div className="flex justify-end">
  1927. {isBatchTesting ? (
  1928. <Button
  1929. theme='light'
  1930. type='warning'
  1931. onClick={handleCloseModal}
  1932. >
  1933. {t('停止测试')}
  1934. </Button>
  1935. ) : (
  1936. <Button
  1937. theme='light'
  1938. type='tertiary'
  1939. onClick={handleCloseModal}
  1940. >
  1941. {t('取消')}
  1942. </Button>
  1943. )}
  1944. <Button
  1945. theme='light'
  1946. type='primary'
  1947. onClick={batchTestModels}
  1948. loading={isBatchTesting}
  1949. disabled={isBatchTesting}
  1950. >
  1951. {isBatchTesting ? t('测试中...') : t('批量测试${count}个模型').replace(
  1952. '${count}',
  1953. currentTestChannel
  1954. ? currentTestChannel.models
  1955. .split(',')
  1956. .filter((model) =>
  1957. model.toLowerCase().includes(modelSearchKeyword.toLowerCase())
  1958. ).length
  1959. : 0
  1960. )}
  1961. </Button>
  1962. </div>
  1963. }
  1964. maskClosable={!isBatchTesting}
  1965. className="!rounded-lg"
  1966. size={isMobile() ? 'full-width' : 'large'}
  1967. >
  1968. <div className="model-test-scroll">
  1969. {currentTestChannel && (
  1970. <div>
  1971. <Table
  1972. columns={[
  1973. {
  1974. title: t('模型名称'),
  1975. dataIndex: 'model',
  1976. render: (text) => (
  1977. <div className="flex items-center">
  1978. <Typography.Text strong>{text}</Typography.Text>
  1979. </div>
  1980. )
  1981. },
  1982. {
  1983. title: t('状态'),
  1984. dataIndex: 'status',
  1985. render: (text, record) => {
  1986. const testResult = modelTestResults[`${currentTestChannel.id}-${record.model}`];
  1987. const isTesting = testingModels.has(record.model);
  1988. if (isTesting) {
  1989. return (
  1990. <Tag size='large' color='blue' shape='circle'>
  1991. {t('测试中')}
  1992. </Tag>
  1993. );
  1994. }
  1995. if (!testResult) {
  1996. return (
  1997. <Tag size='large' color='grey' shape='circle'>
  1998. {t('未开始')}
  1999. </Tag>
  2000. );
  2001. }
  2002. return (
  2003. <div className="flex items-center gap-2">
  2004. <Tag
  2005. size='large'
  2006. color={testResult.success ? 'green' : 'red'}
  2007. shape='circle'
  2008. >
  2009. {testResult.success ? t('成功') : t('失败')}
  2010. </Tag>
  2011. {testResult.success && (
  2012. <Typography.Text type="tertiary">
  2013. {t('请求时长: ${time}s').replace('${time}', testResult.time.toFixed(2))}
  2014. </Typography.Text>
  2015. )}
  2016. </div>
  2017. );
  2018. }
  2019. },
  2020. {
  2021. title: '',
  2022. dataIndex: 'operate',
  2023. render: (text, record) => {
  2024. const isTesting = testingModels.has(record.model);
  2025. return (
  2026. <Button
  2027. theme='light'
  2028. type='primary'
  2029. onClick={() => testChannel(currentTestChannel, record.model)}
  2030. loading={isTesting}
  2031. size='small'
  2032. >
  2033. {t('测试')}
  2034. </Button>
  2035. );
  2036. }
  2037. }
  2038. ]}
  2039. dataSource={(() => {
  2040. const filtered = currentTestChannel.models
  2041. .split(',')
  2042. .filter((model) =>
  2043. model.toLowerCase().includes(modelSearchKeyword.toLowerCase()),
  2044. );
  2045. const start = (modelTablePage - 1) * MODEL_TABLE_PAGE_SIZE;
  2046. const end = start + MODEL_TABLE_PAGE_SIZE;
  2047. return filtered.slice(start, end).map((model) => ({
  2048. model,
  2049. key: model,
  2050. }));
  2051. })()}
  2052. rowSelection={{
  2053. selectedRowKeys: selectedModelKeys,
  2054. onChange: (keys) => {
  2055. if (allSelectingRef.current) {
  2056. allSelectingRef.current = false;
  2057. return;
  2058. }
  2059. setSelectedModelKeys(keys);
  2060. },
  2061. onSelectAll: (checked) => {
  2062. const filtered = currentTestChannel.models
  2063. .split(',')
  2064. .filter((m) => m.toLowerCase().includes(modelSearchKeyword.toLowerCase()));
  2065. allSelectingRef.current = true;
  2066. setSelectedModelKeys(checked ? filtered : []);
  2067. },
  2068. }}
  2069. pagination={{
  2070. currentPage: modelTablePage,
  2071. pageSize: MODEL_TABLE_PAGE_SIZE,
  2072. total: currentTestChannel.models
  2073. .split(',')
  2074. .filter((model) =>
  2075. model.toLowerCase().includes(modelSearchKeyword.toLowerCase()),
  2076. ).length,
  2077. showSizeChanger: false,
  2078. onPageChange: (page) => setModelTablePage(page),
  2079. }}
  2080. />
  2081. </div>
  2082. )}
  2083. </div>
  2084. </Modal>
  2085. </>
  2086. );
  2087. };
  2088. export default ChannelsTable;