ChannelsTable.js 68 KB

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