Use RywData everywhere we were previously using String (rywToken)

Motivation: update all usages of `rywToken` to now make use of the new `RywData` object. In `ConsistencyManager` we also renamed the method `setRywToken` to `setRywData`
OneSignal · Oct 25, 2024 · bb4a0b7 · bb4a0b7
1 parent 8b79aac
commit bb4a0b7
Show file tree

Hide file tree

Showing 18 changed files with 132 additions and 89 deletions.
diff --git a/...K/onesignal/core/src/main/java/com/onesignal/common/consistency/IamFetchReadyCondition.kt b/...K/onesignal/core/src/main/java/com/onesignal/common/consistency/IamFetchReadyCondition.kt
@@ -20,7 +20,7 @@ class IamFetchReadyCondition(
     override val id: String
         get() = ID
 
-    override fun isMet(indexedTokens: Map<String, Map<IConsistencyKeyEnum, String>>): Boolean {
+    override fun isMet(indexedTokens: Map<String, Map<IConsistencyKeyEnum, RywData>>): Boolean {
         val tokenMap = indexedTokens[key] ?: return false
         val userUpdateTokenSet = tokenMap[IamFetchRywTokenKey.USER] != null
 
@@ -33,9 +33,13 @@ class IamFetchReadyCondition(
         return userUpdateTokenSet
     }
 
-    override fun getNewestToken(indexedTokens: Map<String, Map<IConsistencyKeyEnum, String?>>): String? {
+    override fun getNewestToken(indexedTokens: Map<String, Map<IConsistencyKeyEnum, RywData?>>): RywData? {
         val tokenMap = indexedTokens[key] ?: return null
-        // maxOrNull compares lexicographically
-        return listOfNotNull(tokenMap[IamFetchRywTokenKey.USER], tokenMap[IamFetchRywTokenKey.SUBSCRIPTION]).maxOrNull()
+
+        // Collect non-null RywData objects and find the one with the largest rywToken lexicographically
+        return listOfNotNull(
+            tokenMap[IamFetchRywTokenKey.USER],
+            tokenMap[IamFetchRywTokenKey.SUBSCRIPTION],
+        ).maxByOrNull { it.rywToken ?: "" }
     }
 }
diff --git a/.../onesignal/core/src/main/java/com/onesignal/common/consistency/impl/ConsistencyManager.kt b/.../onesignal/core/src/main/java/com/onesignal/common/consistency/impl/ConsistencyManager.kt
@@ -1,5 +1,6 @@
 package com.onesignal.common.consistency.impl
 
+import com.onesignal.common.consistency.RywData
 import com.onesignal.common.consistency.models.ICondition
 import com.onesignal.common.consistency.models.IConsistencyKeyEnum
 import com.onesignal.common.consistency.models.IConsistencyManager
@@ -18,8 +19,8 @@ import kotlinx.coroutines.sync.withLock
  */
 class ConsistencyManager : IConsistencyManager {
     private val mutex = Mutex()
-    private val indexedTokens: MutableMap<String, MutableMap<IConsistencyKeyEnum, String>> = mutableMapOf()
-    private val conditions: MutableList<Pair<ICondition, CompletableDeferred<String?>>> =
+    private val indexedTokens: MutableMap<String, MutableMap<IConsistencyKeyEnum, RywData>> = mutableMapOf()
+    private val conditions: MutableList<Pair<ICondition, CompletableDeferred<RywData?>>> =
         mutableListOf()
 
     /**
@@ -29,10 +30,10 @@ class ConsistencyManager : IConsistencyManager {
      *      key: K - corresponds to the operation for which we have a read-your-write token
      *      value: String? - the token (read-your-write token)
      */
-    override suspend fun setRywToken(
+    override suspend fun setRywData(
         id: String,
         key: IConsistencyKeyEnum,
-        value: String,
+        value: RywData,
     ) {
         mutex.withLock {
             val rywTokens = indexedTokens.getOrPut(id) { mutableMapOf() }
@@ -46,7 +47,7 @@ class ConsistencyManager : IConsistencyManager {
      */
     override suspend fun getRywDataFromAwaitableCondition(condition: ICondition): CompletableDeferred<RywData?> {
         mutex.withLock {
-            val deferred = CompletableDeferred<String?>()
+            val deferred = CompletableDeferred<RywData?>()
             val pair = Pair(condition, deferred)
             conditions.add(pair)
             checkConditionsAndComplete()
@@ -55,7 +56,7 @@ class ConsistencyManager : IConsistencyManager {
     }
 
     override suspend fun resolveConditionsWithID(id: String) {
-        val completedConditions = mutableListOf<Pair<ICondition, CompletableDeferred<String?>>>()
+        val completedConditions = mutableListOf<Pair<ICondition, CompletableDeferred<RywData?>>>()
 
         for ((condition, deferred) in conditions) {
             if (condition.id == id) {
@@ -74,13 +75,13 @@ class ConsistencyManager : IConsistencyManager {
      * IMPORTANT: calling code should be protected by mutex to avoid potential inconsistencies
      */
     private fun checkConditionsAndComplete() {
-        val completedConditions = mutableListOf<Pair<ICondition, CompletableDeferred<String?>>>()
+        val completedConditions = mutableListOf<Pair<ICondition, CompletableDeferred<RywData?>>>()
 
         for ((condition, deferred) in conditions) {
             if (condition.isMet(indexedTokens)) {
-                val newestToken = condition.getNewestToken(indexedTokens)
+                val rywDataForNewestToken = condition.getNewestToken(indexedTokens)
                 if (!deferred.isCompleted) {
-                    deferred.complete(newestToken)
+                    deferred.complete(rywDataForNewestToken)
                 }
                 completedConditions.add(Pair(condition, deferred))
             }

diff --git a/...nalSDK/onesignal/core/src/main/java/com/onesignal/common/consistency/models/ICondition.kt b/...nalSDK/onesignal/core/src/main/java/com/onesignal/common/consistency/models/ICondition.kt
@@ -1,5 +1,7 @@
 package com.onesignal.common.consistency.models
 
+import com.onesignal.common.consistency.RywData
+
 interface ICondition {
     /**
      * Every implementation should define a unique ID & make available via a companion object for
@@ -11,11 +13,11 @@ interface ICondition {
      * Define a condition that "unblocks" execution
      * e.g. we have token (A && B) || A
      */
-    fun isMet(indexedTokens: Map<String, Map<IConsistencyKeyEnum, String>>): Boolean
+    fun isMet(indexedTokens: Map<String, Map<IConsistencyKeyEnum, RywData>>): Boolean
 
     /**
      * Used to process tokens according to their format & return the newest token.
      * e.g. numeric strings would be compared differently from JWT tokens
      */
-    fun getNewestToken(indexedTokens: Map<String, Map<IConsistencyKeyEnum, String?>>): String?
+    fun getNewestToken(indexedTokens: Map<String, Map<IConsistencyKeyEnum, RywData?>>): RywData?
 }
diff --git a/...esignal/core/src/main/java/com/onesignal/common/consistency/models/IConsistencyManager.kt b/...esignal/core/src/main/java/com/onesignal/common/consistency/models/IConsistencyManager.kt
@@ -1,5 +1,6 @@
 package com.onesignal.common.consistency.models
 
+import com.onesignal.common.consistency.RywData
 import kotlinx.coroutines.CompletableDeferred
 
 interface IConsistencyManager {
@@ -10,10 +11,10 @@ interface IConsistencyManager {
      *  key: IConsistencyKeyEnum - corresponds to the operation for which we have a read-your-write token
      *  value: String? - the read-your-write token
      */
-    suspend fun setRywToken(
+    suspend fun setRywData(
         id: String,
         key: IConsistencyKeyEnum,
-        value: String,
+        value: RywData,
     )
 
     /**
@@ -22,7 +23,7 @@ interface IConsistencyManager {
      *  condition: ICondition - the condition to be registered
      * Returns: CompletableDeferred<String?> - a deferred action that completes when the condition is met
      */
-    suspend fun getRywDataFromAwaitableCondition(condition: ICondition): CompletableDeferred<String?>
+    suspend fun getRywDataFromAwaitableCondition(condition: ICondition): CompletableDeferred<RywData?>
 
     /**
      * Resolve all conditions with a specific ID

diff --git a/...nal/core/src/main/java/com/onesignal/user/internal/backend/ISubscriptionBackendService.kt b/...nal/core/src/main/java/com/onesignal/user/internal/backend/ISubscriptionBackendService.kt
@@ -1,5 +1,6 @@
 package com.onesignal.user.internal.backend
 
+import com.onesignal.common.consistency.RywData
 import com.onesignal.common.exceptions.BackendException
 
 interface ISubscriptionBackendService {
@@ -21,7 +22,7 @@ interface ISubscriptionBackendService {
         aliasLabel: String,
         aliasValue: String,
         subscription: SubscriptionObject,
-    ): Pair<String, String?>?
+    ): Pair<String, RywData>?
 
     /**
      * Update an existing subscription with the properties provided.
@@ -34,7 +35,7 @@ interface ISubscriptionBackendService {
         appId: String,
         subscriptionId: String,
         subscription: SubscriptionObject,
-    ): String?
+    ): RywData
 
     /**
      * Delete an existing subscription.

diff --git a/...K/onesignal/core/src/main/java/com/onesignal/user/internal/backend/IUserBackendService.kt b/...K/onesignal/core/src/main/java/com/onesignal/user/internal/backend/IUserBackendService.kt
@@ -1,5 +1,6 @@
 package com.onesignal.user.internal.backend
 
+import com.onesignal.common.consistency.RywData
 import com.onesignal.common.exceptions.BackendException
 
 interface IUserBackendService {
@@ -47,7 +48,7 @@ interface IUserBackendService {
         properties: PropertiesObject,
         refreshDeviceMetadata: Boolean,
         propertyiesDelta: PropertiesDeltasObject,
-    ): String?
+    ): RywData
 
     /**
      * Retrieve a user from the backend.

diff --git a/...core/src/main/java/com/onesignal/user/internal/backend/impl/SubscriptionBackendService.kt b/...core/src/main/java/com/onesignal/user/internal/backend/impl/SubscriptionBackendService.kt
@@ -1,5 +1,6 @@
 package com.onesignal.user.internal.backend.impl
 
+import com.onesignal.common.consistency.RywData
 import com.onesignal.common.exceptions.BackendException
 import com.onesignal.common.safeJSONObject
 import com.onesignal.common.toMap
@@ -16,7 +17,7 @@ internal class SubscriptionBackendService(
         aliasLabel: String,
         aliasValue: String,
         subscription: SubscriptionObject,
-    ): Pair<String, String?>? {
+    ): Pair<String, RywData>? {
         val jsonSubscription = JSONConverter.convertToJSON(subscription)
         jsonSubscription.remove("id")
         val requestJSON = JSONObject().put("subscription", jsonSubscription)
@@ -38,14 +39,21 @@ internal class SubscriptionBackendService(
             rywToken = responseJSON.getString("ryw_token")
         }
 
-        return Pair(subscriptionJSON.getString("id"), rywToken)
+        var rywDelay: Long? = null
+        if (responseJSON.has("ryw_delay")) {
+            rywDelay = responseJSON.getLong("ryw_delay")
+        }
+
+        var rywData = RywData(rywToken, rywDelay)
+
+        return Pair(subscriptionJSON.getString("id"), rywData)
     }
 
     override suspend fun updateSubscription(
         appId: String,
         subscriptionId: String,
         subscription: SubscriptionObject,
-    ): String? {
+    ): RywData {
         val requestJSON =
             JSONObject()
                 .put("subscription", JSONConverter.convertToJSON(subscription))
@@ -56,12 +64,18 @@ internal class SubscriptionBackendService(
             throw BackendException(response.statusCode, response.payload, response.retryAfterSeconds)
         }
 
-        val responseBody = JSONObject(response.payload)
-        return if (responseBody.has("ryw_token")) {
-            responseBody.getString("ryw_token")
-        } else {
-            null
+        val responseJSON = JSONObject(response.payload)
+        var rywToken: String? = null
+        if (responseJSON.has("ryw_token")) {
+            rywToken = responseJSON.getString("ryw_token")
+        }
+
+        var rywDelay: Long? = null
+        if (responseJSON.has("ryw_delay")) {
+            rywDelay = responseJSON.getLong("ryw_delay")
         }
+
+        return RywData(rywToken, rywDelay)
     }
 
     override suspend fun deleteSubscription(

diff --git a/...esignal/core/src/main/java/com/onesignal/user/internal/backend/impl/UserBackendService.kt b/...esignal/core/src/main/java/com/onesignal/user/internal/backend/impl/UserBackendService.kt
@@ -1,5 +1,6 @@
 package com.onesignal.user.internal.backend.impl
 
+import com.onesignal.common.consistency.RywData
 import com.onesignal.common.exceptions.BackendException
 import com.onesignal.common.putMap
 import com.onesignal.core.internal.http.IHttpClient
@@ -52,7 +53,7 @@ internal class UserBackendService(
         properties: PropertiesObject,
         refreshDeviceMetadata: Boolean,
         propertyiesDelta: PropertiesDeltasObject,
-    ): String? {
+    ): RywData {
         val jsonObject =
             JSONObject()
                 .put("refresh_device_metadata", refreshDeviceMetadata)
@@ -71,12 +72,18 @@ internal class UserBackendService(
             throw BackendException(response.statusCode, response.payload, response.retryAfterSeconds)
         }
 
-        val responseBody = JSONObject(response.payload)
-        return if (responseBody.has("ryw_token")) {
-            responseBody.getString("ryw_token")
-        } else {
-            null
+        val responseJSON = JSONObject(response.payload)
+        var rywToken: String? = null
+        if (responseJSON.has("ryw_token")) {
+            rywToken = responseJSON.getString("ryw_token")
         }
+
+        var rywDelay: Long? = null
+        if (responseJSON.has("ryw_delay")) {
+            rywDelay = responseJSON.getLong("ryw_delay")
+        }
+
+        return RywData(rywToken, rywDelay)
     }
 
     override suspend fun getUser(

diff --git a/...va/com/onesignal/user/internal/operations/impl/executors/SubscriptionOperationExecutor.kt b/...va/com/onesignal/user/internal/operations/impl/executors/SubscriptionOperationExecutor.kt
@@ -114,10 +114,10 @@ internal class SubscriptionOperationExecutor(
                 ) ?: return ExecutionResponse(ExecutionResult.SUCCESS)
 
             val backendSubscriptionId = result.first
-            val rywToken = result.second
+            val rywData = result.second
 
-            if (rywToken != null) {
-                _consistencyManager.setRywToken(createOperation.onesignalId, IamFetchRywTokenKey.SUBSCRIPTION, rywToken)
+            if (rywData.rywToken != null) {
+                _consistencyManager.setRywData(createOperation.onesignalId, IamFetchRywTokenKey.SUBSCRIPTION, rywData)
             } else {
                 _consistencyManager.resolveConditionsWithID(IamFetchReadyCondition.ID)
             }
@@ -188,10 +188,10 @@ internal class SubscriptionOperationExecutor(
                     AndroidUtils.getAppVersion(_applicationService.appContext),
                 )
 
-            val rywToken = _subscriptionBackend.updateSubscription(lastOperation.appId, lastOperation.subscriptionId, subscription)
+            val rywData = _subscriptionBackend.updateSubscription(lastOperation.appId, lastOperation.subscriptionId, subscription)
 
-            if (rywToken != null) {
-                _consistencyManager.setRywToken(startingOperation.onesignalId, IamFetchRywTokenKey.SUBSCRIPTION, rywToken)
+            if (rywData.rywToken != null) {
+                _consistencyManager.setRywData(startingOperation.onesignalId, IamFetchRywTokenKey.SUBSCRIPTION, rywData)
             } else {
                 _consistencyManager.resolveConditionsWithID(IamFetchReadyCondition.ID)
             }

diff --git a/...java/com/onesignal/user/internal/operations/impl/executors/UpdateUserOperationExecutor.kt b/...java/com/onesignal/user/internal/operations/impl/executors/UpdateUserOperationExecutor.kt
@@ -138,7 +138,7 @@ internal class UpdateUserOperationExecutor(
 
         if (appId != null && onesignalId != null) {
             try {
-                val rywToken =
+                val rywData =
                     _userBackend.updateUser(
                         appId,
                         IdentityConstants.ONESIGNAL_ID,
@@ -148,8 +148,8 @@ internal class UpdateUserOperationExecutor(
                         deltasObject,
                     )
 
-                if (rywToken != null) {
-                    _consistencyManager.setRywToken(onesignalId, IamFetchRywTokenKey.USER, rywToken)
+                if (rywData.rywToken != null) {
+                    _consistencyManager.setRywData(onesignalId, IamFetchRywTokenKey.USER, rywData)
                 } else {
                     _consistencyManager.resolveConditionsWithID(IamFetchReadyCondition.ID)
                 }