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(framework): scope shutdown-only Instances and computeInstances helpers inside installShutdownHook

Browse Source
This commit is contained in:
slhafzjw
2026-02-27 00:34:07 +08:00
parent 333d087979
commit 528e88f613
1 changed files with 26 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
Partner-Framework/src/main/java/work/slhaf/partner/api/agent/factory/context/AgentContext.kt
View File

@@ -93,6 +93,32 @@ object AgentContext {
private fun installShutdownHook() { private fun installShutdownHook() {
class Instances(
val running: MutableMap<Class<out AbstractAgentModule.Running<out RunningFlowContext>>, Any> = mutableMapOf(),
val standalone: MutableMap<Class<out AbstractAgentModule.Standalone>, Any> = mutableMapOf(),
val sub: MutableMap<Class<out AbstractAgentModule.Sub<*, *>>, Any> = mutableMapOf(),
val additional: MutableMap<Class<*>, Any> = mutableMapOf(),
val capability: MutableMap<Class<*>, Any> = mutableMapOf()
)
fun computeInstances(): Instances {
val instances = Instances()
modules.values.forEach {
when (it) {
is ModuleContextData.Running<*> -> instances.running[it.clazz] = it.instance
is ModuleContextData.Standalone<*> -> instances.standalone[it.clazz] = it.instance
is ModuleContextData.Sub<*> -> instances.sub[it.clazz] = it.instance
}
}
instances.additional.putAll(additionalComponents)
capabilities.values.forEach {
instances.capability.putAll(it.cores)
}
return instances
}
fun getModuleInstance(clazz: Class<*>, instances: Instances): Any? { fun getModuleInstance(clazz: Class<*>, instances: Instances): Any? {
return if (AbstractAgentModule.Running::class.java.isAssignableFrom(clazz)) { return if (AbstractAgentModule.Running::class.java.isAssignableFrom(clazz)) {
instances.running[clazz] instances.running[clazz]
@@ -141,32 +167,6 @@ object AgentContext {
}) })
} }
private fun computeInstances(): Instances {
val instances = Instances()
modules.values.forEach {
when (it) {
is ModuleContextData.Running<*> -> instances.running[it.clazz] = it.instance
is ModuleContextData.Standalone<*> -> instances.standalone[it.clazz] = it.instance
is ModuleContextData.Sub<*> -> instances.sub[it.clazz] = it.instance
}
}
instances.additional.putAll(additionalComponents)
capabilities.values.forEach {
instances.capability.putAll(it.cores)
}
return instances
}
private class Instances(
val running: MutableMap<Class<out AbstractAgentModule.Running<out RunningFlowContext>>, Any> = mutableMapOf(),
val standalone: MutableMap<Class<out AbstractAgentModule.Standalone>, Any> = mutableMapOf(),
val sub: MutableMap<Class<out AbstractAgentModule.Sub<*, *>>, Any> = mutableMapOf(),
val additional: MutableMap<Class<*>, Any> = mutableMapOf(),
val capability: MutableMap<Class<*>, Any> = mutableMapOf()
)
data class MetaDataContent( data class MetaDataContent(
val clazz: Class<*>, val clazz: Class<*>,
val value: String val value: String