slhaf/Partner
1
0
Fork 0
mirror of https://github.com/slhaf/Partner.git synced 2026-05-12 16:53:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(exception): register exception reporters on Agent launcher

Browse Source
This commit is contained in:
slhafzjw
2026-04-10 19:04:11 +08:00
parent 7b963df991
commit ec52d64e73
1 changed files with 33 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
Partner-Framework/src/main/java/work/slhaf/partner/framework/agent/Agent.java
View File

@@ -3,8 +3,11 @@ package work.slhaf.partner.framework.agent;
import lombok.NonNull; import lombok.NonNull;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import work.slhaf.partner.framework.agent.config.ConfigCenter; import work.slhaf.partner.framework.agent.config.ConfigCenter;
import work.slhaf.partner.framework.agent.exception.deprecated.AgentLaunchFailedException; import work.slhaf.partner.framework.agent.exception.AgentStartupException;
import work.slhaf.partner.framework.agent.exception.ExceptionReporter;
import work.slhaf.partner.framework.agent.exception.ExceptionReporterHandler;
import work.slhaf.partner.framework.agent.factory.AgentRegisterFactory; import work.slhaf.partner.framework.agent.factory.AgentRegisterFactory;
import work.slhaf.partner.framework.agent.factory.component.annotation.AgentComponent;
import work.slhaf.partner.framework.agent.factory.context.AgentContext; import work.slhaf.partner.framework.agent.factory.context.AgentContext;
import work.slhaf.partner.framework.agent.interaction.AgentGatewayRegistration; import work.slhaf.partner.framework.agent.interaction.AgentGatewayRegistration;
import work.slhaf.partner.framework.agent.interaction.AgentGatewayRegistry; import work.slhaf.partner.framework.agent.interaction.AgentGatewayRegistry;
@@ -30,6 +33,7 @@ public final class Agent {
private final Class<?> applicationClass; private final Class<?> applicationClass;
private final Set<AgentGatewayRegistration> gatewayRegistrations = new LinkedHashSet<>(); private final Set<AgentGatewayRegistration> gatewayRegistrations = new LinkedHashSet<>();
private final Set<ExceptionReporter> exceptionReporters = new LinkedHashSet<>();
private final Set<LifecycleHook> preShutdownHooks = new LinkedHashSet<>(); private final Set<LifecycleHook> preShutdownHooks = new LinkedHashSet<>();
private final Set<LifecycleHook> postShutdownHooks = new LinkedHashSet<>(); private final Set<LifecycleHook> postShutdownHooks = new LinkedHashSet<>();
@@ -42,6 +46,11 @@ public final class Agent {
return this; return this;
} }
public AgentApp addExceptionReporter(ExceptionReporter... exceptionReporters) {
this.exceptionReporters.addAll(Set.of(exceptionReporters));
return this;
}
public AgentApp addPreShutdownHook(String name, Runnable action) { public AgentApp addPreShutdownHook(String name, Runnable action) {
return addPreShutdownHook(name, 0, action); return addPreShutdownHook(name, 0, action);
} }
@@ -60,23 +69,43 @@ public final class Agent {
return this; return this;
} }
public void launch() { public boolean launch() {
try { try {
// Keep startup order explicit so registries are ready before component scanning. // Keep startup order explicit so registries are ready before component scanning.
for (ExceptionReporter exceptionReporter : exceptionReporters) {
// AgentComponent will be initialized by factory
if (exceptionReporter.getClass().isAnnotationPresent(AgentComponent.class)) {
continue;
}
exceptionReporter.register();
}
// Load class
StateCenter.INSTANCE.toString(); StateCenter.INSTANCE.toString();
// Register into config center
ModelRuntimeRegistry.INSTANCE.register(); ModelRuntimeRegistry.INSTANCE.register();
AgentGatewayRegistry.INSTANCE.register(); AgentGatewayRegistry.INSTANCE.register();
for (AgentGatewayRegistration registration : gatewayRegistrations) { for (AgentGatewayRegistration registration : gatewayRegistrations) {
registration.register(); registration.register();
} }
registerShutdownHooks(); registerShutdownHooks();
Path externalModuleDir = ConfigCenter.INSTANCE.getPaths().getResourcesDir().resolve("module"); Path externalModuleDir = ConfigCenter.INSTANCE.getPaths().getResourcesDir().resolve("module");
AgentRegisterFactory.addScanDir(externalModuleDir.toString()); AgentRegisterFactory.addScanDir(externalModuleDir.toString());
AgentRegisterFactory.launch(applicationClass.getPackageName()); AgentRegisterFactory.launch(applicationClass.getPackageName());
// Try to init configurable, and start config listening
ConfigCenter.INSTANCE.initAll(); ConfigCenter.INSTANCE.initAll();
ConfigCenter.INSTANCE.start(); ConfigCenter.INSTANCE.start();
} catch (Exception e) { return true;
throw new AgentLaunchFailedException("Agent 启动失败", e); } catch (AgentStartupException e) {
ExceptionReporterHandler.INSTANCE.report(e);
return false;
} catch (Throwable t) {
AgentStartupException wrapped = new AgentStartupException("Unexpected startup failure", "launcher", t);
ExceptionReporterHandler.INSTANCE.report(wrapped);
return false;
} }
} }