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

refactor(framework): validate and collect @Shutdown hooks during agent

Browse Source 
registration
This commit is contained in:
slhafzjw
2026-02-28 20:38:09 +08:00
parent 1abfc729f8
commit d5095359db
2 changed files with 47 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
Partner-Framework/src/main/java/work/slhaf/partner/api/agent/factory/AgentRegisterFactory.kt
View File

@@ -10,6 +10,7 @@ import work.slhaf.partner.api.agent.factory.component.ComponentInjectorFactory
import work.slhaf.partner.api.agent.factory.component.ComponentRegisterFactory
import work.slhaf.partner.api.agent.factory.config.ConfigLoaderFactory
import work.slhaf.partner.api.agent.factory.context.AgentRegisterContext
import work.slhaf.partner.api.agent.factory.context.ShutdownHookCollectorFactory
import work.slhaf.partner.api.agent.factory.exception.ExternalModuleLoadFailedException
import work.slhaf.partner.api.agent.factory.exception.ExternalModulePathNotExistException
import java.io.File
@@ -38,6 +39,8 @@ object AgentRegisterFactory {
CapabilityInjectorFactory().execute(registerContext)
// 7. 执行模块 Init Hook 逻辑
ComponentInitHookExecuteFactory().execute(registerContext)
// 8. 校验并收集 Shutdown Hook 逻辑，并添加至 AgentContext 中
ShutdownHookCollectorFactory().execute(registerContext)
}
@JvmStatic

44
Partner-Framework/src/main/java/work/slhaf/partner/api/agent/factory/context/ShutdownHookCollectorFactory.kt Normal file
View File

@@ -0,0 +1,44 @@
package work.slhaf.partner.api.agent.factory.context
import work.slhaf.partner.api.agent.factory.AgentBaseFactory
import work.slhaf.partner.api.agent.factory.capability.annotation.CapabilityCore
import work.slhaf.partner.api.agent.factory.component.annotation.AgentComponent
import work.slhaf.partner.api.agent.factory.component.exception.ModuleCheckException
import work.slhaf.partner.api.agent.util.AgentUtil
class ShutdownHookCollectorFactory : AgentBaseFactory() {
override fun execute(context: AgentRegisterContext) {
val reflections = context.reflections
val agentContext = context.agentContext
reflections.getMethodsAnnotatedWith(Shutdown::class.java)
.forEach { method ->
val declaringClass = method.declaringClass
val isAgentComponentRelated =
AgentUtil.isAssignableFromAnnotation(declaringClass, AgentComponent::class.java)
val isCapabilityCoreRelated =
AgentUtil.isAssignableFromAnnotation(declaringClass, CapabilityCore::class.java)
if (!isAgentComponentRelated && !isCapabilityCoreRelated) {
throw ModuleCheckException(
"@Shutdown 仅能用于 AgentComponent/CapabilityCore 相关类: " +
"${declaringClass.name}#${method.name}"
)
}
if (method.parameterCount > 0) {
throw ModuleCheckException(
"@Shutdown 标注的方法不能包含形参: " +
"${declaringClass.name}#${method.name}"
)
}
val order = method.getAnnotation(Shutdown::class.java).order
val added = agentContext.addShutdownHook(method, order)
if (!added) {
throw ModuleCheckException(
"ShutdownHook 收集失败: ${declaringClass.name}#${method.name}"
)
}
}
}
}