X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FReplyGroupFilter.java;h=eff8c7f381c1f6511f65726482c68307a3d487e6;hb=2e03e9dddbea4b81aacaf1aa316f5c3ccffd4bf9;hp=bcf73d8ab97371b881479b36bf533e0b2481cb18;hpb=8f02544d31f323ae9053dd9a11a99eacd8cf5bcd;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/template/ReplyGroupFilter.java b/src/main/java/net/pterodactylus/sone/template/ReplyGroupFilter.java
index bcf73d8..eff8c7f 100644
--- a/src/main/java/net/pterodactylus/sone/template/ReplyGroupFilter.java
+++ b/src/main/java/net/pterodactylus/sone/template/ReplyGroupFilter.java
@@ -21,6 +21,7 @@ import java.util.HashMap;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
+import java.util.Map.Entry;
 import java.util.Set;
 
 import net.pterodactylus.sone.data.Post;
@@ -40,9 +41,6 @@ import com.google.common.base.Optional;
  */
 public class ReplyGroupFilter implements Filter {
 
-	/**
-	 * {@inheritDoc}
-	 */
 	@Override
 	public Object format(TemplateContext templateContext, Object data, Map<String, Object> parameters) {
 		@SuppressWarnings("unchecked")
@@ -69,14 +67,14 @@ public class ReplyGroupFilter implements Filter {
 			replies.add(reply);
 		}
 		Map<Post, Map<String, Set<?>>> result = new HashMap<Post, Map<String, Set<?>>>();
-		for (Post post : postSones.keySet()) {
-			if (result.containsKey(post)) {
+		for (Entry<Post, Set<Sone>> postEntry : postSones.entrySet()) {
+			if (result.containsKey(postEntry.getKey())) {
 				continue;
 			}
 			Map<String, Set<?>> postResult = new HashMap<String, Set<?>>();
-			postResult.put("sones", postSones.get(post));
-			postResult.put("replies", postReplies.get(post));
-			result.put(post, postResult);
+			postResult.put("sones", postEntry.getValue());
+			postResult.put("replies", postReplies.get(postEntry.getKey()));
+			result.put(postEntry.getKey(), postResult);
 		}
 		return result;
 	}