refactor: Optimize channel testing and model menu generation (fix #761)
This commit is contained in:
@@ -357,6 +357,13 @@ const ChannelsTable = () => {
|
|||||||
dataIndex: 'operate',
|
dataIndex: 'operate',
|
||||||
render: (text, record, index) => {
|
render: (text, record, index) => {
|
||||||
if (record.children === undefined) {
|
if (record.children === undefined) {
|
||||||
|
// 构建模型测试菜单
|
||||||
|
const modelMenuItems = record.models.split(',').map(model => ({
|
||||||
|
node: 'item',
|
||||||
|
name: model,
|
||||||
|
onClick: () => testChannel(record, model)
|
||||||
|
}));
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div>
|
<div>
|
||||||
<SplitButtonGroup
|
<SplitButtonGroup
|
||||||
@@ -374,7 +381,7 @@ const ChannelsTable = () => {
|
|||||||
<Dropdown
|
<Dropdown
|
||||||
trigger="click"
|
trigger="click"
|
||||||
position="bottomRight"
|
position="bottomRight"
|
||||||
menu={record.test_models}
|
menu={modelMenuItems} // 使用即时生成的菜单项
|
||||||
>
|
>
|
||||||
<Button
|
<Button
|
||||||
style={{ padding: '8px 4px' }}
|
style={{ padding: '8px 4px' }}
|
||||||
@@ -545,17 +552,6 @@ const ChannelsTable = () => {
|
|||||||
let channelTags = {};
|
let channelTags = {};
|
||||||
for (let i = 0; i < channels.length; i++) {
|
for (let i = 0; i < channels.length; i++) {
|
||||||
channels[i].key = '' + channels[i].id;
|
channels[i].key = '' + channels[i].id;
|
||||||
let test_models = [];
|
|
||||||
channels[i].models.split(',').forEach((item, index) => {
|
|
||||||
test_models.push({
|
|
||||||
node: 'item',
|
|
||||||
name: item,
|
|
||||||
onClick: () => {
|
|
||||||
testChannel(channels[i], item);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
});
|
|
||||||
channels[i].test_models = test_models;
|
|
||||||
if (!enableTagMode) {
|
if (!enableTagMode) {
|
||||||
channelDates.push(channels[i]);
|
channelDates.push(channels[i]);
|
||||||
} else {
|
} else {
|
||||||
@@ -801,6 +797,7 @@ const ChannelsTable = () => {
|
|||||||
const updateChannelProperty = (channelId, updateFn) => {
|
const updateChannelProperty = (channelId, updateFn) => {
|
||||||
// Create a new copy of channels array
|
// Create a new copy of channels array
|
||||||
const newChannels = [...channels];
|
const newChannels = [...channels];
|
||||||
|
let updated = false;
|
||||||
|
|
||||||
// Find and update the correct channel
|
// Find and update the correct channel
|
||||||
newChannels.forEach(channel => {
|
newChannels.forEach(channel => {
|
||||||
@@ -809,26 +806,32 @@ const ChannelsTable = () => {
|
|||||||
channel.children.forEach(child => {
|
channel.children.forEach(child => {
|
||||||
if (child.id === channelId) {
|
if (child.id === channelId) {
|
||||||
updateFn(child);
|
updateFn(child);
|
||||||
|
updated = true;
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
} else if (channel.id === channelId) {
|
} else if (channel.id === channelId) {
|
||||||
// Direct channel match
|
// Direct channel match
|
||||||
updateFn(channel);
|
updateFn(channel);
|
||||||
|
updated = true;
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
// Update state with new array to trigger re-render
|
// Only update state if we actually modified a channel
|
||||||
setChannels(newChannels);
|
if (updated) {
|
||||||
|
setChannels(newChannels);
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
const testChannel = async (record, model) => {
|
const testChannel = async (record, model) => {
|
||||||
const res = await API.get(`/api/channel/test/${record.id}?model=${model}`);
|
const res = await API.get(`/api/channel/test/${record.id}?model=${model}`);
|
||||||
const { success, message, time } = res.data;
|
const { success, message, time } = res.data;
|
||||||
if (success) {
|
if (success) {
|
||||||
|
// Also update the channels state to persist the change
|
||||||
updateChannelProperty(record.id, (channel) => {
|
updateChannelProperty(record.id, (channel) => {
|
||||||
channel.response_time = time * 1000;
|
channel.response_time = time * 1000;
|
||||||
channel.test_time = Date.now() / 1000;
|
channel.test_time = Date.now() / 1000;
|
||||||
});
|
});
|
||||||
|
|
||||||
showInfo(t('通道 ${name} 测试成功,耗时 ${time.toFixed(2)} 秒。').replace('${name}', record.name).replace('${time.toFixed(2)}', time.toFixed(2)));
|
showInfo(t('通道 ${name} 测试成功,耗时 ${time.toFixed(2)} 秒。').replace('${name}', record.name).replace('${time.toFixed(2)}', time.toFixed(2)));
|
||||||
} else {
|
} else {
|
||||||
showError(message);
|
showError(message);
|
||||||
|
|||||||
Reference in New Issue
Block a user