- ViewModels moved to Hilt, Cleaned up dagger code.

WaheedNazir · Jun 19, 2021 · acbf2f3 · acbf2f3
1 parent dc46dca
commit acbf2f3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/app/src/main/java/com/kotlin/mvvm/ui/countries/CountriesViewModel.kt b/app/src/main/java/com/kotlin/mvvm/ui/countries/CountriesViewModel.kt
@@ -6,16 +6,17 @@ import androidx.lifecycle.ViewModel
 import androidx.lifecycle.viewModelScope
 import com.kotlin.mvvm.repository.model.countries.Country
 import com.kotlin.mvvm.repository.repo.countries.CountriesRepository
+import dagger.hilt.android.lifecycle.HiltViewModel
 import kotlinx.coroutines.launch
 import javax.inject.Inject
 
 /**
  * Created by Waheed on 08,November,2019
- */
-
-/**
+ *
  * A container for [Country] related data to show on the UI.
  */
+
+@HiltViewModel
 class CountriesViewModel @Inject constructor(val countriesRepository: CountriesRepository) :
     ViewModel() {
 

diff --git a/app/src/main/java/com/kotlin/mvvm/ui/news/NewsViewModel.kt b/app/src/main/java/com/kotlin/mvvm/ui/news/NewsViewModel.kt
@@ -5,6 +5,7 @@ import androidx.lifecycle.ViewModel
 import com.kotlin.mvvm.repository.api.network.Resource
 import com.kotlin.mvvm.repository.model.news.News
 import com.kotlin.mvvm.repository.repo.news.NewsRepository
+import dagger.hilt.android.lifecycle.HiltViewModel
 import javax.inject.Inject
 
 /**
@@ -14,9 +15,9 @@ import javax.inject.Inject
 /**
  * A container for [News] related data to show on the UI.
  */
-class NewsViewModel @Inject constructor(
-    private val newsRepository: NewsRepository
-) : ViewModel() {
+
+@HiltViewModel
+class NewsViewModel @Inject constructor(private val newsRepository: NewsRepository) : ViewModel() {
 
     /**
      * Loading news articles from internet and database