X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdatabase%2Fmemory%2FMemoryDatabase.java;h=00bb5c94ebce934c2224220c714aad47dfd8bec2;hp=0b2c5a21c14869ec2f3545c0f4847bf69ba60b61;hb=667a3a2d8544cf91f00baba1aaf9f55cead7d006;hpb=b1057c2e5f77a74a3d64c5ad9ff6b0c5894c4a73

diff --git a/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java b/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
index 0b2c5a2..00bb5c9 100644
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
@@ -19,12 +19,15 @@ package net.pterodactylus.sone.database.memory;
 
 import static com.google.common.base.Optional.fromNullable;
 import static com.google.common.base.Preconditions.checkNotNull;
+import static com.google.common.base.Predicates.not;
 import static com.google.common.collect.FluentIterable.from;
+import static java.util.Collections.unmodifiableCollection;
 import static net.pterodactylus.sone.data.Reply.TIME_COMPARATOR;
+import static net.pterodactylus.sone.data.Sone.LOCAL_SONE_FILTER;
+import static net.pterodactylus.sone.data.Sone.toAllAlbums;
+import static net.pterodactylus.sone.data.Sone.toAllImages;
 
-import java.util.ArrayList;
 import java.util.Collection;
-import java.util.Collections;
 import java.util.Comparator;
 import java.util.HashMap;
 import java.util.HashSet;
@@ -79,6 +82,8 @@ public class MemoryDatabase extends AbstractService implements Database {
 	/** The configuration. */
 	private final Configuration configuration;
 
+	private final Map<String, Sone> allSones = new HashMap<String, Sone>();
+
 	/** All posts by their ID. */
 	private final Map<String, Post> allPosts = new HashMap<String, Post>();
 
@@ -100,15 +105,6 @@ public class MemoryDatabase extends AbstractService implements Database {
 		}
 	}, TIME_COMPARATOR);
 
-	/** Replies by post. */
-	private final SortedSetMultimap<String, PostReply> postReplies = TreeMultimap.create(new Comparator<String>() {
-
-		@Override
-		public int compare(String leftString, String rightString) {
-			return leftString.compareTo(rightString);
-		}
-	}, TIME_COMPARATOR);
-
 	/** Whether post replies are known. */
 	private final Set<String> knownPostReplies = new HashSet<String>();
 
@@ -171,6 +167,93 @@ public class MemoryDatabase extends AbstractService implements Database {
 		}
 	}
 
