Add File

agents-flex-core/src/main/java/com/agentsflex/core/chain/ChainHolder.java

@@ -0,0 +1,239 @@
+/*
+ *  Copyright (c) 2023-2025, Agents-Flex (fuhai999@gmail.com).
+ *  <p>
+ *  Licensed under the Apache License, Version 2.0 (the "License");
+ *  you may not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ *  <p>
+ *  http://www.apache.org/licenses/LICENSE-2.0
+ *  <p>
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ */
+package com.agentsflex.core.chain;
+
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.parser.DefaultJSONParser;
+import com.alibaba.fastjson.parser.Feature;
+import com.alibaba.fastjson.parser.ParserConfig;
+import com.alibaba.fastjson.parser.deserializer.ObjectDeserializer;
+import com.alibaba.fastjson.serializer.JSONSerializer;
+import com.alibaba.fastjson.serializer.ObjectSerializer;
+import com.alibaba.fastjson.serializer.SerializeConfig;
+import com.alibaba.fastjson.serializer.SerializerFeature;
+
+import java.io.IOException;
+import java.io.Serializable;
+import java.lang.reflect.Type;
+import java.util.List;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+
+public class ChainHolder implements Serializable {
+
+    private String id;
+    private String name;
+    private String description;
+
+    private List<ChainNode> nodes;
+    private List<ChainEdge> edges;
+
+    private Map<String, Object> executeResult;
+
+    private Map<String, NodeContext> nodeContexts;
+
+    protected Map<String, ChainNode> suspendNodes = new ConcurrentHashMap<>();
+    private List<Parameter> suspendForParameters;
+    private ChainStatus status;
+    private String message;
+
+    public ChainHolder() {
+    }
+
+
+    public static ChainHolder fromChain(Chain chain) {
+        ChainHolder holder = new ChainHolder();
+        holder.id = chain.getId();
+        holder.name = chain.getName();
+        holder.description = chain.getDescription();
+
+        holder.nodes = chain.getNodes();
+        holder.edges = chain.getEdges();
+
+        holder.executeResult = chain.getExecuteResult();
+        holder.nodeContexts = chain.getNodeContexts();
+
+        holder.suspendNodes = chain.getSuspendNodes();
+        holder.suspendForParameters = chain.getSuspendForParameters();
+        holder.status = chain.getStatus();
+
+        holder.message = chain.getMessage();
+
+        return holder;
+    }
+
+
+    public static ChainHolder fromJSON(String jsonString) {
+        ParserConfig config = new ParserConfig();
+        config.putDeserializer(Chain.class, new ChainDeserializer());
+        return JSON.parseObject(jsonString, ChainHolder.class, config, Feature.SupportAutoType);
+    }
+
+    public String toJSON() {
+        SerializeConfig config = new SerializeConfig();
+        config.put(Chain.class, new ChainSerializer());
+        return JSON.toJSONString(this, config, SerializerFeature.WriteClassName);
+    }
+
+    public Chain toChain() {
+        return new Chain(this);
+    }
+
+
+    public String getId() {
+        return id;
+    }
+
+    public void setId(String id) {
+        this.id = id;
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public void setName(String name) {
+        this.name = name;
+    }
+
+    public String getDescription() {
+        return description;
+    }
+
+    public void setDescription(String description) {
+        this.description = description;
+    }
+
+//    public ChainHolder getParent() {
+//        return parent;
+//    }
+//
+//    public void setParent(ChainHolder parent) {
+//        this.parent = parent;
+//    }
+//
+//    public List<ChainHolder> getChildren() {
+//        return children;
+//    }
+//
+//    public void setChildren(List<ChainHolder> children) {
+//        this.children = children;
+//    }
+
+    public List<ChainNode> getNodes() {
+        return nodes;
+    }
+
+    public void setNodes(List<ChainNode> nodes) {
+        this.nodes = nodes;
+    }
+
+    public List<ChainEdge> getEdges() {
+        return edges;
+    }
+
+    public void setEdges(List<ChainEdge> edges) {
+        this.edges = edges;
+    }
+
+    public Map<String, Object> getExecuteResult() {
+        return executeResult;
+    }
+
+    public void setExecuteResult(Map<String, Object> executeResult) {
+        this.executeResult = executeResult;
+    }
+
+    public Map<String, NodeContext> getNodeContexts() {
+        return nodeContexts;
+    }
+
+    public void setNodeContexts(Map<String, NodeContext> nodeContexts) {
+        this.nodeContexts = nodeContexts;
+    }
+
+    public Map<String, ChainNode> getSuspendNodes() {
+        return suspendNodes;
+    }
+
+    public void setSuspendNodes(Map<String, ChainNode> suspendNodes) {
+        this.suspendNodes = suspendNodes;
+    }
+
+    public List<Parameter> getSuspendForParameters() {
+        return suspendForParameters;
+    }
+
+    public void setSuspendForParameters(List<Parameter> suspendForParameters) {
+        this.suspendForParameters = suspendForParameters;
+    }
+
+    public ChainStatus getStatus() {
+        return status;
+    }
+
+    public void setStatus(ChainStatus status) {
+        this.status = status;
+    }
+
+    public String getMessage() {
+        return message;
+    }
+
+    public void setMessage(String message) {
+        this.message = message;
+    }
+
+    public static class ChainSerializer implements ObjectSerializer {
+        @Override
+        public void write(JSONSerializer serializer, Object object, Object fieldName, Type fieldType, int features) throws IOException {
+            if (object == null) {
+                serializer.writeNull();
+                return;
+            }
+            Chain chain = (Chain) object;
+            serializer.write(chain.toJSON());
+        }
+    }
+
+    public static class ChainDeserializer implements ObjectDeserializer {
+        @Override
+        public <T> T deserialze(DefaultJSONParser parser, Type type, Object fieldName) {
+            String value = parser.parseObject(String.class);
+            //noinspection unchecked
+            return (T) Chain.fromJSON(value);
+        }
+    }
+
+    @Override
+    public String toString() {
+        return "ChainHolder{" +
+            "id='" + id + '\'' +
+            ", name='" + name + '\'' +
+            ", description='" + description + '\'' +
+//            ", parent=" + parent +
+//            ", children=" + children +
+            ", nodes=" + nodes +
+            ", edges=" + edges +
+            ", executeResult=" + executeResult +
+            ", nodeContexts=" + nodeContexts +
+            ", suspendNodes=" + suspendNodes +
+            ", suspendForParameters=" + suspendForParameters +
+            ", status=" + status +
+            ", message='" + message + '\'' +
+            '}';
+    }
+}