From 7b415d1019795fa39d9c9727c3123e638a021ffa Mon Sep 17 00:00:00 2001 From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?= Date: Thu, 14 Oct 2010 12:15:31 +0200 Subject: [PATCH] Store replies in post. --- .../java/net/pterodactylus/sone/data/Post.java | 41 ++++++++++++++++++++++ 1 file changed, 41 insertions(+) diff --git a/src/main/java/net/pterodactylus/sone/data/Post.java b/src/main/java/net/pterodactylus/sone/data/Post.java index fc8a125..a42cfaa 100644 --- a/src/main/java/net/pterodactylus/sone/data/Post.java +++ b/src/main/java/net/pterodactylus/sone/data/Post.java @@ -17,6 +17,10 @@ package net.pterodactylus.sone.data; +import java.util.Collection; +import java.util.Collections; +import java.util.HashSet; +import java.util.Set; import java.util.UUID; /** @@ -39,6 +43,9 @@ public class Post { /** The text of the post. */ private final String text; + /** The replies that have been loaded for this post. */ + private final Set replies = new HashSet(); + /** * Creates a new post. * @@ -124,6 +131,40 @@ public class Post { return text; } + /** + * Returns all replies to this post in unspecified order. + * + * @return All replies to this post + */ + public Collection getReplies() { + return Collections.unmodifiableSet(replies); + } + + /** + * Adds a reply to this post. The reply will not be added if its + * {@link Reply#getPost() post} is not equal to this post. + * + * @param reply + * The reply to add + */ + public void addReply(Reply reply) { + if (reply.getPost().equals(this)) { + replies.add(reply); + } + } + + /** + * Removes a reply from this post. + * + * @param reply + * The reply to remove + */ + public void removeReply(Reply reply) { + if (reply.getPost().equals(this)) { + replies.remove(reply); + } + } + // // OBJECT METHODS // -- 2.7.4