ChatGPT/FastGPT
1
0
Fork 0
mirror of https://github.com/labring/FastGPT.git synced 2025-07-23 13:03:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Perf: vector queue and app ui (#1750)

Browse Source
This commit is contained in:
Archer
2024-06-13 12:35:26 +08:00
committed by GitHub
parent 05611df056
commit b8b26ad700
23 changed files with 157 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
packages/service/common/mongo/sessionRun.ts
View File

@@ -1,17 +1,24 @@
import { addLog } from '../system/log';
import { connectionMongo, ClientSession } from './index';
export const mongoSessionRun = async <T = unknown>(fn: (session: ClientSession) => Promise<T>) => {
const session = await connectionMongo.startSession();
let committed = false;
try {
session.startTransaction();
const result = await fn(session);
await session.commitTransaction();
committed = true;
return result as T;
} catch (error) {
await session.abortTransaction();
if (!committed) {
await session.abortTransaction();
} else {
addLog.warn('Un catch mongo session error', { error });
}
return Promise.reject(error);
} finally {
await session.endSession();

6
packages/service/core/workflow/dispatch/index.ts
View File

@@ -217,16 +217,16 @@ export async function dispatchWorkFlow(data: Props): Promise<DispatchFlowRespons
if (res?.closed || props.maxRunTimes <= 0) return;
props.maxRunTimes--;
console.log(props.maxRunTimes, user._id);
addLog.debug(`Run node`, { maxRunTimes: props.maxRunTimes, uid: user._id });
await surrenderProcess();
if (status === 'run') {
addLog.info(`[dispatchWorkFlow] nodeRunWithActive: ${node.name}`);
addLog.debug(`[dispatchWorkFlow] nodeRunWithActive: ${node.name}`);
return nodeRunWithActive(node);
}
if (status === 'skip') {
addLog.info(`[dispatchWorkFlow] nodeRunWithSkip: ${node.name}`);
addLog.debug(`[dispatchWorkFlow] nodeRunWithSkip: ${node.name}`);
return nodeRunWithSkip(node);
}