From 425bc857cfe094d8b9be5815030ef3e6b0cf173d Mon Sep 17 00:00:00 2001
From: Alex Ning <chineseperson5@gmail.com>
Date: Wed, 7 Aug 2019 23:28:02 +0800
Subject: Refactored all the other classes to support multi user. Clearing the
 app data is required before launching the app.

---
 .../main/java/SubredditDatabase/SubredditViewModel.java | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

(limited to 'app/src/main/java/SubredditDatabase/SubredditViewModel.java')

diff --git a/app/src/main/java/SubredditDatabase/SubredditViewModel.java b/app/src/main/java/SubredditDatabase/SubredditViewModel.java
index 8899abdd..dc0247d2 100644
--- a/app/src/main/java/SubredditDatabase/SubredditViewModel.java
+++ b/app/src/main/java/SubredditDatabase/SubredditViewModel.java
@@ -7,13 +7,15 @@ import androidx.lifecycle.ViewModel;
 import androidx.lifecycle.ViewModelProvider;
 import androidx.annotation.NonNull;
 
+import ml.docilealligator.infinityforreddit.RedditDataRoomDatabase;
+
 public class SubredditViewModel extends AndroidViewModel {
     private SubredditRepository mSubredditRepository;
     private LiveData<SubredditData> mSubredditLiveData;
 
-    public SubredditViewModel(Application application, String id) {
+    public SubredditViewModel(Application application, RedditDataRoomDatabase redditDataRoomDatabase, String id) {
         super(application);
-        mSubredditRepository = new SubredditRepository(application, id);
+        mSubredditRepository = new SubredditRepository(redditDataRoomDatabase, id);
         mSubredditLiveData = mSubredditRepository.getSubredditLiveData();
     }
 
@@ -29,19 +31,20 @@ public class SubredditViewModel extends AndroidViewModel {
 
         @NonNull
         private final Application mApplication;
+        private final RedditDataRoomDatabase mRedditDataRoomDatabase;
+        private final String mSubredditName;
 
-        private final String subredditName;
-
-        public Factory(@NonNull Application application, String subredditName) {
+        public Factory(@NonNull Application application, RedditDataRoomDatabase redditDataRoomDatabase, String subredditname) {
             mApplication = application;
-            this.subredditName = subredditName;
+            mRedditDataRoomDatabase = redditDataRoomDatabase;
+            mSubredditName = subredditname;
         }
 
         @NonNull
         @Override
         public <T extends ViewModel> T create(@NonNull Class<T> modelClass) {
             //noinspection unchecked
-            return (T) new SubredditViewModel(mApplication, subredditName);
+            return (T) new SubredditViewModel(mApplication, mRedditDataRoomDatabase, mSubredditName);
         }
     }
 }
-- 
cgit v1.2.3