X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fdatabase%2Fmemory%2FMemoryDatabase.kt;h=d48ef1e9f7c315e8063670a641d73fa79b4e77d4;hp=3b5a6a934c153919359a3df0ad67683659e10341;hb=9fab35abd7dc86afe186ed69a86758a3a0ac426c;hpb=80538cb39f2f708a5811b00bd95cdee8efa0e05d

diff --git a/src/main/kotlin/net/pterodactylus/sone/database/memory/MemoryDatabase.kt b/src/main/kotlin/net/pterodactylus/sone/database/memory/MemoryDatabase.kt
index 3b5a6a9..d48ef1e 100644
--- a/src/main/kotlin/net/pterodactylus/sone/database/memory/MemoryDatabase.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/database/memory/MemoryDatabase.kt
@@ -63,7 +63,7 @@ class MemoryDatabase @Inject constructor(private val configuration: Configuratio
 	private val allPosts = mutableMapOf<String, Post>()
 	private val sonePosts: Multimap<String, Post> = HashMultimap.create<String, Post>()
 	private val knownPosts = mutableSetOf<String>()
-	private val allPostReplies = mutableMapOf<String, PostReply>()
+	private val allPostReplies = mutableMapOf<String, MemoryPostReply.Shell>()
 	private val sonePostReplies: Multimap<String, PostReply> = TreeMultimap.create<String, PostReply>(Comparator { leftString, rightString -> leftString.compareTo(rightString) }, newestReplyFirst)
 	private val knownPostReplies = mutableSetOf<String>()
 	private val allAlbums = mutableMapOf<String, Album>()
@@ -123,7 +123,7 @@ class MemoryDatabase @Inject constructor(private val configuration: Configuratio
 			}
 			sonePostReplies.putAll(sone.id, sone.replies)
 			for (postReply in sone.replies) {
-				allPostReplies[postReply.id] = postReply
+				allPostReplies[postReply.id] = postReply.toShell()
 			}
 			sone.allAlbums.let { albums ->
 				soneAlbums.putAll(sone.id, albums)
@@ -223,12 +223,15 @@ class MemoryDatabase @Inject constructor(private val configuration: Configuratio
 		}
 	}
 
-	override fun getPostReply(id: String) = readLock.withLock { allPostReplies[id] }
+	override fun getPostReply(id: String) = readLock.withLock {
+		allPostReplies[id]?.build(newPostReplyBuilder())
+	}
 
 	override fun getReplies(postId: String) =
 			readLock.withLock {
 				allPostReplies.values
 						.filter { it.postId == postId }
+						.map { it.build(newPostReplyBuilder()) }
 						.sortedWith(newestReplyFirst.reversed())
 			}
 
@@ -237,7 +240,7 @@ class MemoryDatabase @Inject constructor(private val configuration: Configuratio
 
 	override fun storePostReply(postReply: PostReply) =
 			writeLock.withLock {
-				allPostReplies[postReply.id] = postReply
+				allPostReplies[postReply.id] = postReply.toShell()
 			}
 
 	override fun removePostReply(postReply: PostReply) =
@@ -297,7 +300,7 @@ class MemoryDatabase @Inject constructor(private val configuration: Configuratio
 				saveKnownPosts()
 			}
 
-	protected fun isPostReplyKnown(postReply: PostReply) = readLock.withLock { postReply.id in knownPostReplies }
+	internal fun isPostReplyKnown(postReply: PostReply) = readLock.withLock { postReply.id in knownPostReplies }
 
 	override fun setPostReplyKnown(postReply: PostReply): Unit =
 			writeLock.withLock {