From a2f73de90fb9c9d0ea7a5e7e9e6b9e445a8094ee Mon Sep 17 00:00:00 2001 From: Linnea Gräf Date: Sun, 15 Dec 2024 19:31:46 +0100 Subject: refactor: Move DI to its own package --- src/main/kotlin/moe/nea/ledger/utils/DI.kt | 68 ---------------------- src/main/kotlin/moe/nea/ledger/utils/DIProvider.kt | 51 ---------------- src/main/kotlin/moe/nea/ledger/utils/ErrorUtil.kt | 1 + src/main/kotlin/moe/nea/ledger/utils/Inject.kt | 6 -- src/main/kotlin/moe/nea/ledger/utils/di/DI.kt | 68 ++++++++++++++++++++++ .../kotlin/moe/nea/ledger/utils/di/DIProvider.kt | 51 ++++++++++++++++ src/main/kotlin/moe/nea/ledger/utils/di/Inject.kt | 6 ++ 7 files changed, 126 insertions(+), 125 deletions(-) delete mode 100644 src/main/kotlin/moe/nea/ledger/utils/DI.kt delete mode 100644 src/main/kotlin/moe/nea/ledger/utils/DIProvider.kt delete mode 100644 src/main/kotlin/moe/nea/ledger/utils/Inject.kt create mode 100644 src/main/kotlin/moe/nea/ledger/utils/di/DI.kt create mode 100644 src/main/kotlin/moe/nea/ledger/utils/di/DIProvider.kt create mode 100644 src/main/kotlin/moe/nea/ledger/utils/di/Inject.kt (limited to 'src/main/kotlin/moe/nea/ledger/utils') diff --git a/src/main/kotlin/moe/nea/ledger/utils/DI.kt b/src/main/kotlin/moe/nea/ledger/utils/DI.kt deleted file mode 100644 index e998c65..0000000 --- a/src/main/kotlin/moe/nea/ledger/utils/DI.kt +++ /dev/null @@ -1,68 +0,0 @@ -package moe.nea.ledger.utils - -import java.lang.reflect.AnnotatedElement -import java.util.Collections -import java.util.Stack - -@Suppress("UNCHECKED_CAST") -class DI { - private fun internalProvide(type: Class, element: AnnotatedElement? = null): T { - val provider = providers[type] as BaseDIProvider - val context = if (element == null) provider.createEmptyContext() else provider.createContext(element) - val key = Pair(type, context) - val existingValue = values[key] - if (existingValue != null) return existingValue as T - if (type in injectionStack) { - error("Found injection cycle: ${injectionStack.joinToString(" -> ")} -> $type") - } - injectionStack.push(type) - val value = try { - provider.provideWithContext(this, context) - } catch (ex: Exception) { - throw RuntimeException("Could not create instance for type $type", ex) - } - val cycleCheckCookie = injectionStack.pop() - require(cycleCheckCookie == type) { "Unbalanced stack cookie: $cycleCheckCookie != $type" } - values[key] = value - return value - } - - fun provide(type: Class, element: AnnotatedElement? = null): T { - return internalProvide(type, element) - } - - inline fun provide(): T = provide(T::class.java) - - fun register(type: Class, provider: BaseDIProvider) { - providers[type] = provider - } - - fun registerInjectableClasses(vararg type: Class<*>) { - type.forEach { internalRegisterInjectableClass(it) } - } - - private fun internalRegisterInjectableClass(type: Class) { - register(type, DIProvider.fromInjectableClass(type)) - } - - fun instantiateAll() { - providers.keys.forEach { - provide(it, null) - } - } - - fun getAllInstances(): Collection = - Collections.unmodifiableCollection(values.values) - - fun registerSingleton(value: T) { - register(value.javaClass, DIProvider.singeleton(value)) - } - - private val injectionStack: Stack> = Stack() - private val values = mutableMapOf, *>, Any>() - private val providers = mutableMapOf, BaseDIProvider<*, *>>() - - init { - registerSingleton(this) - } -} \ No newline at end of file diff --git a/src/main/kotlin/moe/nea/ledger/utils/DIProvider.kt b/src/main/kotlin/moe/nea/ledger/utils/DIProvider.kt deleted file mode 100644 index 98d1bbc..0000000 --- a/src/main/kotlin/moe/nea/ledger/utils/DIProvider.kt +++ /dev/null @@ -1,51 +0,0 @@ -package moe.nea.ledger.utils - -import java.lang.reflect.AnnotatedElement -import java.lang.reflect.Constructor - -fun interface DIProvider : BaseDIProvider { - override fun provideWithContext(di: DI, context: Unit): T { - return provide(di) - } - - override fun createContext(element: AnnotatedElement) { - } - - override fun createEmptyContext() { - } - - fun provide(di: DI): T - - companion object { - - fun fromInjectableClass(clazz: Class): DIProvider { - @Suppress("UNCHECKED_CAST") - val cons = (clazz.constructors.find { it.getAnnotation(Inject::class.java) != null } - ?: clazz.constructors.find { it.parameterCount == 0 } - ?: error("Could not find DI injection entrypoint for class $clazz")) - as Constructor - return DIProvider { di -> - val typArgs = cons.parameters.map { - di.provide(it.type, it) - }.toTypedArray() - val instance = cons.newInstance(*typArgs) - for (it in clazz.fields) { - if (it.getAnnotation(Inject::class.java) == null) continue - it.set(instance, di.provide(it.type, it)) - } - instance - } - } - - fun singeleton(value: T): DIProvider { - return DIProvider { _ -> value } - } - } - -} - -interface BaseDIProvider { - fun createContext(element: AnnotatedElement): C - fun provideWithContext(di: DI, context: C): T - fun createEmptyContext(): C -} diff --git a/src/main/kotlin/moe/nea/ledger/utils/ErrorUtil.kt b/src/main/kotlin/moe/nea/ledger/utils/ErrorUtil.kt index 9b6a153..4ba313e 100644 --- a/src/main/kotlin/moe/nea/ledger/utils/ErrorUtil.kt +++ b/src/main/kotlin/moe/nea/ledger/utils/ErrorUtil.kt @@ -1,5 +1,6 @@ package moe.nea.ledger.utils +import moe.nea.ledger.utils.di.Inject import moe.nea.ledger.utils.telemetry.ContextValue import moe.nea.ledger.utils.telemetry.EventRecorder import moe.nea.ledger.utils.telemetry.Span diff --git a/src/main/kotlin/moe/nea/ledger/utils/Inject.kt b/src/main/kotlin/moe/nea/ledger/utils/Inject.kt deleted file mode 100644 index 654f77e..0000000 --- a/src/main/kotlin/moe/nea/ledger/utils/Inject.kt +++ /dev/null @@ -1,6 +0,0 @@ -package moe.nea.ledger.utils - -@Retention(AnnotationRetention.RUNTIME) -@Target(AnnotationTarget.CONSTRUCTOR, AnnotationTarget.FIELD) -annotation class Inject( -) diff --git a/src/main/kotlin/moe/nea/ledger/utils/di/DI.kt b/src/main/kotlin/moe/nea/ledger/utils/di/DI.kt new file mode 100644 index 0000000..6940f72 --- /dev/null +++ b/src/main/kotlin/moe/nea/ledger/utils/di/DI.kt @@ -0,0 +1,68 @@ +package moe.nea.ledger.utils.di + +import java.lang.reflect.AnnotatedElement +import java.util.Collections +import java.util.Stack + +@Suppress("UNCHECKED_CAST") +class DI { + private fun internalProvide(type: Class, element: AnnotatedElement? = null): T { + val provider = providers[type] as BaseDIProvider + val context = if (element == null) provider.createEmptyContext() else provider.createContext(element) + val key = Pair(type, context) + val existingValue = values[key] + if (existingValue != null) return existingValue as T + if (type in injectionStack) { + error("Found injection cycle: ${injectionStack.joinToString(" -> ")} -> $type") + } + injectionStack.push(type) + val value = try { + provider.provideWithContext(this, context) + } catch (ex: Exception) { + throw RuntimeException("Could not create instance for type $type", ex) + } + val cycleCheckCookie = injectionStack.pop() + require(cycleCheckCookie == type) { "Unbalanced stack cookie: $cycleCheckCookie != $type" } + values[key] = value + return value + } + + fun provide(type: Class, element: AnnotatedElement? = null): T { + return internalProvide(type, element) + } + + inline fun provide(): T = provide(T::class.java) + + fun register(type: Class, provider: BaseDIProvider) { + providers[type] = provider + } + + fun registerInjectableClasses(vararg type: Class<*>) { + type.forEach { internalRegisterInjectableClass(it) } + } + + private fun internalRegisterInjectableClass(type: Class) { + register(type, DIProvider.fromInjectableClass(type)) + } + + fun instantiateAll() { + providers.keys.forEach { + provide(it, null) + } + } + + fun getAllInstances(): Collection = + Collections.unmodifiableCollection(values.values) + + fun registerSingleton(value: T) { + register(value.javaClass, DIProvider.singeleton(value)) + } + + private val injectionStack: Stack> = Stack() + private val values = mutableMapOf, *>, Any>() + private val providers = mutableMapOf, BaseDIProvider<*, *>>() + + init { + registerSingleton(this) + } +} \ No newline at end of file diff --git a/src/main/kotlin/moe/nea/ledger/utils/di/DIProvider.kt b/src/main/kotlin/moe/nea/ledger/utils/di/DIProvider.kt new file mode 100644 index 0000000..b5ce550 --- /dev/null +++ b/src/main/kotlin/moe/nea/ledger/utils/di/DIProvider.kt @@ -0,0 +1,51 @@ +package moe.nea.ledger.utils.di + +import java.lang.reflect.AnnotatedElement +import java.lang.reflect.Constructor + +fun interface DIProvider : BaseDIProvider { + override fun provideWithContext(di: DI, context: Unit): T { + return provide(di) + } + + override fun createContext(element: AnnotatedElement) { + } + + override fun createEmptyContext() { + } + + fun provide(di: DI): T + + companion object { + + fun fromInjectableClass(clazz: Class): DIProvider { + @Suppress("UNCHECKED_CAST") + val cons = (clazz.constructors.find { it.getAnnotation(Inject::class.java) != null } + ?: clazz.constructors.find { it.parameterCount == 0 } + ?: error("Could not find DI injection entrypoint for class $clazz")) + as Constructor + return DIProvider { di -> + val typArgs = cons.parameters.map { + di.provide(it.type, it) + }.toTypedArray() + val instance = cons.newInstance(*typArgs) + for (it in clazz.fields) { + if (it.getAnnotation(Inject::class.java) == null) continue + it.set(instance, di.provide(it.type, it)) + } + instance + } + } + + fun singeleton(value: T): DIProvider { + return DIProvider { _ -> value } + } + } + +} + +interface BaseDIProvider { + fun createContext(element: AnnotatedElement): C + fun provideWithContext(di: DI, context: C): T + fun createEmptyContext(): C +} diff --git a/src/main/kotlin/moe/nea/ledger/utils/di/Inject.kt b/src/main/kotlin/moe/nea/ledger/utils/di/Inject.kt new file mode 100644 index 0000000..a8fdd87 --- /dev/null +++ b/src/main/kotlin/moe/nea/ledger/utils/di/Inject.kt @@ -0,0 +1,6 @@ +package moe.nea.ledger.utils.di + +@Retention(AnnotationRetention.RUNTIME) +@Target(AnnotationTarget.CONSTRUCTOR, AnnotationTarget.FIELD) +annotation class Inject( +) -- cgit