add #82

Author: xlorne

flow-engine-starter-api/src/main/java/com/codingapi/flow/api/controller/WorkflowController.java

@@ -76,6 +76,12 @@ public SingleResponse<String> mock() {
         return SingleResponse.of(FlowServiceMockFactory.getInstance().create(GatewayContext.getInstance().getFlowOperatorGateway()));
     }
 
+    @PostMapping("/cleanMock")
+    public Response mock(@RequestBody IdRequest request) {
+        FlowServiceMockFactory.getInstance().clear(request.getStringId());
+        return Response.buildSuccess();
+    }
+
 
     @PostMapping("/create")
     public SingleResponse<JSONObject> create() {

flow-engine-starter-query/src/main/java/com/codingapi/flow/query/controller/FlowRecordMockQueryController.java

@@ -5,19 +5,25 @@
 import com.codingapi.flow.infra.entity.FlowTodoRecordEntity;
 import com.codingapi.flow.infra.jpa.FlowRecordEntityRepository;
 import com.codingapi.flow.infra.jpa.FlowTodoRecordEntityRepository;
+import com.codingapi.flow.mock.FlowQueryMockService;
+import com.codingapi.flow.mock.FlowServiceMockFactory;
 import com.codingapi.flow.operator.IFlowOperator;
 import com.codingapi.flow.pojo.response.FlowRecordContent;
 import com.codingapi.springboot.framework.dto.request.Relation;
 import com.codingapi.springboot.framework.dto.request.SearchRequest;
 import com.codingapi.springboot.framework.dto.response.MultiResponse;
 import com.codingapi.springboot.framework.user.UserContext;
+import jakarta.servlet.http.HttpServletRequest;
 import lombok.AllArgsConstructor;
 import org.springframework.data.domain.Page;
 import org.springframework.data.domain.PageRequest;
 import org.springframework.data.domain.Sort;
+import org.springframework.util.StringUtils;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.context.request.RequestContextHolder;
+import org.springframework.web.context.request.ServletRequestAttributes;
 
 @RestController
 @RequestMapping("/api/query/record")
@@ -27,11 +33,28 @@ public class FlowRecordQueryController {
     private final FlowRecordEntityRepository flowRecordEntityRepository;
     private final FlowTodoRecordEntityRepository flowTodoRecordEntityRepository;
 
+
+    private FlowQueryMockService loadFlowService(){
+        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.currentRequestAttributes();
+        HttpServletRequest request = attributes.getRequest();
+        String key = request.getParameter("mockKey");
+        if(StringUtils.hasText(key)) {
+            return FlowServiceMockFactory.getInstance().getFlowQueryService(key);
+        }else {
+            return null;
+        }
+    }
+
     /**
      * 全部流程记录接口
      */
     @GetMapping("/list")
     public MultiResponse<FlowRecordContent> list(SearchRequest request) {
+        FlowQueryMockService flowQueryMockService = loadFlowService();
+        if(flowQueryMockService!=null){
+            PageRequest pageRequest = PageRequest.of(request.getCurrent(),request.getPageSize());
+            return MultiResponse.of(flowQueryMockService.findAll(pageRequest));
+        }
         request.addSort(Sort.by("id").descending());
         request.addFilter("revoked", Relation.EQUAL,false);
         Page<FlowRecordEntity> page = flowRecordEntityRepository.searchRequest(request);
@@ -45,6 +68,12 @@ public MultiResponse<FlowRecordContent> list(SearchRequest request) {
     @GetMapping("/todo")
     public MultiResponse<FlowRecordContent> todo(SearchRequest request) {
         IFlowOperator current = (IFlowOperator) UserContext.getInstance().current();
+        FlowQueryMockService flowQueryMockService = loadFlowService();
+        if(flowQueryMockService!=null){
+            PageRequest pageRequest = PageRequest.of(request.getCurrent(),request.getPageSize());
+            return MultiResponse.of(flowQueryMockService.findTodoRecordPage(current.getUserId(),pageRequest.withSort(Sort.by("id").descending())));
+        }
+
         PageRequest pageRequest = request.toPageRequest(FlowRecordEntity.class);
         Page<FlowTodoRecordEntity> page = flowTodoRecordEntityRepository.findTodoRecordPage(current.getUserId(),pageRequest.withSort(Sort.by("id").descending()));
         return MultiResponse.of(page.map(FlowRecordContentConvertor::convert));
@@ -56,6 +85,11 @@ public MultiResponse<FlowRecordContent> todo(SearchRequest request) {
     @GetMapping("/notify")
     public MultiResponse<FlowRecordContent> notify(SearchRequest request) {
         IFlowOperator current = (IFlowOperator) UserContext.getInstance().current();
+        FlowQueryMockService flowQueryMockService = loadFlowService();
+        if(flowQueryMockService!=null){
+            PageRequest pageRequest = PageRequest.of(request.getCurrent(),request.getPageSize());
+            return MultiResponse.of(flowQueryMockService.findNotifyRecordPage(current.getUserId(),pageRequest.withSort(Sort.by("id").descending())));
+        }
         PageRequest pageRequest = request.toPageRequest(FlowRecordEntity.class);
         Page<FlowRecordEntity> page =flowRecordEntityRepository.findNotifyRecordPage(current.getUserId(),pageRequest.withSort(Sort.by("id").descending()));
         return MultiResponse.of(page.map(FlowRecordContentConvertor::convert));
@@ -68,6 +102,11 @@ public MultiResponse<FlowRecordContent> notify(SearchRequest request) {
     @GetMapping("/done")
     public MultiResponse<FlowRecordContent> done(SearchRequest request) {
         IFlowOperator current = (IFlowOperator) UserContext.getInstance().current();
+        FlowQueryMockService flowQueryMockService = loadFlowService();
+        if(flowQueryMockService!=null){
+            PageRequest pageRequest = PageRequest.of(request.getCurrent(),request.getPageSize());
+            return MultiResponse.of(flowQueryMockService.findDoneRecordPage(current.getUserId(),pageRequest.withSort(Sort.by("id").descending())));
+        }
         PageRequest pageRequest = request.toPageRequest(FlowRecordEntity.class);
         Page<FlowRecordEntity> page =flowRecordEntityRepository.findDoneRecordPage(current.getUserId(),pageRequest.withSort(Sort.by("id").descending()));
         return MultiResponse.of(page.map(FlowRecordContentConvertor::convert));