From 897ffe5e29ab022d75ad948ecf894e0a3ed3b2f5 Mon Sep 17 00:00:00 2001
From: cloudroam <cloudroam>
Date: 星期二, 04 三月 2025 18:01:41 +0800
Subject: [PATCH] fix: 1

---
 app/src/main/java/com/example/firstapp/database/repository/CodeRepository.kt |   34 ++++++++++++++++++++++++++++++++++
 1 files changed, 34 insertions(+), 0 deletions(-)

diff --git a/app/src/main/java/com/example/firstapp/database/repository/CodeRepository.kt b/app/src/main/java/com/example/firstapp/database/repository/CodeRepository.kt
index 2d9eceb..7c14c1e 100644
--- a/app/src/main/java/com/example/firstapp/database/repository/CodeRepository.kt
+++ b/app/src/main/java/com/example/firstapp/database/repository/CodeRepository.kt
@@ -3,6 +3,9 @@
 import androidx.annotation.WorkerThread
 import com.example.firstapp.database.dao.CodeDao
 import com.example.firstapp.database.entity.Code
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.flow.Flow
+import kotlinx.coroutines.withContext
 
 
 class CodeRepository(private val codeDao: CodeDao) {
@@ -17,5 +20,36 @@
 
     fun getAllDesc() = codeDao.getAllCodesDesc()
 
+    @WorkerThread
+    fun getByKeyword(keyword: String): List<Code> {
+        return codeDao.getByKeyword(keyword)
+    }
+
+    fun queryByTypeAndCodeAndDate(content: String, code: String, dateString: String): Code {
+        return codeDao.queryByTypeAndCodeAndDate(content, code, dateString)
+    }
+
+    @WorkerThread
+    fun pickup(id: Long) = codeDao.pickup(id)
+
+
+    fun getPackages(date: Long, dateType: String): Flow<List<Code>> {
+        return when (dateType) {
+            "DAY" -> codeDao.getPackagesByDay(date)
+            "WEEK" -> codeDao.getPackagesByWeek(date)
+            else -> codeDao.getPackagesByDay(date)
+        }
+    }
+
+    fun getCourierStats(date: Long) = codeDao.getCourierStatsByWeek(date)
+
+    fun getDailyStats() = codeDao.getDailyStatsByWeek()
+
+
+    @WorkerThread
+    fun getPackagesByDay(date: Long): List<Code> {
+        return codeDao.getNewPackagesByDay(date)
+    }
+
 
 }
\ No newline at end of file

--
Gitblit v1.9.3