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=975b8ab42a9b31a5a968555c4657ddbb035fecd7;hb=90d8e2eacb9888735865d8f890d105885959ca65;hpb=4aec89e6f9c6eb0972be5e9a3e1363bb487a2898

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 975b8ab..df39511 100644
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
@@ -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,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
@@ -309,6 +321,15 @@ public class MemoryDatabase extends AbstractService implements Database {
 		}
 	}
 
+	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();
@@ -433,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
 	//