Opt/multi database

app/src/main/java/one/mixin/android/MixinApplication.kt

@@ -32,6 +32,7 @@ import com.mapbox.maps.loader.MapboxMapsInitializer
 import dagger.hilt.EntryPoint
 import dagger.hilt.InstallIn
 import dagger.hilt.android.EntryPointAccessors
+import dagger.hilt.android.internal.managers.ApplicationComponentManager
 import dagger.hilt.components.SingletonComponent
 import io.reactivex.plugins.RxJavaPlugins
 import io.sentry.SentryLevel
@@ -50,6 +51,7 @@ import one.mixin.android.crypto.MixinSignalProtocolLogger
 import one.mixin.android.crypto.PrivacyPreference.clearPrivacyPreferences
 import one.mixin.android.crypto.removeValueFromEncryptedPreferences
 import one.mixin.android.crypto.db.SignalDatabase
+import one.mixin.android.db.DatabaseProvider
 import one.mixin.android.db.MixinDatabase
 import one.mixin.android.di.AppModule.API_UA
 import one.mixin.android.di.ApplicationScope
@@ -94,6 +96,7 @@ import java.util.concurrent.atomic.AtomicBoolean
 import javax.inject.Inject
 import kotlin.system.exitProcess
 
+@SuppressLint("UnsafeOptInUsageError")
 open class MixinApplication :
     Application(),
     Application.ActivityLifecycleCallbacks,
@@ -118,11 +121,9 @@ open class MixinApplication :
         fun getHiltWorkerFactory(): HiltWorkerFactory
     }
 
-    @InstallIn(SingletonComponent::class)
-    @EntryPoint
-    interface AppEntryPoint {
-        fun inject(app: MixinApplication)
-    }
+
+    @Inject
+    lateinit var databaseProvider: DatabaseProvider
 
     private fun getWorkerFactory() = EntryPointAccessors.fromApplication(this, HiltWorkerFactoryEntryPoint::class.java).getHiltWorkerFactory()
 
@@ -317,26 +318,11 @@ open class MixinApplication :
             applicationScope.launch {
                 clearData(sessionId)
                 withContext(Dispatchers.Main) {
-                    val entryPoint =
-                        EntryPointAccessors.fromApplication(
-                            this@MixinApplication,
-                            AppEntryPoint::class.java,
-                        )
-                    entryPoint.inject(this@MixinApplication)
                     LandingActivity.show(this@MixinApplication)
                 }
             }
         }
