From 816d344ad60d674ee7a591ccda8113ef2f9fdba6 Mon Sep 17 00:00:00 2001
From: huangfei1994 <1045431781@qq.com>
Date: Fri, 13 Dec 2024 09:52:08 +0800
Subject: [PATCH] =?UTF-8?q?wedpr-web=E6=9B=B4=E6=96=B0=E7=BB=93=E6=9E=9C?=
 =?UTF-8?q?=E8=8E=B7=E5=8F=96=20(#201)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Co-authored-by: flyhuang <flyhuang@webank.com>
---
 wedpr-web/src/views/jobDetail/index.vue | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/wedpr-web/src/views/jobDetail/index.vue b/wedpr-web/src/views/jobDetail/index.vue
index b5e43031..c58e7dbb 100644
--- a/wedpr-web/src/views/jobDetail/index.vue
+++ b/wedpr-web/src/views/jobDetail/index.vue
@@ -398,16 +398,25 @@ export default {
           this.jobInfo.sql = sql
         }
         // 查询任务模型和结果
-        if (this.jobStatusInfo.status === 'RunSuccess' && this.receiverList.includes(this.agencyId)) {
-          if ([jobEnum.XGB_TRAINING, jobEnum.XGB_PREDICTING, jobEnum.LR_TRAINING, jobEnum.LR_PREDICTING].includes(this.jobInfo.jobType)) {
+        if (this.jobStatusInfo.status === 'RunSuccess') {
+          if (
+            [jobEnum.XGB_TRAINING, jobEnum.XGB_PREDICTING, jobEnum.LR_TRAINING, jobEnum.LR_PREDICTING].includes(this.jobInfo.jobType) &&
+            this.receiverList.includes(this.agencyId)
+          ) {
             this.getJobResult()
-          } else {
+          }
+          if (
+            [jobEnum.PIR, jobEnum.PSI, jobEnum.MPC, jobEnum.SQL].includes(this.jobInfo.jobType) &&
+            this.jobInfo.owner === this.userId &&
+            this.jobInfo.ownerAgency === this.agencyId
+          ) {
             // pir psi sql mpc 直接展示任务结果
             if (resultFileInfo) {
               this.resultFileInfo = resultFileInfo
             }
           }
         }
+
         this.startInterVal()
       } else {
         this.jobInfo = {}