Initial clean project import
This commit is contained in:
cat-shark
@@ -0,0 +1,554 @@
|
||||
import {net} from "electron";
|
||||
import {Client, handle_file} from "@gradio/client";
|
||||
import {platformArch, platformName, platformUUID} from "../../lib/env";
|
||||
import {Events} from "../event/main";
|
||||
import {Apps} from "../app";
|
||||
import {Files} from "../file/main";
|
||||
import fs from "node:fs";
|
||||
import User, {UserApi} from "../user/main";
|
||||
import {EncodeUtil, MemoryMapCacheUtil} from "../../lib/util";
|
||||
import {ServerContext, ServerFunctionDataType} from "./type";
|
||||
import {Log} from "../log/main";
|
||||
import {TaskLogMain} from "../taskLog/main";
|
||||
|
||||
type RequestOptionType = {
|
||||
method?: "POST" | "GET";
|
||||
timeout?: number;
|
||||
headers?: Record<string, string>;
|
||||
responseType?: "json" | "text";
|
||||
retry?: number;
|
||||
retryTimes?: number;
|
||||
retryInterval?: number;
|
||||
};
|
||||
|
||||
const request = async (url, data?: {}, option?: RequestOptionType) => {
|
||||
option = Object.assign(
|
||||
{
|
||||
method: "GET",
|
||||
timeout: 60 * 1000,
|
||||
headers: {
|
||||
"Content-Type": "application/json",
|
||||
},
|
||||
responseType: "json" as "json",
|
||||
retry: 0,
|
||||
retryTimes: 0,
|
||||
retryInterval: 5,
|
||||
},
|
||||
option
|
||||
);
|
||||
if (option["method"] === "GET") {
|
||||
url += "?";
|
||||
for (let key in data) {
|
||||
url += `${key}=${data[key]}&`;
|
||||
}
|
||||
}
|
||||
return new Promise((resolve, reject) => {
|
||||
const req = net.request({
|
||||
url,
|
||||
method: option["method"],
|
||||
headers: option["headers"],
|
||||
});
|
||||
req.on("response", response => {
|
||||
let body = "";
|
||||
response.on("data", chunk => {
|
||||
body += chunk.toString();
|
||||
});
|
||||
response.on("end", () => {
|
||||
if ("json" === option["responseType"]) {
|
||||
try {
|
||||
resolve(JSON.parse(body));
|
||||
} catch (e) {
|
||||
if (option.retry > 0 && option.retryTimes < option.retry) {
|
||||
option.retryTimes++;
|
||||
Log.info("request", `retry ${option.retryTimes} ${url}`);
|
||||
setTimeout(() => {
|
||||
request(url, data, option).then(resolve).catch(reject);
|
||||
}, option.retryInterval * 1000);
|
||||
} else {
|
||||
resolve({code: -1, msg: `ResponseError: ${body}`});
|
||||
}
|
||||
}
|
||||
} else {
|
||||
resolve(body);
|
||||
}
|
||||
});
|
||||
});
|
||||
req.on("error", err => {
|
||||
if (option.retry > 0 && option.retryTimes < option.retry) {
|
||||
option.retryTimes++;
|
||||
Log.info("request", `retry ${option.retryTimes} ${url}`);
|
||||
setTimeout(() => {
|
||||
request(url, data, option).then(resolve).catch(reject);
|
||||
}, option.retryInterval * 1000);
|
||||
} else {
|
||||
reject(err);
|
||||
}
|
||||
});
|
||||
if (option["method"] === "POST") {
|
||||
req.write(JSON.stringify(data));
|
||||
}
|
||||
req.end();
|
||||
});
|
||||
};
|
||||
|
||||
const requestPost = async (url, data?: {}, option?: RequestOptionType) => {
|
||||
option = Object.assign(
|
||||
{
|
||||
method: "POST",
|
||||
},
|
||||
option
|
||||
);
|
||||
return request(url, data, option);
|
||||
};
|
||||
|
||||
const requestGet = async (url, data?: {}, option?: RequestOptionType) => {
|
||||
option = Object.assign(
|
||||
{
|
||||
method: "GET",
|
||||
},
|
||||
option
|
||||
);
|
||||
return request(url, data, option);
|
||||
};
|
||||
|
||||
const requestPostSuccess = async (url, data?: {}, option?: RequestOptionType) => {
|
||||
const res = await requestPost(url, data, option);
|
||||
if (res["code"] === 0) {
|
||||
return res;
|
||||
}
|
||||
throw new Error(res["msg"]);
|
||||
};
|
||||
|
||||
const requestUrlFileToLocal = async (url, path) => {
|
||||
return new Promise((resolve, reject) => {
|
||||
const req = net.request(url);
|
||||
req.on("response", response => {
|
||||
const file = fs.createWriteStream(path);
|
||||
// @ts-ignore
|
||||
response.pipe(file);
|
||||
file.on("finish", () => {
|
||||
file.close();
|
||||
resolve("x");
|
||||
});
|
||||
});
|
||||
req.on("error", err => {
|
||||
reject(err);
|
||||
});
|
||||
req.end();
|
||||
});
|
||||
};
|
||||
|
||||
const requestEventSource = async (
|
||||
url: string,
|
||||
param: any,
|
||||
option?: {
|
||||
method?: "POST" | "GET";
|
||||
headers?: Record<string, string>;
|
||||
onMessage: (data: any) => void;
|
||||
onEnd?: () => void;
|
||||
}
|
||||
) => {
|
||||
option = Object.assign(
|
||||
{
|
||||
method: "POST",
|
||||
headers: {},
|
||||
onMessage: (data: any) => {
|
||||
console.log("onMessage", data);
|
||||
},
|
||||
onEnd: () => {
|
||||
console.log("onEnd");
|
||||
},
|
||||
},
|
||||
option
|
||||
);
|
||||
// return new Promise((resolve, reject) => {
|
||||
// const req = net.request({
|
||||
// url,
|
||||
// method: option.method,
|
||||
// headers: {
|
||||
// 'Content-Type': 'application/json',
|
||||
// ...option.headers,
|
||||
// },
|
||||
// })
|
||||
// req.on('response', (response) => {
|
||||
// console.log('response', response)
|
||||
// let buffer = ''
|
||||
// response.on('data', (chunk) => {
|
||||
// console.log('response.data', chunk)
|
||||
// buffer += chunk.toString()
|
||||
// const lines = buffer.split("\n")
|
||||
// buffer = lines.pop()
|
||||
// for (const line of lines) {
|
||||
// if (line.startsWith("data: ")) {
|
||||
// const data = line.slice(6)
|
||||
// if ('[END]' === data) {
|
||||
// option.onEnd()
|
||||
// return;
|
||||
// }
|
||||
// const eventData = line.slice(6).trim();
|
||||
// option.onMessage(EncodeUtil.base64Decode(eventData))
|
||||
// }
|
||||
// }
|
||||
// })
|
||||
// response.on('end', () => {
|
||||
// resolve(undefined)
|
||||
// })
|
||||
// })
|
||||
// req.on('error', (err) => {
|
||||
// reject(err)
|
||||
// })
|
||||
// req.end()
|
||||
// });
|
||||
const response = await net.fetch(url, {
|
||||
method: option.method,
|
||||
headers: {
|
||||
"Content-Type": "application/json",
|
||||
...option.headers,
|
||||
},
|
||||
body: JSON.stringify(param || {}),
|
||||
});
|
||||
if (!response.ok) {
|
||||
throw new Error(`HTTP error! Status: ${response.status}`);
|
||||
}
|
||||
const reader = response.body.getReader();
|
||||
const decoder = new TextDecoder();
|
||||
let buffer = "";
|
||||
while (true) {
|
||||
const {done, value} = await reader.read();
|
||||
if (done) {
|
||||
break;
|
||||
}
|
||||
buffer += decoder.decode(value, {stream: true});
|
||||
const lines = buffer.split("\n");
|
||||
buffer = lines.pop();
|
||||
// console.log('fetchEventSource', JSON.stringify(buffer))
|
||||
for (const line of lines) {
|
||||
// console.log('line', JSON.stringify(line))
|
||||
if (line.startsWith("data: ")) {
|
||||
const data = line.slice(6);
|
||||
if ("[END]" === data) {
|
||||
option.onEnd();
|
||||
return;
|
||||
}
|
||||
const eventData = line.slice(6).trim();
|
||||
option.onMessage(EncodeUtil.base64Decode(eventData));
|
||||
}
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
const env = async () => {
|
||||
const result = {};
|
||||
result["AIGCPANEL_SERVER_API_TOKEN"] = await User.getApiToken();
|
||||
result["AIGCPANEL_SERVER_UUID"] = platformUUID();
|
||||
result["AIGCPANEL_SERVER_LAUNCHER_MODE"] = "api";
|
||||
return result;
|
||||
};
|
||||
|
||||
const sleep = async ms => {
|
||||
return new Promise(resolve => {
|
||||
setTimeout(resolve, ms);
|
||||
});
|
||||
};
|
||||
|
||||
/**
|
||||
* 解析日志文本并保存到数据库
|
||||
* @param taskId 任务ID
|
||||
* @param logs 日志文本
|
||||
* @param context 服务器上下文
|
||||
*/
|
||||
const parseAndSaveLogsToDatabase = async (taskId: string, logs: string, context: ServerContext) => {
|
||||
const lines = logs.split("\n").filter(line => line.trim());
|
||||
|
||||
for (const line of lines) {
|
||||
try {
|
||||
// 解析日志行，格式通常为: 2024-11-24 16:31:08 - INFO - 标签 - 消息
|
||||
const parts = line.split(" - ");
|
||||
if (parts.length < 3) {
|
||||
// 如果格式不匹配，作为普通信息日志记录
|
||||
await TaskLogMain.addLog({
|
||||
taskId: taskId,
|
||||
level: "info",
|
||||
message: line,
|
||||
processName: context.ServerInfo?.name || "模型",
|
||||
});
|
||||
continue;
|
||||
}
|
||||
|
||||
// 提取时间戳、级别、标签和消息
|
||||
const timestamp = new Date(parts[0]).getTime();
|
||||
const level = (parts[1]?.toLowerCase() || "info") as "info" | "warn" | "error" | "debug";
|
||||
const label = parts[2] || "";
|
||||
const message = parts.slice(3).join(" - ");
|
||||
|
||||
// 检测是否包含进度信息
|
||||
let progress: number | undefined = undefined;
|
||||
const progressMatch = line.match(/(\d+)%/);
|
||||
if (progressMatch) {
|
||||
progress = parseInt(progressMatch[1]);
|
||||
}
|
||||
|
||||
// 判断日志级别
|
||||
let logLevel: "info" | "warn" | "error" | "debug" = "info";
|
||||
if (level === "error" || line.toLowerCase().includes("error") || line.toLowerCase().includes("失败")) {
|
||||
logLevel = "error";
|
||||
} else if (
|
||||
level === "warn" ||
|
||||
line.toLowerCase().includes("warning") ||
|
||||
line.toLowerCase().includes("警告") ||
|
||||
line.includes("FutureWarning") ||
|
||||
line.includes("DeprecationWarning") ||
|
||||
line.includes("UserWarning") ||
|
||||
line.includes("RuntimeWarning")
|
||||
) {
|
||||
logLevel = "warn";
|
||||
} else if (level === "debug") {
|
||||
logLevel = "debug";
|
||||
}
|
||||
|
||||
await TaskLogMain.addLog({
|
||||
taskId: taskId,
|
||||
level: logLevel,
|
||||
message: message || label,
|
||||
processName: context.ServerInfo?.name || "模型",
|
||||
progress: progress,
|
||||
timestamp: isNaN(timestamp) ? undefined : timestamp,
|
||||
});
|
||||
} catch (error) {
|
||||
// 解析失败时，将整行作为信息日志记录
|
||||
await TaskLogMain.addLog({
|
||||
taskId: taskId,
|
||||
level: "info",
|
||||
message: line,
|
||||
processName: context.ServerInfo?.name || "模型",
|
||||
});
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
const launcherCancel = async (context: ServerContext) => {
|
||||
const ret = (await requestPost(`${context.url()}cancel`, {})) as any;
|
||||
// console.log('cancel', JSON.stringify(ret))
|
||||
if (ret.code) {
|
||||
throw new Error(`cancel ${ret.msg}`);
|
||||
}
|
||||
};
|
||||
|
||||
const launcherSubmitAndQuery = async (
|
||||
context: ServerContext,
|
||||
data: ServerFunctionDataType,
|
||||
option?: {
|
||||
timeout: number;
|
||||
}
|
||||
): Promise<{
|
||||
result: {
|
||||
[key: string]: any;
|
||||
};
|
||||
endTime: number;
|
||||
}> => {
|
||||
option = Object.assign(
|
||||
{
|
||||
timeout: 24 * 3600,
|
||||
},
|
||||
option
|
||||
);
|
||||
|
||||
// 记录任务开始日志
|
||||
await TaskLogMain.addLog({
|
||||
taskId: data.id,
|
||||
level: "info",
|
||||
message: `任务开始执行`,
|
||||
processName: context.ServerInfo?.name || "模型",
|
||||
progress: 0,
|
||||
});
|
||||
|
||||
const submitRet = (await requestPost(`${context.url()}submit`, data)) as any;
|
||||
// console.log('submitRet', JSON.stringify(submitRet))
|
||||
if (submitRet.code) {
|
||||
await TaskLogMain.addLog({
|
||||
taskId: data.id,
|
||||
level: "error",
|
||||
message: `任务提交失败: ${submitRet.msg}`,
|
||||
processName: context.ServerInfo?.name || "模型",
|
||||
});
|
||||
throw new Error(`submit ${submitRet.msg}`);
|
||||
}
|
||||
const launcherResult = {
|
||||
result: {} as {
|
||||
[key: string]: any;
|
||||
},
|
||||
endTime: null,
|
||||
};
|
||||
const totalWait = Math.ceil(option.timeout / 5);
|
||||
for (let i = 0; i < totalWait; i++) {
|
||||
if (i >= totalWait - 1) {
|
||||
throw new Error("timeout");
|
||||
}
|
||||
await sleep(5000);
|
||||
const queryRet = (await requestPost(
|
||||
`${context.url()}query`,
|
||||
{
|
||||
token: submitRet.data.token,
|
||||
},
|
||||
{
|
||||
retry: 3,
|
||||
}
|
||||
)) as any;
|
||||
// console.log('queryRet', JSON.stringify(queryRet))
|
||||
if (queryRet.code) {
|
||||
throw new Error(queryRet.msg);
|
||||
}
|
||||
let logs = queryRet.data.logs;
|
||||
if (logs) {
|
||||
logs = EncodeUtil.base64Decode(logs);
|
||||
if (logs) {
|
||||
await Files.appendText(context.ServerInfo.logFile, logs);
|
||||
const result = extractResultFromLogs(data.id, logs);
|
||||
if (result) {
|
||||
launcherResult.result = Object.assign(launcherResult.result, result);
|
||||
context.send("taskResult", {id: data.id, result});
|
||||
}
|
||||
|
||||
// 将日志写入到任务日志数据库
|
||||
await parseAndSaveLogsToDatabase(data.id, logs, context);
|
||||
}
|
||||
}
|
||||
if (queryRet.data.status === "running") {
|
||||
continue;
|
||||
} else if (queryRet.data.status === "success") {
|
||||
launcherResult.endTime = Date.now();
|
||||
await Files.appendText(context.ServerInfo.logFile, "success");
|
||||
|
||||
// 记录任务成功日志
|
||||
await TaskLogMain.addLog({
|
||||
taskId: data.id,
|
||||
level: "info",
|
||||
message: "任务执行成功",
|
||||
processName: context.ServerInfo?.name || "模型",
|
||||
progress: 100,
|
||||
});
|
||||
} else {
|
||||
let msg = "请在模型日志中查看错误日志";
|
||||
if (launcherResult.result && launcherResult.result.msg) {
|
||||
msg = launcherResult.result.msg;
|
||||
}
|
||||
|
||||
// 记录任务失败日志
|
||||
await TaskLogMain.addLog({
|
||||
taskId: data.id,
|
||||
level: "error",
|
||||
message: `任务执行失败: ${msg}`,
|
||||
processName: context.ServerInfo?.name || "模型",
|
||||
});
|
||||
|
||||
throw `运行错误：${msg}`;
|
||||
}
|
||||
break;
|
||||
}
|
||||
return launcherResult;
|
||||
};
|
||||
|
||||
|
||||
const launcherPrepareConfigJson = async (data: any) => {
|
||||
const configJson = await Files.temp("json");
|
||||
await Files.write(configJson, JSON.stringify(data), {isDataPath: false});
|
||||
return configJson;
|
||||
};
|
||||
|
||||
const launcherSubmitConfigJsonAndQuery = async (context: ServerContext, configData: any) => {
|
||||
if (!("setting" in configData)) {
|
||||
configData.setting = context.ServerInfo.setting;
|
||||
}
|
||||
const configJsonPath = await launcherPrepareConfigJson(configData);
|
||||
// 使用传统的轮询方式（暂时保持原样）
|
||||
const result = await launcherSubmitAndQuery(context, {
|
||||
id: configData.id,
|
||||
result: {},
|
||||
entryPlaceholders: {
|
||||
CONFIG: configJsonPath,
|
||||
},
|
||||
root: context.ServerInfo.localPath,
|
||||
});
|
||||
await Files.deletes(configJsonPath, {isDataPath: false});
|
||||
return result;
|
||||
};
|
||||
|
||||
const extractResultFromLogs = (dataId: string, logs: string) => {
|
||||
let result = null;
|
||||
logs.split("\n").forEach(line => {
|
||||
// 🔧 修复：兼容 AigcPanelRunResult 和 ZhenQianBaRunResult 两种格式
|
||||
// 因为模型服务器可能还在使用旧的 AigcPanelRunResult 前缀
|
||||
const match = line.match(new RegExp(`(?:ZhenQianBaRunResult|AigcPanelRunResult)\\[${dataId}\\]\\[(.*?)\\]`));
|
||||
// console.log('match', {_data, match})
|
||||
if (match) {
|
||||
const matchResult = JSON.parse(EncodeUtil.base64Decode(match[1]));
|
||||
result = Object.assign(result || {}, matchResult);
|
||||
}
|
||||
});
|
||||
return result;
|
||||
};
|
||||
|
||||
const availablePort = async (
|
||||
port: number,
|
||||
setting: {
|
||||
port?: number;
|
||||
}
|
||||
) => {
|
||||
setting = setting || {};
|
||||
if (port) {
|
||||
return port;
|
||||
}
|
||||
if (setting["port"]) {
|
||||
port = parseInt(setting["port"] as any);
|
||||
} else if (!port || !(await Apps.isPortAvailable(port))) {
|
||||
port = await Apps.availablePort(50617);
|
||||
}
|
||||
return port;
|
||||
};
|
||||
|
||||
const pathSep = () => {
|
||||
return process.platform === "win32" ? ";" : ":";
|
||||
};
|
||||
|
||||
const getPathEnv = (addition: string | string[] = null) => {
|
||||
let p = process.env["PATH"] || "";
|
||||
if (addition) {
|
||||
const sep = pathSep();
|
||||
if (typeof addition === "string") {
|
||||
addition = [addition];
|
||||
}
|
||||
for (let path of addition) {
|
||||
if (p.indexOf(path) === -1) {
|
||||
p = `${path}${sep}${p}`;
|
||||
}
|
||||
}
|
||||
}
|
||||
return p;
|
||||
};
|
||||
|
||||
export default {
|
||||
GradioClient: Client,
|
||||
GradioHandleFile: handle_file,
|
||||
event: Events,
|
||||
file: Files,
|
||||
app: Apps,
|
||||
request,
|
||||
requestPost,
|
||||
requestGet,
|
||||
requestPostSuccess,
|
||||
requestUrlFileToLocal,
|
||||
requestEventSource,
|
||||
platformName: platformName(),
|
||||
platformArch: platformArch(),
|
||||
env,
|
||||
sleep,
|
||||
base64Encode: EncodeUtil.base64Encode,
|
||||
base64Decode: EncodeUtil.base64Decode,
|
||||
launcherCancel,
|
||||
launcherSubmitAndQuery,
|
||||
launcherPrepareConfigJson,
|
||||
launcherSubmitConfigJsonAndQuery,
|
||||
extractResultFromLogs,
|
||||
availablePort,
|
||||
getPathEnv,
|
||||
};
|
||||
@@ -0,0 +1,27 @@
|
||||
export const mapError = e => {
|
||||
if (e === undefined || e === null) {
|
||||
return "Unknown error";
|
||||
}
|
||||
let msg = e;
|
||||
if (e instanceof Error) {
|
||||
msg = [e.message, e.stack].join("\n");
|
||||
} else if (typeof e !== "string") {
|
||||
msg = e.toString();
|
||||
}
|
||||
const map = {
|
||||
// 'fetch error': '网络错误',
|
||||
};
|
||||
for (let key in map) {
|
||||
if (msg.includes(key)) {
|
||||
let error = map[key];
|
||||
// regex PluginReleaseDocFormatError:-11
|
||||
const regex = new RegExp(`${key}:(-?\\d+)`);
|
||||
const match = msg.match(regex);
|
||||
if (match) {
|
||||
error += `(${match[1]})`;
|
||||
}
|
||||
return error;
|
||||
}
|
||||
}
|
||||
return msg;
|
||||
};
|
||||
@@ -0,0 +1,295 @@
|
||||
import ServerApi from "./api";
|
||||
import {ipcMain} from "electron";
|
||||
import {Log} from "../log/main";
|
||||
import {mapError} from "./error";
|
||||
import {AigcServer} from "../../aigcserver";
|
||||
import {SendType, ServerContext, ServerInfo} from "./type";
|
||||
import {Files} from "../file/main";
|
||||
import {getGpuInfo} from "../../lib/env-main";
|
||||
|
||||
ipcMain.handle("server:listGpus", async event => {
|
||||
return await getGpuInfo();
|
||||
});
|
||||
|
||||
let runningServerCount = 0;
|
||||
ipcMain.handle("server:runningServerCount", async (event, count: number | null) => {
|
||||
if (count === null) {
|
||||
return runningServerCount;
|
||||
}
|
||||
// console.log('runningServerCount', count)
|
||||
runningServerCount = count;
|
||||
return runningServerCount;
|
||||
});
|
||||
const getRunningServerCount = () => {
|
||||
return runningServerCount;
|
||||
};
|
||||
|
||||
const serverModule: {
|
||||
[key: string]: ServerContext;
|
||||
} = {};
|
||||
|
||||
// 清理单个模块
|
||||
const cleanupModule = async (localPath: string) => {
|
||||
if (serverModule[localPath]) {
|
||||
try {
|
||||
const server = serverModule[localPath];
|
||||
if (server && server.stop) {
|
||||
await server.stop();
|
||||
}
|
||||
} catch (e) {
|
||||
Log.warn(`[cleanupModule] 停止模块失败: ${localPath}`, e);
|
||||
}
|
||||
delete serverModule[localPath];
|
||||
Log.info(`[cleanupModule] ✓ 已清理模块: ${localPath}`);
|
||||
}
|
||||
};
|
||||
|
||||
// 清理所有模块缓存
|
||||
const clearAllModules = async () => {
|
||||
const paths = Object.keys(serverModule);
|
||||
Log.info(`[clearAllModules] 开始清理 ${paths.length} 个模块缓存`);
|
||||
for (const path of paths) {
|
||||
delete serverModule[path];
|
||||
}
|
||||
Log.info(`[clearAllModules] ✓ 已清理所有模块缓存`);
|
||||
};
|
||||
|
||||
const init = () => {
|
||||
};
|
||||
|
||||
const getModule = async (
|
||||
serverInfo: ServerInfo,
|
||||
option?: {
|
||||
throwException: boolean;
|
||||
}
|
||||
): Promise<ServerContext> => {
|
||||
option = Object.assign(
|
||||
{
|
||||
throwException: true,
|
||||
},
|
||||
option
|
||||
);
|
||||
// console.log('getModule', serverInfo)
|
||||
if (!serverModule[serverInfo.localPath]) {
|
||||
try {
|
||||
if (serverInfo.name.startsWith("Cloud")) {
|
||||
const server = new AigcServer["Cloud"]();
|
||||
server.type = "buildIn";
|
||||
server.ServerApi = ServerApi;
|
||||
await server.init();
|
||||
serverModule[serverInfo.localPath] = server;
|
||||
} else if (serverInfo.name in AigcServer) {
|
||||
const server = AigcServer[serverInfo.name] as ServerContext;
|
||||
server.type = "buildIn";
|
||||
server.ServerApi = ServerApi;
|
||||
await server.init();
|
||||
serverModule[serverInfo.localPath] = server;
|
||||
} else {
|
||||
const serverPath = `${serverInfo.localPath}/server.js`;
|
||||
const configPath = `${serverInfo.localPath}/config.json`;
|
||||
|
||||
let server = null;
|
||||
if (
|
||||
await Files.exists(serverPath, {
|
||||
isDataPath: false,
|
||||
})
|
||||
) {
|
||||
const module = await import(`file://${serverPath}`);
|
||||
server = module.default;
|
||||
}
|
||||
if (
|
||||
!server &&
|
||||
(await Files.exists(configPath, {
|
||||
isDataPath: false,
|
||||
}))
|
||||
) {
|
||||
const configContent = await Files.read(configPath, {isDataPath: false});
|
||||
try {
|
||||
const config = JSON.parse(configContent);
|
||||
if (config.entry === "__EasyServer__") {
|
||||
server = new AigcServer["EasyServer"](config);
|
||||
} else {
|
||||
throw `ServerEntryNotFound : ${config.entry}`;
|
||||
}
|
||||
} catch (e) {
|
||||
const errorMessage = e instanceof Error ? e.message : String(e);
|
||||
console.error(`Failed to parse config file: ${configPath}`, {
|
||||
error: e,
|
||||
content: configContent.substring(0, 500) + (configContent.length > 500 ? "..." : ""),
|
||||
contentLength: configContent.length
|
||||
});
|
||||
throw `ConfigParseError : ${configPath} - ${errorMessage}`;
|
||||
}
|
||||
}
|
||||
if (!server) {
|
||||
throw `ServerFileNotFound : ${serverPath}`;
|
||||
}
|
||||
server.type = "custom";
|
||||
server.ServerApi = ServerApi;
|
||||
if (server.init) {
|
||||
await server.init();
|
||||
}
|
||||
server.send = (type: SendType, data: any) => {
|
||||
server.ServerApi.event.sendChannel(server.ServerInfo.eventChannelName, {type, data});
|
||||
};
|
||||
server.sendLog = (data: any) => {
|
||||
server.ServerApi.file.appendText(server.ServerInfo.logFile, data, {isDataPath: true});
|
||||
};
|
||||
serverModule[serverInfo.localPath] = server;
|
||||
}
|
||||
} catch (e) {
|
||||
if (!option.throwException) {
|
||||
return null;
|
||||
}
|
||||
const error = mapError(e);
|
||||
Log.error("mapi.server.getModule.error", error);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
// console.log('getModule', serverInfo, serverModule[serverInfo.localPath])
|
||||
serverModule[serverInfo.localPath].ServerInfo = serverInfo;
|
||||
return serverModule[serverInfo.localPath];
|
||||
};
|
||||
|
||||
ipcMain.handle("server:isSupport", async (event, serverInfo: ServerInfo) => {
|
||||
try {
|
||||
const module = await getModule(serverInfo, {
|
||||
throwException: false,
|
||||
});
|
||||
return !!module;
|
||||
} catch (e) {
|
||||
return false;
|
||||
}
|
||||
});
|
||||
|
||||
ipcMain.handle("server:start", async (event, serverInfo: ServerInfo) => {
|
||||
const module = await getModule(serverInfo);
|
||||
try {
|
||||
return await module.start();
|
||||
} catch (e) {
|
||||
const error = mapError(e);
|
||||
Log.error("mapi.server.start.error", error);
|
||||
throw error;
|
||||
}
|
||||
});
|
||||
|
||||
ipcMain.handle("server:ping", async (event, serverInfo: ServerInfo) => {
|
||||
const module = await getModule(serverInfo);
|
||||
try {
|
||||
return await module.ping();
|
||||
} catch (e) {
|
||||
const error = mapError(e);
|
||||
Log.error("mapi.server.ping.error", error);
|
||||
throw error;
|
||||
}
|
||||
return false;
|
||||
});
|
||||
|
||||
ipcMain.handle("server:stop", async (event, serverInfo: ServerInfo) => {
|
||||
const module = await getModule(serverInfo);
|
||||
try {
|
||||
const result = await module.stop();
|
||||
|
||||
// ✅ 关键修复：停止后立即删除缓存，释放内存
|
||||
if (serverModule[serverInfo.localPath]) {
|
||||
delete serverModule[serverInfo.localPath];
|
||||
Log.info(`[server:stop] ✓ 已释放模块缓存: ${serverInfo.name}`);
|
||||
}
|
||||
|
||||
return result;
|
||||
} catch (e) {
|
||||
const error = mapError(e);
|
||||
Log.error("mapi.server.stop.error", error);
|
||||
throw error;
|
||||
}
|
||||
});
|
||||
|
||||
ipcMain.handle("server:cancel", async (event, serverInfo: ServerInfo) => {
|
||||
const module = await getModule(serverInfo);
|
||||
try {
|
||||
return await module.cancel();
|
||||
} catch (e) {
|
||||
const error = mapError(e);
|
||||
Log.error("mapi.server.cancel.error", error);
|
||||
throw error;
|
||||
}
|
||||
});
|
||||
|
||||
ipcMain.handle("server:deletes", async (event, serverInfo: ServerInfo) => {
|
||||
if (serverModule[serverInfo.localPath]) {
|
||||
delete serverModule[serverInfo.localPath];
|
||||
}
|
||||
return true;
|
||||
});
|
||||
|
||||
ipcMain.handle("server:config", async (event, serverInfo: ServerInfo) => {
|
||||
const module = await getModule(serverInfo);
|
||||
try {
|
||||
return await module.config();
|
||||
} catch (e) {
|
||||
const error = mapError(e);
|
||||
Log.error("mapi.server.config.error", error);
|
||||
throw error;
|
||||
}
|
||||
});
|
||||
|
||||
ipcMain.handle("server:callFunction", async (event, serverInfo: ServerInfo, method: string, data: any, option: any) => {
|
||||
// console.log('getModule.before', serverInfo, method)
|
||||
const module = await getModule(serverInfo);
|
||||
// console.log('getModule.end', serverInfo, method, module)
|
||||
const func = module[method];
|
||||
if (!func) {
|
||||
throw new Error(`MethodNotFound : ${method}`);
|
||||
}
|
||||
try {
|
||||
return await func.bind(module)(data, option || {});
|
||||
} catch (e) {
|
||||
const error = mapError(e);
|
||||
Log.error("mapi.server.callFunction.error", {
|
||||
type: typeof e,
|
||||
error,
|
||||
serverInfo,
|
||||
method,
|
||||
data,
|
||||
option,
|
||||
});
|
||||
return {
|
||||
code: -1,
|
||||
msg: error,
|
||||
};
|
||||
}
|
||||
});
|
||||
|
||||
export default {
|
||||
init,
|
||||
};
|
||||
|
||||
// 关闭所有运行中的服务器
|
||||
const stopAllServers = async () => {
|
||||
const servers = Object.entries(serverModule);
|
||||
Log.info(`[stopAllServers] 开始停止 ${servers.length} 个服务器`);
|
||||
|
||||
for (const [localPath, server] of servers) {
|
||||
try {
|
||||
if (server && server.stop) {
|
||||
Log.info(`[stopAllServers] 正在停止: ${server.ServerInfo?.name || localPath}`);
|
||||
await server.stop();
|
||||
}
|
||||
} catch (e) {
|
||||
const error = mapError(e);
|
||||
Log.error(`[stopAllServers] 停止失败: ${server.ServerInfo?.name || localPath}`, error);
|
||||
}
|
||||
// 无论成功失败，都从缓存中删除
|
||||
delete serverModule[localPath];
|
||||
Log.info(`[stopAllServers] ✓ 已释放模块缓存: ${server.ServerInfo?.name || localPath}`);
|
||||
}
|
||||
|
||||
Log.info(`[stopAllServers] ✓ 所有服务器已停止，模块缓存已清空`);
|
||||
};
|
||||
|
||||
export const ServerMain = {
|
||||
getRunningServerCount,
|
||||
stopAllServers,
|
||||
clearAllModules,
|
||||
getModule, // 导出给其他模块使用
|
||||
};
|
||||
@@ -0,0 +1,67 @@
|
||||
import {ServerInfo} from "./type";
|
||||
|
||||
const listGpus = async () => {
|
||||
const ipcRenderer = (window as any).ipcRenderer;
|
||||
return ipcRenderer.invoke("server:listGpus");
|
||||
};
|
||||
|
||||
const runningServerCount = async (count: number | null) => {
|
||||
return ipcRenderer.invoke("server:runningServerCount", count);
|
||||
};
|
||||
|
||||
const isSupport = async (serverInfo: ServerInfo) => {
|
||||
return ipcRenderer.invoke("server:isSupport", serverInfo);
|
||||
};
|
||||
|
||||
const start = async (serverInfo: ServerInfo) => {
|
||||
return ipcRenderer.invoke("server:start", serverInfo);
|
||||
};
|
||||
|
||||
const ping = async (serverInfo: ServerInfo) => {
|
||||
return ipcRenderer.invoke("server:ping", serverInfo);
|
||||
};
|
||||
|
||||
const stop = async (serverInfo: ServerInfo) => {
|
||||
return ipcRenderer.invoke("server:stop", serverInfo);
|
||||
};
|
||||
|
||||
const cancel = async (serverInfo: ServerInfo) => {
|
||||
return ipcRenderer.invoke("server:cancel", serverInfo);
|
||||
};
|
||||
|
||||
const deletes = async (serverInfo: ServerInfo) => {
|
||||
return ipcRenderer.invoke("server:deletes", serverInfo);
|
||||
}
|
||||
|
||||
const config = async (serverInfo: ServerInfo) => {
|
||||
return ipcRenderer.invoke("server:config", serverInfo);
|
||||
};
|
||||
|
||||
const callFunction = async (serverInfo: ServerInfo, method: string, data: any, option: any) => {
|
||||
return ipcRenderer.invoke("server:callFunction", serverInfo, method, data, option);
|
||||
};
|
||||
|
||||
const callFunctionWithException = async (serverInfo: ServerInfo, method: string, data: any, option: any) => {
|
||||
try {
|
||||
return ipcRenderer.invoke("server:callFunction", serverInfo, method, data, option);
|
||||
} catch (e) {
|
||||
return {
|
||||
code: -1,
|
||||
msg: e + "",
|
||||
};
|
||||
}
|
||||
};
|
||||
|
||||
export default {
|
||||
listGpus,
|
||||
runningServerCount,
|
||||
isSupport,
|
||||
start,
|
||||
ping,
|
||||
stop,
|
||||
cancel,
|
||||
deletes,
|
||||
config,
|
||||
callFunction,
|
||||
callFunctionWithException,
|
||||
};
|
||||
@@ -0,0 +1,59 @@
|
||||
import api from "./api";
|
||||
|
||||
export type ServerApiType = typeof api;
|
||||
|
||||
export type SendType =
|
||||
| never
|
||||
// 服务
|
||||
| "starting"
|
||||
| "stopping"
|
||||
| "stopped"
|
||||
| "success"
|
||||
| "error"
|
||||
// 其他
|
||||
| "action"
|
||||
// 任务
|
||||
| "taskRunning"
|
||||
| "taskResult"
|
||||
| "taskStatus";
|
||||
|
||||
export type ServerInfo = {
|
||||
localPath: string;
|
||||
name: string;
|
||||
version: string;
|
||||
setting: {
|
||||
[key: string]: any;
|
||||
};
|
||||
logFile: string;
|
||||
eventChannelName: string;
|
||||
config: any;
|
||||
};
|
||||
|
||||
export type ServerContext = {
|
||||
ServerApi: ServerApiType | null;
|
||||
ServerInfo: ServerInfo | null;
|
||||
|
||||
send: (type: SendType, data: any) => void;
|
||||
|
||||
init: () => Promise<void>;
|
||||
start: () => Promise<void>;
|
||||
cancel: () => Promise<void>;
|
||||
stop: () => Promise<void>;
|
||||
url: () => string;
|
||||
ping: () => Promise<boolean>;
|
||||
config: () => Promise<any>;
|
||||
|
||||
[key: string]: any;
|
||||
};
|
||||
|
||||
export type ServerFunctionDataType = {
|
||||
id: string;
|
||||
result: {
|
||||
[key: string]: any;
|
||||
};
|
||||
[key: string]: any;
|
||||
};
|
||||
|
||||
export type ServerFunctionOptionType = {
|
||||
[key: string]: any;
|
||||
};
|
||||
Reference in New Issue
Block a user