empathicqubit/Signal-Android
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Add extra transaction protections.

Browse source
This commit is contained in:
Greyson Parrelli 2024-09-12 14:44:09 -04:00 • committed by Cody Henthorne
parent 0d37013481
commit 5c6644d1a1
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
core-util/src/main/java/org/signal/core/util/SQLiteDatabaseExtensions.kt
View file

@ -17,12 +17,16 @@ inline fun <T : SupportSQLiteDatabase, R> T.withinTransaction(block: (T) -> R):
beginTransaction() beginTransaction()
try { try {
val toReturn = block(this) val toReturn = block(this)
if (inTransaction()) {
setTransactionSuccessful() setTransactionSuccessful()
}
return toReturn return toReturn
} finally { } finally {
if (inTransaction()) {
endTransaction() endTransaction()
} }
} }
}
fun SupportSQLiteDatabase.getTableRowCount(table: String): Int { fun SupportSQLiteDatabase.getTableRowCount(table: String): Int {
return this.query("SELECT COUNT(*) FROM $table").use { return this.query("SELECT COUNT(*) FROM $table").use {