pr suggestion

customerio · Sep 10, 2024 · 50f95d6 · 50f95d6
1 parent f384d57
commit 50f95d6
Showing 1 changed file with 9 additions and 17 deletions.
diff --git a/android/src/main/java/io/customer/reactnative/sdk/logging/CustomerIOReactNativeLogger.kt b/android/src/main/java/io/customer/reactnative/sdk/logging/CustomerIOReactNativeLogger.kt
@@ -44,31 +44,23 @@ class CustomerIOReactNativeLoggingEmitter(
 }
 
 class CustomerIOReactNativeLoggingWrapper private constructor(
-    private var moduleRegistry: ReactApplicationContext,
+    private var reactContext: ReactApplicationContext,
     override var logLevel: CioLogLevel): Logger {
 
     private val emitter: CustomerIOReactNativeLoggingEmitter?
-        get() = moduleRegistry?.getNativeModule(CustomerIOReactNativeLoggingEmitter::class.java)
+        get() = reactContext?.getNativeModule(CustomerIOReactNativeLoggingEmitter::class.java)
 
     companion object {
-        fun getInstance(moduleRegistry: ReactApplicationContext, logLevel: CioLogLevel): CustomerIOReactNativeLoggingWrapper {
-            val instance = CustomerIOReactNativeLoggingWrapper(moduleRegistry, logLevel)
-            SDKComponent.overrideDependency<Logger>(instance)
-            return instance
+        fun getInstance(reactContext: ReactApplicationContext, logLevel: CioLogLevel): CustomerIOReactNativeLoggingWrapper {
+            return CustomerIOReactNativeLoggingWrapper(reactContext, logLevel).also {
+                SDKComponent.overrideDependency<Logger>(it)
+            }
         }
     }
 
-    override fun debug(message: String) {
-        emit(message, CioLogLevel.DEBUG)
-    }
-
-    override fun info(message: String) {
-        emit(message, CioLogLevel.INFO)
-    }
-
-    override fun error(message: String) {
-        emit(message, CioLogLevel.ERROR)
-    }
+    override fun debug(message: String) = emit(message, CioLogLevel.DEBUG)
+    override fun info(message: String) = emit(message, CioLogLevel.INFO)
+    override fun error(message: String) = emit(message, CioLogLevel.ERROR)
 
     private fun emit(message: String, level: CioLogLevel) {
         if (shouldEmit(level)) {