vertoryao/dify-proxy-backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(dify): 优化 workflow 运行接口返回值类型- 将 DifyWorkflowService接口的 run 方法返回值类型从 WorkflowData 改为 WorkflowRunResponse

Browse Source 
- 更新 DifyWorkflowServiceImpl 中的 run 方法实现
-调整 V1WorkflowController 中的 runWorkflow 方法返回类型
This commit is contained in:
zhuangtianxiang 2025-05-06 19:24:12 +08:00
parent d64f4958d0
commit c1f8f896d6
3 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/main/java/com/zsc/edu/dify/modules/dify/controller/V1WorkflowController.java
View File

@ -37,7 +37,7 @@ public class V1WorkflowController {
*/ */
@PostMapping("/run/{appId}") @PostMapping("/run/{appId}")
@PreAuthorize("hasAuthority('dify:workflow:run')") @PreAuthorize("hasAuthority('dify:workflow:run')")
public WorkflowData runWorkflow(@RequestBody WorkflowRunRequest request, @PathVariable String appId) { public WorkflowRunResponse runWorkflow(@RequestBody WorkflowRunRequest request, @PathVariable String appId) {
return difyWorkflowService.run(request, appId); return difyWorkflowService.run(request, appId);
} }

4
src/main/java/com/zsc/edu/dify/modules/dify/service/DifyWorkflowService.java
View File

@ -3,7 +3,9 @@ package com.zsc.edu.dify.modules.dify.service;
import com.baomidou.mybatisplus.extension.service.IService; import com.baomidou.mybatisplus.extension.service.IService;
import com.zsc.edu.dify.modules.dify.entity.WorkflowData; import com.zsc.edu.dify.modules.dify.entity.WorkflowData;
import io.github.guoshiqiufeng.dify.workflow.dto.request.WorkflowRunRequest; import io.github.guoshiqiufeng.dify.workflow.dto.request.WorkflowRunRequest;
import io.github.guoshiqiufeng.dify.workflow.dto.response.WorkflowRunResponse;
public interface DifyWorkflowService extends IService<WorkflowData> { public interface DifyWorkflowService extends IService<WorkflowData> {
WorkflowData run(WorkflowRunRequest request, String appId); WorkflowRunResponse run(WorkflowRunRequest request, String appId);
} }

8
src/main/java/com/zsc/edu/dify/modules/dify/service/Impl/DifyWorkflowServiceImpl.java
View File

@ -8,6 +8,7 @@ import com.zsc.edu.dify.modules.dify.repo.WorkflowRepository;
import com.zsc.edu.dify.modules.dify.service.DifyWorkflowService; import com.zsc.edu.dify.modules.dify.service.DifyWorkflowService;
import io.github.guoshiqiufeng.dify.workflow.DifyWorkflow; import io.github.guoshiqiufeng.dify.workflow.DifyWorkflow;
import io.github.guoshiqiufeng.dify.workflow.dto.request.WorkflowRunRequest; import io.github.guoshiqiufeng.dify.workflow.dto.request.WorkflowRunRequest;
import io.github.guoshiqiufeng.dify.workflow.dto.response.WorkflowRunResponse;
import jakarta.annotation.Resource; import jakarta.annotation.Resource;
import lombok.AllArgsConstructor; import lombok.AllArgsConstructor;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
@ -23,12 +24,13 @@ public class DifyWorkflowServiceImpl extends ServiceImpl<WorkflowRepository, Wor
WorkflowMapper mapper; WorkflowMapper mapper;
@Override @Override
public WorkflowData run(WorkflowRunRequest request, String appId){ public WorkflowRunResponse run(WorkflowRunRequest request, String appId){
String apiKey = appEntityRepository.selectApiKey(appId); String apiKey = appEntityRepository.selectApiKey(appId);
request.setApiKey(apiKey); request.setApiKey(apiKey);
WorkflowData workflowData =mapper.toEntity(difyWorkflow.runWorkflow(request)); WorkflowRunResponse response = difyWorkflow.runWorkflow(request);
WorkflowData workflowData =mapper.toEntity(response);
workflowData.setAppId(appId); workflowData.setAppId(appId);
baseMapper.insert(workflowData); baseMapper.insert(workflowData);
return workflowData; return response;
} }
} }