Override local unspent outputs when remote index conflicts

Author: SeniorZhai

app/src/main/java/one/mixin/android/db/web3/WalletOutputDao.kt

@@ -1,6 +1,7 @@
 package one.mixin.android.db.web3
 
 import androidx.room.Dao
+import androidx.room.ColumnInfo
 import androidx.room.Query
 import androidx.room.Transaction
 import one.mixin.android.api.response.web3.WalletOutput
@@ -39,8 +40,27 @@ interface WalletOutputDao: BaseDao<WalletOutput> {
     @Query("UPDATE outputs SET status = 'signed' WHERE output_id IN (:outputIds)")
     suspend fun updateOutputsToSigned(outputIds: List<String>): Int
 
+    @Query("SELECT output_id, transaction_hash, output_index FROM outputs WHERE address = :address AND asset_id = :assetId AND status = 'unspent' AND output_index IN (:outputIndexes)")
+    suspend fun findLocalUnspentOutputsByIndexes(address: String, assetId: String, outputIndexes: List<Long>): List<LocalUnspentOutputIndex>
+
     @Transaction
     suspend fun mergeOutputsForAddress(address: String, assetId: String, remoteOutputs: List<WalletOutput>) {
+        val remoteOutputIndexes: List<Long> = remoteOutputs.map { it.outputIndex }.distinct()
+        if (remoteOutputIndexes.isNotEmpty()) {
+            val remoteTxHashesByIndex: Map<Long, Set<String>> = remoteOutputs
+                .groupBy { it.outputIndex }
+                .mapValues { entry -> entry.value.map { it.transactionHash }.toSet() }
+            val localUnspentByIndex: List<LocalUnspentOutputIndex> = findLocalUnspentOutputsByIndexes(address, assetId, remoteOutputIndexes)
+            val conflictOutputIds: List<String> = localUnspentByIndex
+                .filter { local ->
+                    val remoteTxHashes: Set<String> = remoteTxHashesByIndex[local.outputIndex] ?: emptySet()
+                    !remoteTxHashes.contains(local.transactionHash)
+                }
+                .map { it.outputId }
+            if (conflictOutputIds.isNotEmpty()) {
+                updateOutputsToSigned(conflictOutputIds)
+            }
+        }
         val remoteOutputIds: List<String> = remoteOutputs.map { it.outputId }
         val signedOutputIds: List<String> = if (remoteOutputIds.isEmpty()) emptyList() else findSignedOutputIds(remoteOutputIds)
         val outputsToInsert: List<WalletOutput> = if (signedOutputIds.isEmpty()) {
@@ -72,3 +92,12 @@ interface WalletOutputDao: BaseDao<WalletOutput> {
         return total
     }
 }
+
+data class LocalUnspentOutputIndex(
+    @ColumnInfo(name = "output_id")
+    val outputId: String,
+    @ColumnInfo(name = "transaction_hash")
+    val transactionHash: String,
+    @ColumnInfo(name = "output_index")
+    val outputIndex: Long,
+)