+	@Override
+	public void storeSone(Sone sone) {
+		lock.writeLock().lock();
+		try {
+			Collection<Post> removedPosts = sonePosts.removeAll(sone.getId());
+			for (Post removedPost : removedPosts) {
+				allPosts.remove(removedPost.getId());
+			}
+			Collection<PostReply> removedPostReplies =
+					sonePostReplies.removeAll(sone.getId());
+			for (PostReply removedPostReply : removedPostReplies) {
+				allPostReplies.remove(removedPostReply.getId());
+			}
+			Collection<Album> removedAlbums =
+					soneAlbums.removeAll(sone.getId());
+			for (Album removedAlbum : removedAlbums) {
+				allAlbums.remove(removedAlbum.getId());
+			}
+			Collection<Image> removedImages =
+					soneImages.removeAll(sone.getId());
+			for (Image removedImage : removedImages) {
+				allImages.remove(removedImage.getId());
+			}
+
+			allSones.put(sone.getId(), sone);
+			sonePosts.putAll(sone.getId(), sone.getPosts());
+			for (Post post : sone.getPosts()) {
+				allPosts.put(post.getId(), post);
+			}
+			sonePostReplies.putAll(sone.getId(), sone.getReplies());
+			for (PostReply postReply : sone.getReplies()) {
+				allPostReplies.put(postReply.getId(), postReply);
+			}
+			soneAlbums.putAll(sone.getId(), toAllAlbums.apply(sone));
+			for (Album album : toAllAlbums.apply(sone)) {
+				allAlbums.put(album.getId(), album);
+			}
+			soneImages.putAll(sone.getId(), toAllImages.apply(sone));
+			for (Image image : toAllImages.apply(sone)) {
+				allImages.put(image.getId(), image);
+			}
+		} finally {
+			lock.writeLock().unlock();
+		}
+	}
+
+	@Override
+	public Optional<Sone> getSone(String soneId) {
+		lock.readLock().lock();
+		try {
+			return fromNullable(allSones.get(soneId));
+		} finally {
+			lock.readLock().unlock();
+		}
+	}
+
+	@Override
+	public Collection<Sone> getSones() {
+		lock.readLock().lock();
+		try {
+			return unmodifiableCollection(allSones.values());
+		} finally {
+			lock.readLock().unlock();
+		}
+	}
+
+	@Override
+	public Collection<Sone> getLocalSones() {
+		lock.readLock().lock();
+		try {
+			return from(allSones.values()).filter(LOCAL_SONE_FILTER).toSet();
+		} finally {
+			lock.readLock().unlock();
+		}
+	}
+
+	@Override
+	public Collection<Sone> getRemoteSones() {
+		lock.readLock().lock();
+		try {
+			return from(allSones.values())
+					.filter(not(LOCAL_SONE_FILTER)) .toSet();
+		} finally {
+			lock.readLock().unlock();
+		}
+	}
+
 	//
 	// POSTPROVIDER METHODS
 	//
@@ -311,13 +394,16 @@ public class MemoryDatabase extends AbstractService implements Database {
 
 	/** {@inheritDocs} */
 	@Override
-	public List<PostReply> getReplies(String postId) {
+	public List<PostReply> getReplies(final String postId) {
 		lock.readLock().lock();
 		try {
-			if (!postReplies.containsKey(postId)) {
-				return Collections.emptyList();
-			}
-			return new ArrayList<PostReply>(postReplies.get(postId));
+			return from(allPostReplies.values())
+					.filter(new Predicate<PostReply>() {
+						@Override
+						public boolean apply(PostReply postReply) {
+							return postReply.getPostId().equals(postId);
+						}
+					}).toSortedList(TIME_COMPARATOR);
 		} finally {
 			lock.readLock().unlock();
 		}
@@ -343,7 +429,6 @@ public class MemoryDatabase extends AbstractService implements Database {
 		lock.writeLock().lock();
 		try {
 			allPostReplies.put(postReply.getId(), postReply);
-			postReplies.put(postReply.getPostId(), postReply);
 		} finally {
 			lock.writeLock().unlock();
 		}
@@ -369,7 +454,6 @@ public class MemoryDatabase extends AbstractService implements Database {
 			for (PostReply postReply : postReplies) {
 				allPostReplies.put(postReply.getId(), postReply);
 				sonePostReplies.put(postReply.getSone().getId(), postReply);
-				this.postReplies.put(postReply.getPostId(), postReply);
 			}
 		} finally {
 			lock.writeLock().unlock();
@@ -382,9 +466,6 @@ public class MemoryDatabase extends AbstractService implements Database {
 		lock.writeLock().lock();
 		try {
 			allPostReplies.remove(postReply.getId());
-			if (postReplies.containsKey(postReply.getPostId())) {
-				postReplies.get(postReply.getPostId()).remove(postReply);
-			}
 		} finally {
 			lock.writeLock().unlock();
 		}
@@ -651,7 +732,7 @@ public class MemoryDatabase extends AbstractService implements Database {
 	private Collection<PostReply> getRepliesFrom(String id) {
 		lock.readLock().lock();
 		try {
-			return Collections.unmodifiableCollection(sonePostReplies.get(id));
+			return unmodifiableCollection(sonePostReplies.get(id));
 		} finally {
 			lock.readLock().unlock();
 		}