dcs 2 gün önce
ebeveyn
işleme
fb40c21030

+ 13 - 3
virgo.api/src/main/java/com/bosshand/virgo/api/workark/model/DifyWorkFlow.java

@@ -44,7 +44,9 @@ public class DifyWorkFlow {
 
     private int insertType;
 
-    private int organizationId;
+    private long organizationId;
+
+    private long fileId;
 
     public Long getId() {
         return id;
@@ -182,11 +184,19 @@ public class DifyWorkFlow {
         this.insertType = insertType;
     }
 
-    public int getOrganizationId() {
+    public long getOrganizationId() {
         return organizationId;
     }
 
-    public void setOrganizationId(int organizationId) {
+    public void setOrganizationId(long organizationId) {
         this.organizationId = organizationId;
     }
+
+    public long getFileId() {
+        return fileId;
+    }
+
+    public void setFileId(long fileId) {
+        this.fileId = fileId;
+    }
 }

+ 2 - 1
virgo.api/src/main/java/com/bosshand/virgo/api/workark/service/DifyService.java

@@ -125,7 +125,8 @@ public class DifyService {
                     difyWorkFlow.setUserId(userContext.getUserId());
                     difyWorkFlow.setType((int) inputs.get("type"));
                     difyWorkFlow.setInsertType((int) inputs.get("insertType"));
-                    difyWorkFlow.setOrganizationId((int) inputs.get("organizationId"));
+                    difyWorkFlow.setOrganizationId(Long.parseLong(inputs.get("organizationId").toString()));
+                    difyWorkFlow.setOrganizationId(Long.parseLong(inputs.get("fileId").toString()));
                     difyWorkFlow.setWorkflowRunId(event.getWorkflowRunId());
                     difyWorkFlow.setStatus("running");
                     difyWorkFlowDao.save(difyWorkFlow);

+ 6 - 2
virgo.api/src/main/resources/mapper/DifyWorkFlowMapper.xml

@@ -21,11 +21,12 @@
         <result column="type" property="type"/>
         <result column="insertType" property="insertType"/>
         <result column="organizationId" property="organizationId"/>
+        <result column="fileId" property="fileId"/>
     </resultMap>
 
     <insert id="save" parameterType="com.bosshand.virgo.api.workark.model.DifyWorkFlow" useGeneratedKeys="true" keyProperty="id">
-        INSERT INTO dify_work_flow(`simpleUUID`, `difyTypeId`, `date`, `userId`, `workflowRunId`, `status`, `inputs`, `outputs`, `createdAt`, `finishedAt`, `elapsedTime`, `error`, `totalSteps`, `totalTokens`, `type`, `insertType`, `organizationId`)
-        VALUES (#{simpleUUID}, #{difyTypeId}, now(), #{userId}, #{workflowRunId}, #{status}, #{inputs}, #{outputs}, #{createdAt}, #{finishedAt}, #{elapsedTime}, #{error}, #{totalSteps}, #{totalTokens}, #{type}, #{insertType}, #{organizationId})
+        INSERT INTO dify_work_flow(`simpleUUID`, `difyTypeId`, `date`, `userId`, `workflowRunId`, `status`, `inputs`, `outputs`, `createdAt`, `finishedAt`, `elapsedTime`, `error`, `totalSteps`, `totalTokens`, `type`, `insertType`, `organizationId`, `fileId`)
+        VALUES (#{simpleUUID}, #{difyTypeId}, now(), #{userId}, #{workflowRunId}, #{status}, #{inputs}, #{outputs}, #{createdAt}, #{finishedAt}, #{elapsedTime}, #{error}, #{totalSteps}, #{totalTokens}, #{type}, #{insertType}, #{organizationId}, #{fileId})
     </insert>
 
     <select id="getSimpleUUID" resultMap="result">
@@ -73,6 +74,9 @@
             <if test="organizationId != 0">
                 and organizationId = #{organizationId}
             </if>
+            <if test="fileId != 0">
+                and fileId = #{fileId}
+            </if>
             <if test="simpleUUID != null">
                 and simpleUUID = #{simpleUUID}
             </if>