Заготовка

2026-05-07 23:27:37 +03:00
parent 2d2e4b685e
commit 9daa786cfb
33 changed files with 1662 additions and 178 deletions
@@ -0,0 +1,70 @@
+package chats.data.local.dao
+
+import androidx.room.*
+import chats.data.local.database.MessageEntity
+import chats.domain.model.MessageStatus
+import kotlinx.coroutines.flow.Flow
+
+/**
+ * DAO для операций с сообщениями в Room Database
+ */
+@Dao
+interface MessageDao {
+    
+    @Query("SELECT * FROM messages WHERE chatId = :chatId ORDER BY sequenceId ASC")
+    fun getMessagesByChatId(chatId: String): Flow<List<MessageEntity>>
+    
+    @Query("SELECT * FROM messages WHERE chatId = :chatId AND sequenceId > :afterSequenceId ORDER BY sequenceId ASC LIMIT :limit")
+    suspend fun getMessagesAfter(chatId: String, afterSequenceId: Long, limit: Int): List<MessageEntity>
+    
+    @Query("SELECT * FROM messages WHERE chatId = :chatId AND sequenceId < :beforeSequenceId ORDER BY sequenceId DESC LIMIT :limit")
+    suspend fun getMessagesBefore(chatId: String, beforeSequenceId: Long, limit: Int): List<MessageEntity>
+    
+    @Query("SELECT * FROM messages WHERE localId = :localId")
+    suspend fun getMessageByLocalId(localId: String): MessageEntity?
+    
+    @Query("SELECT * FROM messages WHERE serverId = :serverId")
+    suspend fun getMessageByServerId(serverId: String): MessageEntity?
+    
+    @Query("SELECT * FROM messages WHERE status IN (:statuses) AND chatId = :chatId")
+    suspend fun getMessagesByStatus(chatId: String, statuses: List<MessageStatus>): List<MessageEntity>
+    
+    @Query("SELECT * FROM messages WHERE status = :status")
+    fun getMessagesByStatusFlow(status: MessageStatus): Flow<List<MessageEntity>>
+    
+    @Query("SELECT * FROM messages WHERE status IN (:statuses)")
+    suspend fun getPendingMessages(statuses: List<MessageStatus>): List<MessageEntity>
+    
+    @Insert(onConflict = OnConflictStrategy.REPLACE)
+    suspend fun insertMessage(message: MessageEntity)
+    
+    @Insert(onConflict = OnConflictStrategy.REPLACE)
+    suspend fun insertMessages(messages: List<MessageEntity>)
+    
+    @Update
+    suspend fun updateMessage(message: MessageEntity)
+    
+    @Delete
+    suspend fun deleteMessage(message: MessageEntity)
+    
+    @Query("DELETE FROM messages WHERE localId = :localId")
+    suspend fun deleteMessageByLocalId(localId: String)
+    
+    @Query("DELETE FROM messages WHERE chatId = :chatId")
+    suspend fun deleteMessagesByChatId(chatId: String)
+    
+    @Query("UPDATE messages SET status = :status, updatedAtMillis = :updatedAtMillis WHERE localId = :localId")
+    suspend fun updateMessageStatus(localId: String, status: MessageStatus, updatedAtMillis: Long)
+    
+    @Query("UPDATE messages SET serverId = :serverId, status = :status, updatedAtMillis = :updatedAtMillis WHERE localId = :localId")
+    suspend fun updateMessageWithServerId(localId: String, serverId: String, status: MessageStatus, updatedAtMillis: Long)
+    
+    @Query("UPDATE messages SET status = :status, errorMessage = :errorMessage, retryCount = retryCount + 1, updatedAtMillis = :updatedAtMillis WHERE localId = :localId")
+    suspend fun updateMessageError(localId: String, status: MessageStatus, errorMessage: String, updatedAtMillis: Long)
+    
+    @Query("SELECT COUNT(*) FROM messages WHERE chatId = :chatId AND status IN (:statuses)")
+    suspend fun getUnsentCount(chatId: String, statuses: List<MessageStatus>): Int
+    
+    @Query("SELECT * FROM messages WHERE chatId = :chatId AND status = :status LIMIT 1")
+    suspend fun getFirstMessageByStatus(chatId: String, status: MessageStatus): MessageEntity?
+}