X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdatabase%2Fmemory%2FMemoryBookmarkDatabase.java;h=aae93dd3cde38c6530768327487227a547b6afba;hp=78c8a4f093b421465ae8271f564bad4b276a2a12;hb=308b94c1c75e55637644142e2553aec8d7ce01ad;hpb=36b8e20651c14970ca30a1b8ed43af219b367bc9

diff --git a/src/main/java/net/pterodactylus/sone/database/memory/MemoryBookmarkDatabase.java b/src/main/java/net/pterodactylus/sone/database/memory/MemoryBookmarkDatabase.java
index 78c8a4f..aae93dd 100644
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemoryBookmarkDatabase.java
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemoryBookmarkDatabase.java
@@ -45,13 +45,16 @@ public class MemoryBookmarkDatabase implements BookmarkDatabase {
 		}
 	}
 
-	@Override
-	public void bookmarkPost(String postId) {
-		lock.writeLock().lock();
+	public void stop() {
+		saveBookmarkedPosts();
+	}
+
+	private void saveBookmarkedPosts() {
+		lock.readLock().lock();
 		try {
-			bookmarkedPosts.add(postId);
+			configurationLoader.saveBookmarkedPosts(this.bookmarkedPosts);
 		} finally {
-			lock.writeLock().unlock();
+			lock.readLock().unlock();
 		}
 	}
 
@@ -60,6 +63,7 @@ public class MemoryBookmarkDatabase implements BookmarkDatabase {
 		lock.writeLock().lock();
 		try {
 			bookmarkedPosts.add(post.getId());
+			saveBookmarkedPosts();
 		} finally {
 			lock.writeLock().unlock();
 		}