X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdatabase%2Fmemory%2FMemoryDatabase.kt;h=4e960ec5407f1d5a30df049a6a239e8c133252ee;hp=962357ebcfb266e21b843b45a6ebadef6966bc73;hb=a697ce78fc213bd83c696507cc66796610fd189c;hpb=081a7b1344574efb641fe925088e4eb975e2e47e

diff --git a/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.kt b/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.kt
index 962357e..4e960ec 100644
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.kt
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.kt
@@ -281,7 +281,7 @@ class MemoryDatabase @Inject constructor(private val configuration: Configuratio
 
 	protected fun isPostKnown(post: Post) = readLock.withLock { post.id in knownPosts }
 
-	protected fun setPostKnown(post: Post, known: Boolean): Unit =
+	fun setPostKnown(post: Post, known: Boolean): Unit =
 			writeLock.withLock {
 				if (known)
 					knownPosts.add(post.id)
@@ -292,12 +292,13 @@ class MemoryDatabase @Inject constructor(private val configuration: Configuratio
 
 	protected fun isPostReplyKnown(postReply: PostReply) = readLock.withLock { postReply.id in knownPostReplies }
 
-	protected fun setPostReplyKnown(postReply: PostReply, known: Boolean): Unit =
+	fun setPostReplyKnown(postReply: PostReply, known: Boolean): Unit =
 			writeLock.withLock {
 				if (known)
 					knownPostReplies.add(postReply.id)
 				else
 					knownPostReplies.remove(postReply.id)
+				saveKnownPostReplies()
 			}
 
 	private fun loadKnownPosts() =