1 files changed, 17 insertions, 2 deletions
diff --git a/app/src/main/java/SubscribedSubredditDatabase/SubscribedSubredditData.java b/app/src/main/java/SubscribedSubredditDatabase/SubscribedSubredditData.java
index 4c6f1a49..bf2a87b8 100644
--- a/app/src/main/java/SubscribedSubredditDatabase/SubscribedSubredditData.java
+++ b/app/src/main/java/SubscribedSubredditDatabase/SubscribedSubredditData.java
@@ -2,10 +2,14 @@ package SubscribedSubredditDatabase;
 
 import androidx.room.ColumnInfo;
 import androidx.room.Entity;
+import androidx.room.ForeignKey;
 import androidx.room.PrimaryKey;
 import androidx.annotation.NonNull;
 
-@Entity(tableName = "subscribed_subreddits")
+import Account.Account;
+
+@Entity(tableName = "subscribed_subreddits", foreignKeys = @ForeignKey(entity = Account.class, parentColumns = "username",
+        childColumns = "username", onDelete = ForeignKey.CASCADE))
 public class SubscribedSubredditData {
 
     @PrimaryKey
@@ -16,11 +20,14 @@ public class SubscribedSubredditData {
     private String name;
     @ColumnInfo(name = "icon")
     private String iconUrl;
+    @ColumnInfo(name = "username")
+    private String username;
 
-    public SubscribedSubredditData(@NonNull String id, String name, String iconUrl) {
+    public SubscribedSubredditData(@NonNull String id, String name, String iconUrl, String username) {
         this.id = id;
         this.name = name;
         this.iconUrl = iconUrl;
+        this.username = username;
     }
 
     @NonNull
@@ -35,4 +42,12 @@ public class SubscribedSubredditData {
     public String getIconUrl() {
         return iconUrl;
     }
+
+    public void setUsername(String username) {
+        this.username = username;
+    }
+
+    public String getUsername() {
+        return username;
+    }
 }