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=df3951110ea5e2a9f89a1fbab36ce831bc6fa852;hp=a935b75efb209af7b527fb4e96f3dbf1c73f6473;hb=90d8e2eacb9888735865d8f890d105885959ca65;hpb=16cadd4cd69c0c8cbbd6bfc41257c852e6590e40

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 a935b75..df39511 100644
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
@@ -33,8 +33,6 @@ import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
 import java.util.Set;
-import java.util.SortedSet;
-import java.util.TreeSet;
 import java.util.concurrent.locks.ReadWriteLock;
 import java.util.concurrent.locks.ReentrantReadWriteLock;
 
@@ -42,7 +40,6 @@ import net.pterodactylus.sone.data.Album;
 import net.pterodactylus.sone.data.Image;
 import net.pterodactylus.sone.data.Post;
 import net.pterodactylus.sone.data.PostReply;
-import net.pterodactylus.sone.data.Reply;
 import net.pterodactylus.sone.data.Sone;
 import net.pterodactylus.sone.data.impl.DefaultSoneBuilder;
 import net.pterodactylus.sone.database.Database;
@@ -60,6 +57,7 @@ import com.google.common.collect.HashMultimap;
 import com.google.common.collect.ListMultimap;
 import com.google.common.collect.Maps;
 import com.google.common.collect.Multimap;
+import com.google.common.collect.SetMultimap;
 import com.google.common.collect.SortedSetMultimap;
 import com.google.common.collect.TreeMultimap;
 import com.google.common.util.concurrent.AbstractService;
