Add runTransaction methods for jvm, ios and js

Author: michaelprichardson

firebase-database/src/androidMain/kotlin/dev/gitlive/firebase/database/database.kt

@@ -5,26 +5,27 @@
 package dev.gitlive.firebase.database
 
 import com.google.android.gms.tasks.Task
-import com.google.firebase.database.ChildEventListener
-import com.google.firebase.database.Logger
+import com.google.firebase.database.*
 import com.google.firebase.database.ServerValue
-import com.google.firebase.database.ValueEventListener
 import dev.gitlive.firebase.Firebase
 import dev.gitlive.firebase.FirebaseApp
 import dev.gitlive.firebase.database.ChildEvent.Type
 import dev.gitlive.firebase.decode
 import dev.gitlive.firebase.safeOffer
+import kotlinx.coroutines.CompletableDeferred
 import kotlinx.coroutines.FlowPreview
 import kotlinx.coroutines.channels.awaitClose
 import kotlinx.coroutines.coroutineScope
 import kotlinx.coroutines.flow.Flow
 import kotlinx.coroutines.flow.callbackFlow
 import kotlinx.coroutines.flow.filter
 import kotlinx.coroutines.flow.produceIn
+import kotlinx.coroutines.runBlocking
 import kotlinx.coroutines.selects.select
 import kotlinx.coroutines.tasks.asDeferred
 import kotlinx.coroutines.tasks.await
 import kotlinx.serialization.DeserializationStrategy
+import kotlinx.serialization.KSerializer
 import kotlinx.serialization.SerializationStrategy
 
 @PublishedApi
@@ -188,8 +189,30 @@ actual class DatabaseReference internal constructor(
     actual suspend fun removeValue() = android.removeValue()
         .run { if(persistenceEnabled) await() else awaitWhileOnline() }
         .run { Unit }
-}
 
+    actual suspend fun <T> runTransaction(strategy: KSerializer<T>, transactionUpdate: (currentData: T) -> T): DataSnapshot {
+        val deferred = CompletableDeferred<Result<DataSnapshot>>()
+        android.runTransaction(object : Transaction.Handler {
+
+            override fun doTransaction(currentData: MutableData) =
+                Transaction.success(transactionUpdate(decode(strategy, currentData)) as MutableData)
+
+            override fun onComplete(
+                error: DatabaseError?,
+                committed: Boolean,
+                snapshot: com.google.firebase.database.DataSnapshot?
+            ) {
+                if (error == null && snapshot != null) {
+                    deferred.complete(Result.success(DataSnapshot(snapshot)))
+                } else {
+                    deferred.complete(Result.failure(Throwable(error?.message)))
+                }
+            }
+
+        })
+        return deferred.await().getOrThrow()
+    }
+}
 @Suppress("UNCHECKED_CAST")
 actual class DataSnapshot internal constructor(val android: com.google.firebase.database.DataSnapshot) {
 

firebase-database/src/iosMain/kotlin/dev/gitlive/firebase/database/database.kt

@@ -22,6 +22,7 @@ import kotlinx.coroutines.flow.filter
 import kotlinx.coroutines.flow.produceIn
 import kotlinx.coroutines.selects.select
 import kotlinx.serialization.DeserializationStrategy
+import kotlinx.serialization.KSerializer
 import kotlinx.serialization.SerializationStrategy
 import platform.Foundation.*
 import kotlin.collections.component1
@@ -156,6 +157,24 @@ actual class DatabaseReference internal constructor(
     actual suspend fun removeValue() {
         ios.await(persistenceEnabled) { removeValueWithCompletionBlock(it) }
     }
+
+    actual suspend fun <T> runTransaction(strategy: KSerializer<T>, transactionUpdate: (currentData: T) -> T): DataSnapshot {
+        val deferred = CompletableDeferred<Result<DataSnapshot>>()
+        ios.runTransactionBlock(
+            block = { firMutableData ->
+                FIRTransactionResult.successWithValue(transactionUpdate(decode(strategy, firMutableData)) as FIRMutableData)
+            },
+            andCompletionBlock = { error, _, snapshot ->
+                if (error == null) {
+                    deferred.complete(Result.success(DataSnapshot(snapshot!!)))
+                } else {
+                    deferred.complete(Result.failure(Throwable(error.localizedDescription)))
+                }
+            },
+            withLocalEvents = false
+        )
+        return deferred.await().getOrThrow()
+    }
 }
 
 @Suppress("UNCHECKED_CAST")

firebase-database/src/jsMain/kotlin/dev/gitlive/firebase/database/database.kt

@@ -12,6 +12,7 @@ import kotlinx.coroutines.flow.filter
 import kotlinx.coroutines.flow.produceIn
 import kotlinx.coroutines.selects.select
 import kotlinx.serialization.DeserializationStrategy
+import kotlinx.serialization.KSerializer
 import kotlinx.serialization.SerializationStrategy
 import kotlin.js.Promise
 
@@ -127,6 +128,23 @@ actual class DatabaseReference internal constructor(override val js: firebase.da
 
     actual suspend fun <T> setValue(strategy: SerializationStrategy<T>, value: T, encodeDefaults: Boolean) =
         rethrow { js.set(encode(strategy, value, encodeDefaults)).awaitWhileOnline() }
+
+    actual suspend fun <T> runTransaction(strategy: KSerializer<T>, transactionUpdate: (currentData: T) -> T): DataSnapshot {
+        val deferred = CompletableDeferred<Result<DataSnapshot>>()
+        js.runTransaction(
+            transactionUpdate,
+            { error, _, snapshot ->
+                if (error != null) {
+                    deferred.complete(Result.success(DataSnapshot(snapshot!!)))
+                } else {
+                    deferred.complete(Result.failure(Throwable(error?.message)))
+                }
+            },
+            applyLocally = false
+        ).await()
+        return deferred.await().getOrThrow()
+    }
+
 }
 
 actual class DataSnapshot internal constructor(val js: firebase.database.DataSnapshot) {