InsanusMokrassar/MicroUtils
1
0
Fork 0
mirror of https://github.com/InsanusMokrassar/MicroUtils.git synced 2024-06-26 11:37:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

one more time update mechanism of sql transactions in android

Browse Source
This commit is contained in:
InsanusMokrassar 2020-11-22 22:00:50 +06:00
parent 3de5558ed4
commit 615f7f99c3
1 changed files with 22 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
repos/common/src/main/kotlin/dev/inmo/micro_utils/repos/DatabaseTransactions.kt
View File

@ -2,33 +2,36 @@ package dev.inmo.micro_utils.repos
import android.database.sqlite.SQLiteDatabase
import dev.inmo.micro_utils.coroutines.safely
import kotlinx.coroutines.newSingleThreadContext
import kotlinx.coroutines.withContext
import kotlinx.coroutines.*
import java.util.concurrent.Executors
import kotlin.coroutines.CoroutineContext
import kotlin.coroutines.coroutineContext
object InTransaction: CoroutineContext.Element, CoroutineContext.Key<InTransaction> {
override val key: CoroutineContext.Key<InTransaction> = InTransaction
class TransactionContext(
val databaseContext: CoroutineContext
): CoroutineContext.Element {
override val key: CoroutineContext.Key<TransactionContext> = TransactionContext
companion object : CoroutineContext.Key<TransactionContext>
}
suspend fun <T> SQLiteDatabase.transaction(block: suspend SQLiteDatabase.() -> T): T {
return when {
coroutineContext[InTransaction] == InTransaction -> {
return coroutineContext[TransactionContext] ?.let {
withContext(it.databaseContext) {
block()
}
else -> {
withContext(InTransaction) {
beginTransaction()
safely(
{
endTransaction()
throw it
}
) {
block().also {
setTransactionSuccessful()
endTransaction()
}
} ?: Executors.newSingleThreadExecutor().asCoroutineDispatcher().let { context ->
withContext(TransactionContext(context) + context) {
beginTransaction()
safely(
{
endTransaction()
throw it
}
) {
block().also {
setTransactionSuccessful()
endTransaction()
}
}
}