From b6ba156d67ecd6fc1d91650ba32b32791ff66f18 Mon Sep 17 00:00:00 2001
From: zhuangtianxiang <2913129173@qq.com>
Date: Fri, 16 May 2025 18:13:15 +0800
Subject: [PATCH] =?UTF-8?q?refactor(dify):=20=E4=BC=98=E5=8C=96=20workflow?=
 =?UTF-8?q?=20=E6=95=B0=E6=8D=AE=E5=A4=84=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 修改 WorkflowData 实体类，将 data 字段类型从 String 改为 WorkflowRunResponse.WorkflowRunData
- 更新 DifyWorkflowServiceImpl 中的数据处理逻辑，直接使用 WorkflowRunResponse.WorkflowRunData 对象
- 在 WorkflowRepository 中添加 getData 方法，用于获取未解析的 data 字符串
---
 .../zsc/edu/dify/modules/dify/entity/WorkflowData.java    | 8 ++++----
 .../edu/dify/modules/dify/repo/WorkflowRepository.java    | 4 ++++
 .../dify/service/Impl/DifyWorkflowServiceImpl.java        | 8 +++++---
 3 files changed, 13 insertions(+), 7 deletions(-)

diff --git a/src/main/java/com/zsc/edu/dify/modules/dify/entity/WorkflowData.java b/src/main/java/com/zsc/edu/dify/modules/dify/entity/WorkflowData.java
index 9b063a8..28dedcd 100644
--- a/src/main/java/com/zsc/edu/dify/modules/dify/entity/WorkflowData.java
+++ b/src/main/java/com/zsc/edu/dify/modules/dify/entity/WorkflowData.java
@@ -35,9 +35,9 @@ public class WorkflowData{
     @TableField(value = "dept_id", fill = FieldFill.INSERT)
     private Long deptId;
 
-    @TableField(exist = false)
-    private WorkflowRunResponse.WorkflowRunData runData;
-
     @TableField(typeHandler = JsonbTypeHandler.class)
-    private String data;
+    private WorkflowRunResponse.WorkflowRunData data;
+
+//    @TableField(typeHandler = JsonbTypeHandler.class)
+//    private String runData;
 }
diff --git a/src/main/java/com/zsc/edu/dify/modules/dify/repo/WorkflowRepository.java b/src/main/java/com/zsc/edu/dify/modules/dify/repo/WorkflowRepository.java
index 7c4e7a7..aa07660 100644
--- a/src/main/java/com/zsc/edu/dify/modules/dify/repo/WorkflowRepository.java
+++ b/src/main/java/com/zsc/edu/dify/modules/dify/repo/WorkflowRepository.java
@@ -2,6 +2,10 @@ package com.zsc.edu.dify.modules.dify.repo;
 
 import com.baomidou.mybatisplus.core.mapper.BaseMapper;
 import com.zsc.edu.dify.modules.dify.entity.WorkflowData;
+import org.apache.ibatis.annotations.Select;
 
 public interface WorkflowRepository extends BaseMapper<WorkflowData> {
+
+    @Select("select data from workflow_data where id = #{id}")
+    String getData(Long id);
 }
diff --git a/src/main/java/com/zsc/edu/dify/modules/dify/service/Impl/DifyWorkflowServiceImpl.java b/src/main/java/com/zsc/edu/dify/modules/dify/service/Impl/DifyWorkflowServiceImpl.java
index 9e903cf..450c89a 100644
--- a/src/main/java/com/zsc/edu/dify/modules/dify/service/Impl/DifyWorkflowServiceImpl.java
+++ b/src/main/java/com/zsc/edu/dify/modules/dify/service/Impl/DifyWorkflowServiceImpl.java
@@ -16,6 +16,8 @@ import jakarta.annotation.Resource;
 import lombok.AllArgsConstructor;
 import org.springframework.stereotype.Service;
 
+import java.nio.ByteBuffer;
+
 @AllArgsConstructor
 @Service
 public class DifyWorkflowServiceImpl extends ServiceImpl<WorkflowRepository, WorkflowData> implements DifyWorkflowService {
@@ -39,7 +41,7 @@ public class DifyWorkflowServiceImpl extends ServiceImpl<WorkflowRepository, Wor
         WorkflowRunResponse response = difyWorkflow.runWorkflow(request);
         WorkflowData workflowData =mapper.toEntity(response);
         workflowData.setAppId(appId);
-        workflowData.setData(String.valueOf(response.getData()));
+        workflowData.setData(response.getData());
         baseMapper.insert(workflowData);
         return response;
     }
@@ -51,8 +53,8 @@ public class DifyWorkflowServiceImpl extends ServiceImpl<WorkflowRepository, Wor
             throw new RuntimeException("未找到数据");
         }
         WorkflowRunResponse.WorkflowRunData parsedData =
-                JSON.parseObject(workflowData.getData(), WorkflowRunResponse.WorkflowRunData.class);
-        workflowData.setRunData(parsedData);
+                JSON.parseObject(baseMapper.getData(id), WorkflowRunResponse.WorkflowRunData.class);
+        workflowData.setData(parsedData);
         return workflowData;
     }