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=6fce7e0c3701fff4a21284a0be10fbe97bd5398c;hp=962357ebcfb266e21b843b45a6ebadef6966bc73;hb=b4d2d68b5ea4f4edc7337f380cfe078756678126;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..6fce7e0 100644
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.kt
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.kt
@@ -1,5 +1,5 @@
 /*
- * Sone - MemoryDatabase.java - Copyright © 2013–2019 David Roden
+ * Sone - MemoryDatabase.kt - Copyright © 2013–2019 David Roden
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -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() =