add operationRepoLoadedListener and implement it to RecoverFromDroppedLoginBug

Author: jinliu9508

OneSignalSDK/onesignal/core/src/main/java/com/onesignal/core/internal/operations/IOperationRepo.kt

@@ -38,6 +38,8 @@ interface IOperationRepo {
      * Check if the queue contains a specific operation type
      */
     fun <T : Operation> containsInstanceOf(type: KClass<T>): Boolean
+
+    fun addOperationLoadedListener(handler: IOperationRepoLoadedListener)
 }
 
 // Extension function so the syntax containsInstanceOf<Operation>() can be used over

OneSignalSDK/onesignal/core/src/main/java/com/onesignal/core/internal/operations/IOperationRepoLoadedListener.kt

@@ -0,0 +1,5 @@
+package com.onesignal.core.internal.operations
+
+interface IOperationRepoLoadedListener {
+    fun onOperationRepoLoaded()
+}

OneSignalSDK/onesignal/core/src/main/java/com/onesignal/core/internal/operations/impl/OperationRepo.kt

@@ -1,11 +1,14 @@
 package com.onesignal.core.internal.operations.impl
 
+import com.onesignal.common.events.EventProducer
+import com.onesignal.common.events.IEventNotifier
 import com.onesignal.common.threading.WaiterWithValue
 import com.onesignal.core.internal.config.ConfigModelStore
 import com.onesignal.core.internal.operations.ExecutionResult
 import com.onesignal.core.internal.operations.GroupComparisonType
 import com.onesignal.core.internal.operations.IOperationExecutor
 import com.onesignal.core.internal.operations.IOperationRepo
+import com.onesignal.core.internal.operations.IOperationRepoLoadedListener
 import com.onesignal.core.internal.operations.Operation
 import com.onesignal.core.internal.startup.IStartableService
 import com.onesignal.core.internal.time.ITime
@@ -27,7 +30,7 @@ internal class OperationRepo(
     private val _configModelStore: ConfigModelStore,
     private val _time: ITime,
     private val _newRecordState: NewRecordsState,
-) : IOperationRepo, IStartableService {
+) : IOperationRepo, IStartableService, IEventNotifier<IOperationRepoLoadedListener> {
     internal class OperationQueueItem(
         val operation: Operation,
         val waiter: WaiterWithValue<Boolean>? = null,
@@ -49,6 +52,18 @@ internal class OperationRepo(
     private val waiter = WaiterWithValue<LoopWaiterMessage>()
     private var paused = false
     private var coroutineScope = CoroutineScope(newSingleThreadContext(name = "OpRepo"))
+    private val loadedSubscription: EventProducer<IOperationRepoLoadedListener> = EventProducer()
+
+    override val hasSubscribers: Boolean
+        get() = loadedSubscription.hasSubscribers
+
+    override fun unsubscribe(handler: IOperationRepoLoadedListener) {
+        loadedSubscription.subscribe(handler)
+    }
+
+    override fun subscribe(handler: IOperationRepoLoadedListener) {
+        loadedSubscription.unsubscribe(handler)
+    }
 
     /** *** Buckets ***
      * Purpose: Bucketing is a pattern we are using to help save network
@@ -86,6 +101,10 @@ internal class OperationRepo(
         }
     }
 
+    override fun addOperationLoadedListener(handler: IOperationRepoLoadedListener) {
+        subscribe(handler)
+    }
+
     override fun start() {
         paused = false
         coroutineScope.launch {
@@ -393,5 +412,6 @@ internal class OperationRepo(
                 operation.index,
             )
         }
+        loadedSubscription.fire { it.onOperationRepoLoaded() }
     }
 }

OneSignalSDK/onesignal/core/src/main/java/com/onesignal/user/internal/migrations/RecoverFromDroppedLoginBug.kt

@@ -3,6 +3,7 @@ package com.onesignal.user.internal.migrations
 import com.onesignal.common.IDManager
 import com.onesignal.core.internal.config.ConfigModelStore
 import com.onesignal.core.internal.operations.IOperationRepo
+import com.onesignal.core.internal.operations.IOperationRepoLoadedListener
 import com.onesignal.core.internal.operations.containsInstanceOf
 import com.onesignal.core.internal.startup.IStartableService
 import com.onesignal.debug.internal.logging.Logging
@@ -30,8 +31,12 @@ class RecoverFromDroppedLoginBug(
     private val _operationRepo: IOperationRepo,
     private val _identityModelStore: IdentityModelStore,
     private val _configModelStore: ConfigModelStore,
-) : IStartableService {
+) : IStartableService, IOperationRepoLoadedListener {
     override fun start() {
+        _operationRepo.addOperationLoadedListener(this)
+    }
+
+    override fun onOperationRepoLoaded() {
         if (isInBadState()) {
             Logging.warn(
                 "User with externalId:" +

OneSignalSDK/onesignal/core/src/test/java/com/onesignal/core/internal/operations/OperationRepoTests.kt

@@ -9,6 +9,9 @@ import com.onesignal.core.internal.time.impl.Time
 import com.onesignal.debug.LogLevel
 import com.onesignal.debug.internal.logging.Logging
 import com.onesignal.mocks.MockHelper
+import com.onesignal.user.internal.identity.IdentityModel
+import com.onesignal.user.internal.identity.IdentityModelStore
+import com.onesignal.user.internal.migrations.RecoverFromDroppedLoginBug
 import com.onesignal.user.internal.operations.ExecutorMocks.Companion.getNewRecordState
 import io.kotest.core.spec.style.FunSpec
 import io.kotest.matchers.shouldBe
@@ -590,6 +593,20 @@ class OperationRepoTests : FunSpec({
         // Then
         result shouldBe null
     }
+
+    test("IOperationRepoLoadedListener") {
+        val mocks = Mocks()
+        val mockIdentityModel = mockk<IdentityModel>()
+        val mockIdentityModelStore = mockk<IdentityModelStore>()
+        val recovery = RecoverFromDroppedLoginBug(mocks.operationRepo, mockIdentityModelStore, mocks.configModelStore)
+
+        mocks.operationRepo.start()
+        recovery.start()
+
+        verify {
+            mocks.operationRepo.subscribe(recovery)
+        }
+    }
 }) {
     companion object {
         private fun mockOperation(