mirror of
https://github.com/mihonapp/mihon.git
synced 2024-10-24 21:12:35 -04:00
3fd9e021fa
* Use custom QueryPagingSource - Adds placeholder to make the list jump around less - Fixes issue where SQLDelight QueryPagingSource would throw IndexOutOfBounds * Review Changes
90 lines
3.1 KiB
Kotlin
90 lines
3.1 KiB
Kotlin
package eu.kanade.data
|
|
|
|
import androidx.paging.PagingSource
|
|
import com.squareup.sqldelight.Query
|
|
import com.squareup.sqldelight.db.SqlDriver
|
|
import com.squareup.sqldelight.runtime.coroutines.asFlow
|
|
import com.squareup.sqldelight.runtime.coroutines.mapToList
|
|
import com.squareup.sqldelight.runtime.coroutines.mapToOne
|
|
import com.squareup.sqldelight.runtime.coroutines.mapToOneOrNull
|
|
import eu.kanade.tachiyomi.Database
|
|
import kotlinx.coroutines.CoroutineDispatcher
|
|
import kotlinx.coroutines.Dispatchers
|
|
import kotlinx.coroutines.flow.Flow
|
|
import kotlinx.coroutines.withContext
|
|
|
|
class AndroidDatabaseHandler(
|
|
val db: Database,
|
|
private val driver: SqlDriver,
|
|
val queryDispatcher: CoroutineDispatcher = Dispatchers.IO,
|
|
val transactionDispatcher: CoroutineDispatcher = queryDispatcher,
|
|
) : DatabaseHandler {
|
|
|
|
val suspendingTransactionId = ThreadLocal<Int>()
|
|
|
|
override suspend fun <T> await(inTransaction: Boolean, block: suspend Database.() -> T): T {
|
|
return dispatch(inTransaction, block)
|
|
}
|
|
|
|
override suspend fun <T : Any> awaitList(
|
|
inTransaction: Boolean,
|
|
block: suspend Database.() -> Query<T>,
|
|
): List<T> {
|
|
return dispatch(inTransaction) { block(db).executeAsList() }
|
|
}
|
|
|
|
override suspend fun <T : Any> awaitOne(
|
|
inTransaction: Boolean,
|
|
block: suspend Database.() -> Query<T>,
|
|
): T {
|
|
return dispatch(inTransaction) { block(db).executeAsOne() }
|
|
}
|
|
|
|
override suspend fun <T : Any> awaitOneOrNull(
|
|
inTransaction: Boolean,
|
|
block: suspend Database.() -> Query<T>,
|
|
): T? {
|
|
return dispatch(inTransaction) { block(db).executeAsOneOrNull() }
|
|
}
|
|
|
|
override fun <T : Any> subscribeToList(block: Database.() -> Query<T>): Flow<List<T>> {
|
|
return block(db).asFlow().mapToList(queryDispatcher)
|
|
}
|
|
|
|
override fun <T : Any> subscribeToOne(block: Database.() -> Query<T>): Flow<T> {
|
|
return block(db).asFlow().mapToOne(queryDispatcher)
|
|
}
|
|
|
|
override fun <T : Any> subscribeToOneOrNull(block: Database.() -> Query<T>): Flow<T?> {
|
|
return block(db).asFlow().mapToOneOrNull(queryDispatcher)
|
|
}
|
|
|
|
override fun <T : Any> subscribeToPagingSource(
|
|
countQuery: Database.() -> Query<Long>,
|
|
queryProvider: Database.(Long, Long) -> Query<T>,
|
|
): PagingSource<Long, T> {
|
|
return QueryPagingSource(
|
|
handler = this,
|
|
countQuery = countQuery,
|
|
queryProvider = { limit, offset ->
|
|
queryProvider.invoke(db, limit, offset)
|
|
},
|
|
)
|
|
}
|
|
|
|
private suspend fun <T> dispatch(inTransaction: Boolean, block: suspend Database.() -> T): T {
|
|
// Create a transaction if needed and run the calling block inside it.
|
|
if (inTransaction) {
|
|
return withTransaction { block(db) }
|
|
}
|
|
|
|
// If we're currently in the transaction thread, there's no need to dispatch our query.
|
|
if (driver.currentTransaction() != null) {
|
|
return block(db)
|
|
}
|
|
|
|
// Get the current database context and run the calling block.
|
|
val context = getCurrentDatabaseContext()
|
|
return withContext(context) { block(db) }
|
|
}
|
|
}
|