-    }
-
-    fun reject() {
-        MixinDatabase.destroy()
-        val entryPoint =
-            EntryPointAccessors.fromApplication(
-                this@MixinApplication,
-                AppEntryPoint::class.java,
-            )
-        entryPoint.inject(this@MixinApplication)
+        databaseProvider.closeAllDatabases()
     }
 
     private fun clearData(sessionId: String?) {

app/src/main/java/one/mixin/android/crypto/db/SignalDatabase.kt

@@ -12,6 +12,7 @@ import one.mixin.android.crypto.vo.RatchetSenderKey
 import one.mixin.android.crypto.vo.SenderKey
 import one.mixin.android.crypto.vo.Session
 import one.mixin.android.crypto.vo.SignedPreKey
+import one.mixin.android.db.MixinDatabase
 
 @Database(
     entities = [
@@ -67,4 +68,9 @@ abstract class SignalDatabase : RoomDatabase() {
             object : RoomDatabase.Callback() {
             }
     }
+
+    override fun close() {
+        INSTANCE = null
+        super.close()
+    }
 }

app/src/main/java/one/mixin/android/db/CircleDao.kt

@@ -38,7 +38,6 @@ interface CircleDao : BaseDao<Circle> {
         }
     }
 
-
     @Query(
         """
         SELECT c.* FROM circle_conversations cc

app/src/main/java/one/mixin/android/db/DaoExtension.kt

@@ -1,5 +1,6 @@
 package one.mixin.android.db
 
+import androidx.room.Transaction
 import kotlinx.coroutines.withContext
 import one.mixin.android.db.flow.MessageFlow
 import one.mixin.android.db.pending.PendingDatabase
@@ -31,26 +32,25 @@ fun JobDao.insertNoReplace(job: Job) {
     }
 }
 
+@Transaction
 suspend fun OutputDao.insertUnspentOutputs(outputs: List<Output>) =
     withContext(SINGLE_DB_THREAD) {
-        runInTransaction {
-            val signed = findSignedOutput(outputs.map { it.outputId })
-            if (signed.isEmpty()) {
-                insertList(outputs)
-            } else {
-                Timber.e("Insert filter ${signed.joinToString(", ") }")
-                // Exclude signed data
-                val unsignedData = outputs.filterNot { signed.contains(it.outputId) }
-                insertList(unsignedData)
-            }
+        val signed = findSignedOutput(outputs.map { it.outputId })
+        if (signed.isEmpty()) {
+            insertList(outputs)
+        } else {
+            Timber.e("Insert filter ${signed.joinToString(", ")}")
+            // Exclude signed data
+            val unsignedData = outputs.filterNot { signed.contains(it.outputId) }
+            insertList(unsignedData)
         }
     }
 
 // Delete SQL
 fun MixinDatabase.deleteMessageById(messageId: String) {
     runInTransaction {
         pinMessageDao().deleteByMessageId(messageId)
-        mentionMessageDao().deleteMessage(messageId)
+        messageMentionDao().deleteMessage(messageId)
         messageDao().deleteMessageById(messageId)
         remoteMessageStatusDao().deleteByMessageId(messageId)
         expiredMessageDao().deleteByMessageId(messageId)
@@ -63,7 +63,7 @@ fun MixinDatabase.deleteMessageById(
 ) {
     runInTransaction {
         pinMessageDao().deleteByMessageId(messageId)
-        mentionMessageDao().deleteMessage(messageId)
+        messageMentionDao().deleteMessage(messageId)
         messageDao().deleteMessageById(messageId)
         conversationExtDao().decrement(conversationId)
         remoteMessageStatusDao().deleteByMessageId(messageId)
@@ -75,7 +75,7 @@ fun MixinDatabase.deleteMessageById(
 fun MixinDatabase.deleteMessageByIds(messageIds: List<String>) {
     runInTransaction {
         pinMessageDao().deleteByIds(messageIds)
-        mentionMessageDao().deleteMessage(messageIds)
+        messageMentionDao().deleteMessage(messageIds)
         messageDao().deleteMessageById(messageIds)
         remoteMessageStatusDao().deleteByMessageIds(messageIds)
         expiredMessageDao().deleteByMessageId(messageIds)
@@ -143,4 +143,4 @@ fun MixinDatabase.insertMessage(message: Message) {
     messageDao().insert(message)
     conversationExtDao().increment(message.conversationId)
     conversationDao().updateLastMessageId(message.messageId, message.createdAt, message.conversationId)
-}
+}

app/src/main/java/one/mixin/android/db/DatabaseProvider.kt

@@ -0,0 +1,80 @@
+package one.mixin.android.db
+
+import android.content.Context
+import one.mixin.android.crypto.db.SignalDatabase
+import one.mixin.android.fts.FtsDatabase
+import one.mixin.android.db.pending.PendingDatabase
+import one.mixin.android.db.pending.PendingDatabaseImp
+import one.mixin.android.session.Session
+import javax.inject.Inject
+
+class DatabaseProvider @Inject constructor(
+    private val context: Context,
+) {
+    private var mixinDatabase: MixinDatabase? = null
+    private var ftsDatabase: FtsDatabase? = null
+    private var pendingDatabase: PendingDatabaseImp? = null
+    private var identityNumber: String? = null
+
+    @Synchronized
+    fun getMixinDatabase(): MixinDatabase {
+        if (mixinDatabase == null) {
+            val db = MixinDatabase.getDatabase(context)
+            mixinDatabase = db
+        }
+        return mixinDatabase ?: throw IllegalStateException("MixinDatabase is not initialized")
+    }
+
+    @Synchronized
+    fun getFtsDatabase(): FtsDatabase {
+        if (ftsDatabase == null) {
+            val db = FtsDatabase.getDatabase(context)
+            ftsDatabase = db
+        }
+        return ftsDatabase ?: throw IllegalStateException("FtsDatabase is not initialized")
+    }
+
+    @Synchronized
+    fun getPendingDatabase(needCreate: Boolean = false): PendingDatabase {
+        if (pendingDatabase == null) {
+            val db = getMixinDatabase()
+            pendingDatabase = PendingDatabaseImp.getDatabase(context, db.floodMessageDao(), db.jobDao())
+        }
+        return pendingDatabase ?: throw IllegalStateException("PendingDatabase is not initialized")
+    }
+
+    @Synchronized
+    fun initAllDatabases() {
+        val identityNumber = requireNotNull(Session.getAccount()?.identityNumber)
+        if (identityNumber == this.identityNumber && isInit()) return
+
+        this.identityNumber = identityNumber
+
+        mixinDatabase?.close()
+        val db = MixinDatabase.getDatabase(context)
+        mixinDatabase = db
+
+        ftsDatabase?.close()
+        ftsDatabase = FtsDatabase.getDatabase(context)
+
+        pendingDatabase?.close()
+        pendingDatabase = PendingDatabaseImp.getDatabase(context, db.floodMessageDao(), db.jobDao())
+    }
+
+    @Synchronized
+    fun closeAllDatabases() {
+        mixinDatabase?.close()
+        mixinDatabase = null
+
+        ftsDatabase?.close()
+        ftsDatabase = null
+
+        pendingDatabase?.close()
+        pendingDatabase = null
+
+        identityNumber = null
+    }
+
+    @Synchronized
+    fun isInit() = mixinDatabase != null && ftsDatabase != null && pendingDatabase != null
+}

app/src/main/java/one/mixin/android/db/DepositDao.kt

@@ -2,6 +2,7 @@ package one.mixin.android.db
 
 import androidx.room.Dao
 import androidx.room.Query
+import androidx.room.Transaction
 import one.mixin.android.vo.safe.DepositEntry
 import one.mixin.android.vo.safe.DestinationTag
 
@@ -15,4 +16,10 @@ interface DepositDao : BaseDao<DepositEntry> {
 
     @Query("DELETE FROM deposit_entries WHERE chain_id=:chainId")
     fun deleteByChainId(chainId: String)
+
+    @Transaction
+    fun deleteAndInsert(chainId: String, list: List<DepositEntry>) {
+        deleteByChainId(chainId)
+        insertList(list)
+    }
 }

app/src/main/java/one/mixin/android/db/MarketCapRankDao.kt

@@ -3,6 +3,7 @@ package one.mixin.android.db
 import androidx.room.Dao
 import androidx.room.Query
 import androidx.room.Transaction
+import one.mixin.android.vo.market.Market
 import one.mixin.android.vo.market.MarketCapRank
 
 @Dao

app/src/main/java/one/mixin/android/db/MixinDatabase.kt

@@ -77,6 +77,7 @@ import one.mixin.android.db.converter.WithdrawalMemoPossibilityConverter
 import one.mixin.android.ui.wallet.alert.vo.Alert
 import one.mixin.android.util.GsonHelper
 import one.mixin.android.util.SINGLE_DB_EXECUTOR
+import one.mixin.android.util.database.dbDir
 import one.mixin.android.util.debug.getContent
 import one.mixin.android.util.reportException
 import one.mixin.android.vo.Address
@@ -127,6 +128,7 @@ import one.mixin.android.vo.safe.RawTransaction
 import one.mixin.android.vo.safe.SafeSnapshot
 import one.mixin.android.vo.safe.Token
 import one.mixin.android.vo.safe.TokensExtra
+import java.io.File
 import java.util.concurrent.Executors
 import kotlin.math.max
 import kotlin.math.min
@@ -235,7 +237,7 @@ abstract class MixinDatabase : RoomDatabase() {
 
     abstract fun favoriteAppDao(): FavoriteAppDao
 
-    abstract fun mentionMessageDao(): MessageMentionDao
+    abstract fun messageMentionDao(): MessageMentionDao
 
     abstract fun circleDao(): CircleDao
 
@@ -277,23 +279,28 @@ abstract class MixinDatabase : RoomDatabase() {
 
     abstract fun marketCapRankDao(): MarketCapRankDao
 
+    override fun close() {
+        INSTANCE = null
+        super.close()
+    }
+
     companion object {
         private var INSTANCE: MixinDatabase? = null
-
-        private val lock = Any()
         private var supportSQLiteDatabase: SupportSQLiteDatabase? = null
+        private var NAME: String? = null
 
-        fun destroy() {
-            INSTANCE = null
+        fun getCurrentDbName(): String? {
+            return NAME
         }
 
         @Suppress("UNUSED_ANONYMOUS_PARAMETER")
         @SuppressLint("RestrictedApi")
         fun getDatabase(context: Context): MixinDatabase {
-            synchronized(lock) {
                 if (INSTANCE == null) {
+                    val dir = dbDir(context)
+                    NAME = dir.name
                     val builder =
-                        Room.databaseBuilder(context, MixinDatabase::class.java, DB_NAME)
+                        Room.databaseBuilder(context, MixinDatabase::class.java, File(dir, DB_NAME).absolutePath)
                             .openHelperFactory(
                                 MixinOpenHelperFactory(
                                     FrameworkSQLiteOpenHelperFactory(),
@@ -384,7 +391,6 @@ abstract class MixinDatabase : RoomDatabase() {
                     INSTANCE = builder.build()
                 }
                 return INSTANCE as MixinDatabase
-            }
         }
 
         fun query(query: String): String? {
@@ -411,7 +417,12 @@ abstract class MixinDatabase : RoomDatabase() {
         }
 
         fun checkPoint() {
-            supportSQLiteDatabase?.query("PRAGMA wal_checkpoint(FULL)")?.close()
+            supportSQLiteDatabase?.let { db ->
+                db.beginTransaction()
+                db.query("PRAGMA wal_checkpoint(FULL)")?.close()
+                db.setTransactionSuccessful()
+                db.endTransaction()
+            }
         }
 
         fun getWritableDatabase(): SupportSQLiteDatabase? {
@@ -431,12 +442,4 @@ abstract class MixinDatabase : RoomDatabase() {
                 }
             }
     }
-}
-
-fun runInTransaction(block: () -> Unit) {
-    MixinDatabase.getDatabase(MixinApplication.appContext).runInTransaction(block)
-}
-
-suspend fun withTransaction(block: suspend () -> Unit) {
-    MixinDatabase.getDatabase(MixinApplication.appContext).withTransaction(block)
-}
+}

app/src/main/java/one/mixin/android/db/StickerAlbumDao.kt

@@ -67,4 +67,5 @@ interface StickerAlbumDao : BaseDao<StickerAlbum> {
 
     @Query("SELECT max(ordered_at) FROM sticker_albums")
     suspend fun findMaxOrder(): Int?
+
 }

app/src/main/java/one/mixin/android/db/StickerDao.kt

@@ -20,7 +20,6 @@ interface StickerDao : BaseDao<Sticker> {
         insert(s)
     }
 
-
     @Query("SELECT * FROM stickers WHERE last_use_at > 0 ORDER BY last_use_at DESC LIMIT 20")
     fun recentUsedStickers(): LiveData<List<Sticker>>