From 3433f3e9a0254b24cec2836b112267b1af3f9101 Mon Sep 17 00:00:00 2001
From: tj <1378534974@qq.com>
Date: 星期一, 07 四月 2025 10:57:40 +0800
Subject: [PATCH] 1.jks打包相关修改 2.vip续费页面相关修改

---
 app/src/main/java/com/example/firstapp/ui/dashboard/DashboardViewModel.kt |   47 +++++++++++++++++++++++++++++++----------------
 1 files changed, 31 insertions(+), 16 deletions(-)

diff --git a/app/src/main/java/com/example/firstapp/ui/dashboard/DashboardViewModel.kt b/app/src/main/java/com/example/firstapp/ui/dashboard/DashboardViewModel.kt
index dae2211..1a0a2cb 100644
--- a/app/src/main/java/com/example/firstapp/ui/dashboard/DashboardViewModel.kt
+++ b/app/src/main/java/com/example/firstapp/ui/dashboard/DashboardViewModel.kt
@@ -4,32 +4,47 @@
 import androidx.lifecycle.AndroidViewModel
 import androidx.lifecycle.asLiveData
 import androidx.lifecycle.viewModelScope
+import androidx.room.RewriteQueriesToDropUnusedColumns
 import com.example.firstapp.AppDatabase
-import com.example.firstapp.model.PackageInfo
-import com.example.firstapp.repository.PackageRepository
+import com.example.firstapp.database.entity.Code
+import com.example.firstapp.database.repository.CodeRepository
 import kotlinx.coroutines.launch
 
 class DashboardViewModel(application: Application) : AndroidViewModel(application) {
-    
-    private val repository: PackageRepository
-    
+
+    private val repository: CodeRepository
+
     init {
-        val packageDao = AppDatabase.getInstance(application).packageDao()
-        repository = PackageRepository(packageDao)
+        val codeDao = AppDatabase.getInstance(application).codeDao()
+        repository = CodeRepository(codeDao)
     }
-    
-    fun getPackages(date: Long, dateType: String) = 
+
+    fun getPackages(date: Long, dateType: String) =
         repository.getPackages(date, dateType).asLiveData()
 
-    fun getCourierStats(date: Long) = repository.getCourierStats(date).asLiveData()
+    fun getCourierStats(date: Long, dateType: String) = 
+        repository.getCourierStats(date, dateType).asLiveData()
+
+    fun getYearlyHeatmap(date: Long) = repository.getYearlyHeatmap(date).asLiveData()
+
+
+    fun getWeeklyStats(date: Long, weekCount: Int = 6) = 
+        repository.getWeeklyStats(date, weekCount).asLiveData()
     
-    fun getDailyStats(date: Long) = repository.getDailyStats(date).asLiveData()
+    fun getMonthlyStats(date: Long) = repository.getMonthlyStats(date).asLiveData()
+    
+    fun getYearlyStats(date: Long) = repository.getYearlyStats(date).asLiveData()
 
-    fun updatePackageStatus(packageInfo: PackageInfo) = viewModelScope.launch {
-        repository.update(packageInfo)
+    fun getYearMonthlyStats(date: Long) = 
+        repository.getYearMonthlyStats(date).asLiveData()
+
+    fun insert(code: Code) = viewModelScope.launch {
+        repository.insert(code)
     }
 
-    fun insert(packageInfo: PackageInfo) = viewModelScope.launch {
-        repository.insert(packageInfo)
-    }
+    fun getCurrentWeekStats(date: Long) =
+        repository.getCurrentWeekStats(date).asLiveData()
+
+    fun getCurrentYearStats(date: Long) =
+        repository.getCurrentYearStats(date).asLiveData()
 }
\ No newline at end of file

--
Gitblit v1.9.3