X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdatabase%2Fmemory%2FMemoryDatabase.java;h=4bf66437f544557659aa412f8b155c69fa173eb6;hb=72407829d504a0444aadd09cc937bae10b6cb866;hp=c73513d92ae4d95e3f60b0a826e9c5df18fbdc58;hpb=403b51bcf5b736808e3c554b8589759e7d3d5d47;p=Sone.git

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 c73513d..4bf6643 100644
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
@@ -84,7 +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> likedPosts = 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();
@@ -94,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>() {
@@ -175,6 +178,16 @@ 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
@@ -225,6 +238,16 @@ public class MemoryDatabase extends AbstractService implements Database {
 	}
 
 	@Override
+	public void storeSone(Sone sone) {
+		lock.writeLock().lock();
+		try {
+			sones.put(sone.getId(), sone);
+		} finally {
+			lock.writeLock().unlock();
+		}
+	}
+
+	@Override
 	public SoneBuilder newSoneBuilder() {
 		return new DefaultSoneBuilder(this) {
 			@Override
@@ -286,11 +309,45 @@ public class MemoryDatabase extends AbstractService implements Database {
 		}
 	}
 
+	/**
+	 * Returns whether the given post is known.
+	 *
+	 * @param post
+	 * 		The post
+	 * @return {@code true} if the post is known, {@code false} otherwise
+	 */
+	@Override
+	public boolean isPostKnown(Post post) {
+		lock.readLock().lock();
+		try {
+			return knownPosts.contains(post.getId());
+		} finally {
+			lock.readLock().unlock();
+		}
+	}
+
+	/**
+	 * Sets whether the given post is known.
+	 *
+	 * @param post
+	 * 		The post
+	 */
+	@Override
+	public void setPostKnown(Post post) {
+		lock.writeLock().lock();
+		try {
+			knownPosts.add(post.getId());
+		} finally {
+			lock.writeLock().unlock();
+		}
+	}
+
 	@Override
 	public void likePost(Post post, Sone localSone) {
 		lock.writeLock().lock();
 		try {
-			likedPosts.put(localSone.getId(), post.getId());
+			likedPostsBySone.put(localSone.getId(), post.getId());
+			postLikingSones.put(post.getId(), localSone.getId());
 		} finally {
 			lock.writeLock().unlock();
 		}
@@ -300,12 +357,32 @@ public class MemoryDatabase extends AbstractService implements Database {
 	public void unlikePost(Post post, Sone localSone) {
 		lock.writeLock().lock();
 		try {
-			likedPosts.remove(localSone.getId(), post.getId());
+			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
 	//
@@ -420,6 +497,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
 	//
@@ -680,47 +799,6 @@ public class MemoryDatabase extends AbstractService implements Database {
 	}
 
 	//
-	// PACKAGE-PRIVATE METHODS
-	//
-
-	/**
-	 * Returns whether the given post is known.
-	 *
-	 * @param post
-	 * 		The post
-	 * @return {@code true} if the post is known, {@code false} otherwise
-	 */
-	boolean isPostKnown(Post post) {
-		lock.readLock().lock();
-		try {
-			return knownPosts.contains(post.getId());
-		} finally {
-			lock.readLock().unlock();
-		}
-	}
-
-	/**
-	 * Sets whether the given post is known.
-	 *
-	 * @param post
-	 * 		The post
-	 * @param known
-	 * 		{@code true} if the post is known, {@code false} otherwise
-	 */
-	void setPostKnown(Post post, boolean known) {
-		lock.writeLock().lock();
-		try {
-			if (known) {
-				knownPosts.add(post.getId());
-			} else {
-				knownPosts.remove(post.getId());
-			}
-		} finally {
-			lock.writeLock().unlock();
-		}
-	}
-
-	//
 	// PRIVATE METHODS
 	//
 
@@ -837,4 +915,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();
+			}
+		};
+	}
+
 }