From cf9bc941487df8dfb0780d3e8f1281f4e397b5fa Mon Sep 17 00:00:00 2001
From: cloudroam <cloudroam>
Date: 星期四, 06 三月 2025 10:33:57 +0800
Subject: [PATCH] fix: 3

---
 app/src/main/java/com/example/firstapp/database/repository/CodeRepository.kt |   26 ++++++++++++++++++++++----
 1 files changed, 22 insertions(+), 4 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 f187d22..915c13d 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,9 +3,7 @@
 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) {
@@ -41,9 +39,19 @@
         }
     }
 
-    fun getCourierStats(date: Long) = codeDao.getCourierStatsByWeek(date)
+    fun getCourierStats(date: Long, dateType: String) = when(dateType) {
+        "WEEK" -> codeDao.getCourierStatsByWeek(date)
+        "MONTH" -> codeDao.getCourierStatsByMonth(date)
+        "YEAR" -> codeDao.getCourierStatsByYear(date)
+        else -> codeDao.getCourierStatsByWeek(date)
+    }
 
-    fun getDailyStats(date: Long) = codeDao.getDailyStatsByWeek(date)
+//    fun getStats(date: Long, dateType: String) = when(dateType) {
+//        "WEEK" -> codeDao.getDailyStatsByWeek(date)
+//        "MONTH" -> codeDao.getMonthlyStats(date)
+//        "YEAR" -> codeDao.getYearlyStats(date)
+//        else -> codeDao.getDailyStatsByWeek(date)
+//    }
 
 
     @WorkerThread
@@ -51,5 +59,15 @@
         return codeDao.getNewPackagesByDay(date)
     }
 
+    fun getYearlyHeatmap(date: Long) = codeDao.getYearlyHeatmap(date)
+
+    fun getWeeklyStats(date: Long, weekCount: Int) = codeDao.getWeeklyStats(date, weekCount)
+    
+    fun getMonthlyStats(date: Long) = codeDao.getMonthlyStats(date)
+
+
+    fun getYearlyStats(date: Long) = codeDao.getYearlyStats(date)
+
+    fun getYearMonthlyStats(date: Long) = codeDao.getYearMonthlyStats(date)
 
 }
\ No newline at end of file

--
Gitblit v1.9.3