test/agents-flex
1
0
Fork 0
mirror of https://gitee.com/agents-flex/agents-flex.git synced 2024-11-29 18:38:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor: optimize Chain events name

Browse Source
This commit is contained in:
Michael Yang 2024-11-20 10:23:28 +08:00
parent a1be752d92
commit ca6385776e
5 changed files with 22 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
agents-flex-core/src/main/java/com/agentsflex/core/chain/Chain.java
View File

@ -139,7 +139,7 @@ public class Chain extends ChainNode {
this.status = status; this.status = status;
if (before != status) { if (before != status) {
notifyEvent(new OnStatusChangeEvent(this.status, before)); notifyEvent(new OnStatusChangeEvent(this, this.status, before));
} }
} }
@ -260,7 +260,7 @@ public class Chain extends ChainNode {
} finally { } finally {
onNodeExecuteEnd(nodeContext); onNodeExecuteEnd(nodeContext);
ChainContext.clearNode(); ChainContext.clearNode();
notifyEvent(new OnNodeFinishedEvent(currentNode, executeResult)); notifyEvent(new OnNodeEndEvent(currentNode, executeResult));
} }
if (executeResult != null && !executeResult.isEmpty()) { if (executeResult != null && !executeResult.isEmpty()) {
@ -371,7 +371,7 @@ public class Chain extends ChainNode {
} }
try { try {
ChainContext.setChain(this); ChainContext.setChain(this);
notifyEvent(new OnStartEvent()); notifyEvent(new OnChainStartEvent());
try { try {
setStatus(ChainStatus.RUNNING); setStatus(ChainStatus.RUNNING);
runnable.run(); runnable.run();
@ -386,7 +386,7 @@ public class Chain extends ChainNode {
} }
} finally { } finally {
ChainContext.clearChain(); ChainContext.clearChain();
notifyEvent(new OnFinishedEvent()); notifyEvent(new OnChainEndEvent());
} }
} }

2
agents-flex-core/src/main/java/com/agentsflex/core/chain/event/OnFinishedEvent.java → agents-flex-core/src/main/java/com/agentsflex/core/chain/event/OnChainEndEvent.java
View File

@ -17,6 +17,6 @@ package com.agentsflex.core.chain.event;
import com.agentsflex.core.chain.ChainEvent; import com.agentsflex.core.chain.ChainEvent;
public class OnFinishedEvent implements ChainEvent { public class OnChainEndEvent implements ChainEvent {
} }

2
agents-flex-core/src/main/java/com/agentsflex/core/chain/event/OnStartEvent.java → agents-flex-core/src/main/java/com/agentsflex/core/chain/event/OnChainStartEvent.java
View File

@ -17,6 +17,6 @@ package com.agentsflex.core.chain.event;
import com.agentsflex.core.chain.ChainEvent; import com.agentsflex.core.chain.ChainEvent;
public class OnStartEvent implements ChainEvent { public class OnChainStartEvent implements ChainEvent {
} }

6
agents-flex-core/src/main/java/com/agentsflex/core/chain/event/OnNodeFinishedEvent.java → agents-flex-core/src/main/java/com/agentsflex/core/chain/event/OnNodeEndEvent.java
View File

@ -20,13 +20,13 @@ import com.agentsflex.core.chain.ChainNode;
import java.util.Map; import java.util.Map;
public class OnNodeFinishedEvent implements ChainEvent { public class OnNodeEndEvent implements ChainEvent {
private ChainNode node; private ChainNode node;
private Map<String, Object> result; private Map<String, Object> result;
public OnNodeFinishedEvent(ChainNode node, Map<String, Object> result) { public OnNodeEndEvent(ChainNode node, Map<String, Object> result) {
this.node = node; this.node = node;
this.result = result; this.result = result;
} }
@ -49,7 +49,7 @@ public class OnNodeFinishedEvent implements ChainEvent {
@Override @Override
public String toString() { public String toString() {
return "OnNodeFinishedEvent{" + return "OnNodeEndEvent{" +
"node=" + node + "node=" + node +
", result=" + result + ", result=" + result +
'}'; '}';

15
agents-flex-core/src/main/java/com/agentsflex/core/chain/event/OnStatusChangeEvent.java
View File

@ -15,19 +15,29 @@
*/ */
package com.agentsflex.core.chain.event; package com.agentsflex.core.chain.event;
import com.agentsflex.core.chain.Chain;
import com.agentsflex.core.chain.ChainEvent; import com.agentsflex.core.chain.ChainEvent;
import com.agentsflex.core.chain.ChainStatus; import com.agentsflex.core.chain.ChainStatus;
public class OnStatusChangeEvent implements ChainEvent { public class OnStatusChangeEvent implements ChainEvent {
private Chain chain;
private ChainStatus status; private ChainStatus status;
private ChainStatus before; private ChainStatus before;
public OnStatusChangeEvent(ChainStatus status, ChainStatus before) { public OnStatusChangeEvent(Chain chain,ChainStatus status, ChainStatus before) {
this.status = status; this.status = status;
this.before = before; this.before = before;
} }
public Chain getChain() {
return chain;
}
public void setChain(Chain chain) {
this.chain = chain;
}
public ChainStatus getStatus() { public ChainStatus getStatus() {
return status; return status;
} }
@ -47,7 +57,8 @@ public class OnStatusChangeEvent implements ChainEvent {
@Override @Override
public String toString() { public String toString() {
return "OnStatusChangeEvent{" + return "OnStatusChangeEvent{" +
"status=" + status + "chain=" + chain +
", status=" + status +
", before=" + before + ", before=" + before +
'}'; '}';
} }