From 5fc9567cfa6b6beee4f52a9f835f304865d693e1 Mon Sep 17 00:00:00 2001
From: cloudroam <cloudroam>
Date: 星期四, 10 四月 2025 18:13:58 +0800
Subject: [PATCH] add: 消息提醒后版本

---
 app/src/main/java/com/example/firstapp/database/repository/ReminderRepository.kt |   31 +++++++++++++++++++------------
 1 files changed, 19 insertions(+), 12 deletions(-)

diff --git a/app/src/main/java/com/example/firstapp/database/repository/ReminderRepository.kt b/app/src/main/java/com/example/firstapp/database/repository/ReminderRepository.kt
index 20f90f4..861e344 100644
--- a/app/src/main/java/com/example/firstapp/database/repository/ReminderRepository.kt
+++ b/app/src/main/java/com/example/firstapp/database/repository/ReminderRepository.kt
@@ -1,26 +1,33 @@
 package com.example.firstapp.database.repository
 
+import android.content.Context
+import com.example.firstapp.AppDatabase
 import com.example.firstapp.database.dao.ReminderDao
-import androidx.annotation.WorkerThread
 import com.example.firstapp.database.entity.Reminder
 import kotlinx.coroutines.flow.Flow
 
-class ReminderRepository(private val reminderDao: ReminderDao) {
 
-    val allReminders: Flow<List<Reminder>> = reminderDao.getAllReminders()
+class ReminderRepository(context: Context) {
+        private val reminderDao: ReminderDao = AppDatabase.getInstance(context).reminderDao()
+    fun getAllReminders(): Flow<List<Reminder>> = reminderDao.getAllReminders()
 
-    @WorkerThread
-     fun insert(reminder: Reminder) {
-        reminderDao.insert(reminder)
+    suspend fun insertReminder(reminder: Reminder) {
+        reminderDao.insertReminder(reminder)
     }
 
-    @WorkerThread
-     fun delete(reminder: Reminder) {
-        reminderDao.delete(reminder)
+    suspend fun deleteReminder(reminder: Reminder) {
+        reminderDao.deleteReminder(reminder)
     }
 
-    @WorkerThread
-    fun getByType(type: String): List<Reminder> {
-        return reminderDao.getByType(type)
+    suspend fun deleteReminderByCategoryId(categoryId: Int) {
+        reminderDao.deleteReminderByCategoryId(categoryId)
+    }
+
+    suspend fun getReminderByCategoryId(categoryId: Int): Reminder? {
+        return reminderDao.getReminderByCategoryId(categoryId)
+    }
+
+    suspend fun deleteAllReminders() {
+        reminderDao.deleteAllReminders()
     }
 } 
\ No newline at end of file

--
Gitblit v1.9.3