diff --git a/src/main/java/com/commafeed/backend/model/FeedSubscription.java b/src/main/java/com/commafeed/backend/model/FeedSubscription.java index 1874af6a..e311f7ce 100644 --- a/src/main/java/com/commafeed/backend/model/FeedSubscription.java +++ b/src/main/java/com/commafeed/backend/model/FeedSubscription.java @@ -1,9 +1,13 @@ package com.commafeed.backend.model; +import java.util.Set; + +import javax.persistence.CascadeType; import javax.persistence.Column; import javax.persistence.Entity; import javax.persistence.JoinColumn; import javax.persistence.ManyToOne; +import javax.persistence.OneToMany; import javax.persistence.Table; @Entity @@ -25,6 +29,9 @@ public class FeedSubscription extends AbstractModel { @ManyToOne private FeedCategory category; + @OneToMany(mappedBy = "subscription", cascade = CascadeType.REMOVE) + private Set statuses; + public Feed getFeed() { return feed; } @@ -57,4 +64,12 @@ public class FeedSubscription extends AbstractModel { this.category = category; } + public Set getStatuses() { + return statuses; + } + + public void setStatuses(Set statuses) { + this.statuses = statuses; + } + }