add buttons

Author: xlorne

example/example-infra-flow/src/main/java/com/codingapi/example/convert/FlowNodeConvertor.java

@@ -1,5 +1,6 @@
 package com.codingapi.example.convert;
 
+import com.alibaba.fastjson.JSON;
 import com.codingapi.example.entity.FlowNodeEntity;
 import com.codingapi.springboot.flow.domain.FlowNode;
 import com.codingapi.springboot.flow.em.ApprovalType;
@@ -30,6 +31,10 @@ public static FlowNodeEntity convert(FlowNode flowNode, long workId){
         entity.setOperatorMatcher(flowNode.getOperatorMatcher().getScript());
         entity.setUpdateTime(flowNode.getUpdateTime());
         entity.setWorkId(workId);
+        if(flowNode.getButtons()!=null) {
+            String json = JSON.toJSONString(flowNode.getButtons());
+            entity.setButtons(json);
+        }
         return entity;
     }
 
@@ -50,7 +55,8 @@ public static FlowNode convert(FlowNodeEntity entity){
                 entity.getCreateTime(),
                 entity.getUpdateTime(),
                 entity.getTimeout(),
-                StringUtils.hasLength(entity.getErrTrigger())?new ErrTrigger(entity.getErrTrigger()):null);
+                StringUtils.hasLength(entity.getErrTrigger())?new ErrTrigger(entity.getErrTrigger()):null,
+                entity.toFlowButtons());
     }
 
 }

example/example-infra-flow/src/main/java/com/codingapi/example/entity/FlowNodeEntity.java

@@ -1,10 +1,15 @@
 package com.codingapi.example.entity;
 
+import com.alibaba.fastjson.JSON;
+import com.codingapi.springboot.flow.domain.FlowButton;
 import jakarta.persistence.Entity;
 import jakarta.persistence.Id;
 import jakarta.persistence.Lob;
 import lombok.Getter;
 import lombok.Setter;
+import org.springframework.util.StringUtils;
+
+import java.util.List;
 
 @Setter
 @Getter
@@ -81,4 +86,19 @@ public class FlowNodeEntity {
      */
     @Lob
     private String errTrigger;
+
+
+    /**
+     * 自定义按钮
+     */
+    @Lob
+    private String buttons;
+
+
+    public List<FlowButton> toFlowButtons(){
+        if(StringUtils.hasText(buttons)) {
+            return JSON.parseArray(buttons, FlowButton.class);
+        }
+        return null;
+    }
 }