@@ -86,6 +84,8 @@ public class MemoryDatabase extends AbstractService implements Database {
 
 	/** All posts by their Sones. */
 	private final Multimap<String, Post> sonePosts = HashMultimap.create();
+	private final SetMultimap<String, String> likedPostsBySone = HashMultimap.create();
+	private final SetMultimap<String, String> postLikingSones = HashMultimap.create();
 
 	/** All posts by their recipient. */
 	private final Multimap<String, Post> recipientPosts = HashMultimap.create();
@@ -95,6 +95,8 @@ public class MemoryDatabase extends AbstractService implements Database {
 
 	/** All post replies by their ID. */
 	private final Map<String, PostReply> allPostReplies = new HashMap<String, PostReply>();
+	private final SetMultimap<String, String> likedPostRepliesBySone = HashMultimap.create();
+	private final SetMultimap<String, String> postReplyLikingSones = HashMultimap.create();
 
 	/** Replies sorted by Sone. */
 	private final SortedSetMultimap<String, PostReply> sonePostReplies = TreeMultimap.create(new Comparator<String>() {
@@ -176,6 +178,26 @@ public class MemoryDatabase extends AbstractService implements Database {
 	}
 
 	@Override
+	public void storeIdentity(Identity identitiy) {
+		lock.writeLock().lock();
+		try {
+			identities.put(identitiy.getId(), identitiy);
+		} finally {
+			lock.writeLock().unlock();
+		}
+	}
+
+	@Override
+	public Function<String, Optional<Sone>> getSone() {
+		return new Function<String, Optional<Sone>>() {
+			@Override
+			public Optional<Sone> apply(String soneId) {
+				return (soneId == null) ? Optional.<Sone>absent() : getSone(soneId);
+			}
+		};
+	}
+
+	@Override
 	public Optional<Sone> getSone(String soneId) {
 		lock.readLock().lock();
 		try {
@@ -221,7 +243,12 @@ public class MemoryDatabase extends AbstractService implements Database {
 			@Override
 			public Sone build(Optional<SoneCreated> soneCreated) throws IllegalStateException {
 				Sone sone = super.build(soneCreated);
-				sones.put(sone.getId(), sone);
+				lock.writeLock().lock();
+				try {
+					sones.put(sone.getId(), sone);
+				} finally {
+					lock.writeLock().unlock();
+				}
 				return sone;
 			}
 		};
@@ -232,6 +259,16 @@ public class MemoryDatabase extends AbstractService implements Database {
 	//
 
 	@Override
+	public Function<String, Optional<Post>> getPost() {
+		return new Function<String, Optional<Post>>() {
+			@Override
+			public Optional<Post> apply(String postId) {
+				return (postId == null) ? Optional.<Post>absent() : getPost(postId);
+			}
+		};
+	}
+
+	@Override
 	public Optional<Post> getPost(String postId) {
 		lock.readLock().lock();
 		try {
@@ -262,6 +299,47 @@ public class MemoryDatabase extends AbstractService implements Database {
 		}
 	}
 
+	@Override
+	public void likePost(Post post, Sone localSone) {
+		lock.writeLock().lock();
+		try {
+			likedPostsBySone.put(localSone.getId(), post.getId());
+			postLikingSones.put(post.getId(), localSone.getId());
+		} finally {
+			lock.writeLock().unlock();
+		}
+	}
+
+	@Override
+	public void unlikePost(Post post, Sone localSone) {
+		lock.writeLock().lock();
+		try {
+			likedPostsBySone.remove(localSone.getId(), post.getId());
+			postLikingSones.remove(post.getId(), localSone.getId());
+		} finally {
+			lock.writeLock().unlock();
+		}
+	}
+
+	public boolean isLiked(Post post, Sone sone) {
+		lock.readLock().lock();
+		try {
+			return likedPostsBySone.containsEntry(sone.getId(), post.getId());
+		} finally {
+			lock.readLock().unlock();
+		}
+	}
+
+	@Override
+	public Set<Sone> getLikes(Post post) {
+		lock.readLock().lock();
+		try {
+			return from(postLikingSones.get(post.getId())).transform(getSone()).transformAndConcat(this.<Sone>unwrap()).toSet();
+		} finally {
+			lock.readLock().unlock();
+		}
+	}
+
 	//
 	// POSTSTORE METHODS
 	//
@@ -376,6 +454,48 @@ public class MemoryDatabase extends AbstractService implements Database {
 		}
 	}
 
+	@Override
+	public void likePostReply(PostReply postReply, Sone localSone) {
+		lock.writeLock().lock();
+		try {
+			likedPostRepliesBySone.put(localSone.getId(), postReply.getId());
+			postReplyLikingSones.put(postReply.getId(), localSone.getId());
+		} finally {
+			lock.writeLock().unlock();
+		}
+	}
+
+	@Override
+	public void unlikePostReply(PostReply postReply, Sone localSone) {
+		lock.writeLock().lock();
+		try {
+			likedPostRepliesBySone.remove(localSone.getId(), postReply.getId());
+			postReplyLikingSones.remove(postReply.getId(), localSone.getId());
+		} finally {
+			lock.writeLock().unlock();
+		}
+	}
+
+	@Override
+	public boolean isLiked(PostReply postReply, Sone sone) {
+		lock.readLock().lock();
+		try {
+			return postReplyLikingSones.containsEntry(postReply.getId(), sone.getId());
+		} finally {
+			lock.readLock().unlock();
+		}
+	}
+
+	@Override
+	public Set<Sone> getLikes(PostReply postReply) {
+		lock.readLock().lock();
+		try {
+			return from(postReplyLikingSones.get(postReply.getId())).transform(getSone()).transformAndConcat(this.<Sone>unwrap()).toSet();
+		} finally {
+			lock.readLock().unlock();
+		}
+	}
+
 	//
 	// POSTREPLYSTORE METHODS
 	//
@@ -534,7 +654,9 @@ public class MemoryDatabase extends AbstractService implements Database {
 		lock.writeLock().lock();
 		try {
 			allAlbums.put(album.getId(), album);
-			albumChildren.put(album.getParent().getId(), album.getId());
+			if (!album.isRoot()) {
+				albumChildren.put(album.getParent().getId(), album.getId());
+			}
 		} finally {
 			lock.writeLock().unlock();
 		}
@@ -791,4 +913,13 @@ public class MemoryDatabase extends AbstractService implements Database {
 		};
 	}
 
+	private static <T> Function<Optional<T>, Iterable<T>> unwrap() {
+		return new Function<Optional<T>, Iterable<T>>() {
+			@Override
+			public Iterable<T> apply(Optional<T> input) {
+				return (input == null) ? Collections.<T>emptyList() : input.asSet();
+			}
+		};
+	}
+
 }