diff --git a/src/components/Canvas/FlowCanvas/manager/GenerateManager.ts b/src/components/Canvas/FlowCanvas/manager/GenerateManager.ts index f422b01..a16461c 100644 --- a/src/components/Canvas/FlowCanvas/manager/GenerateManager.ts +++ b/src/components/Canvas/FlowCanvas/manager/GenerateManager.ts @@ -1,4 +1,5 @@ import { getTaskidResult } from '@/api/flow-canvas' +import { ElMessage } from 'element-plus' // interface NodeOptions { // } export class GenerateManager { @@ -27,8 +28,8 @@ export class GenerateManager { this.getTaskIdsImgTime = setInterval(()=>{ let taskIds = this.taskIds.map((item)=>item.taskId) getTaskidResult({taskIds}).then((rv:any)=>{ - //找出成功的 - let returnedTasks = rv.filter((item)=>item.status == 'RETURNED') + //找出成功和失败的任务 + let returnedTasks = rv.filter((item)=>item.status == 'RETURNED' || item.status == 'FAILED') if(returnedTasks.length == 0)return //剔除调成功的 let taskIds_ = JSON.parse(JSON.stringify(this.taskIds)) @@ -45,9 +46,14 @@ export class GenerateManager { result.forEach(item => { this.stateManager.getNodeById(item.nodeId).data.data.imageProcessTasks.forEach((nodeDataItem:any)=>{ if(item.taskId == nodeDataItem.taskId){ - nodeDataItem.url = item.url - nodeDataItem.createTime = item.createTime - nodeDataItem.status = item.status + if(item.status == 'FAILED'){ + this.stateManager.deleteNode(item.nodeId) + ElMessage.error(item.errorMessage) + }else if(item.status == 'RETURNED'){ + nodeDataItem.url = item.url + nodeDataItem.createTime = item.createTime + nodeDataItem.status = item.status + } } }) })