X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdatabase%2Fmemory%2FMemoryBookmarkDatabase.java;h=38c5bd05801f4d126689a83a9f8d8d07be8a2251;hb=04fbdc3d76e3b1140d178dc2600141ec92d22637;hp=1aec5734090e61aac828aa3a1a60d99f9f40714b;hpb=aba6bd1d58ef244050703e1b99d7389e024809c6;p=Sone.git

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 1aec573..38c5bd0 100644
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemoryBookmarkDatabase.java
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemoryBookmarkDatabase.java
@@ -21,27 +21,49 @@ public class MemoryBookmarkDatabase implements BookmarkDatabase {
 
 	private final ReadWriteLock lock = new ReentrantReadWriteLock();
 	private final MemoryDatabase memoryDatabase;
+	private final ConfigurationLoader configurationLoader;
 	private final Set<String> bookmarkedPosts = new HashSet<String>();
 
-	public MemoryBookmarkDatabase(MemoryDatabase memoryDatabase) {
+	public MemoryBookmarkDatabase(MemoryDatabase memoryDatabase,
+			ConfigurationLoader configurationLoader) {
 		this.memoryDatabase = memoryDatabase;
+		this.configurationLoader = configurationLoader;
 	}
 
-	@Override
-	public void bookmarkPost(String postId) {
+	public void start() {
+		loadBookmarkedPosts();
+	}
+
+	private void loadBookmarkedPosts() {
+		Set<String> bookmarkedPosts = configurationLoader.loadBookmarkedPosts();
 		lock.writeLock().lock();
 		try {
-			bookmarkedPosts.add(postId);
+			this.bookmarkedPosts.clear();
+			this.bookmarkedPosts.addAll(bookmarkedPosts);
 		} finally {
 			lock.writeLock().unlock();
 		}
 	}
 
+	public void stop() {
+		saveBookmarkedPosts();
+	}
+
+	private void saveBookmarkedPosts() {
+		lock.readLock().lock();
+		try {
+			configurationLoader.saveBookmarkedPosts(this.bookmarkedPosts);
+		} finally {
+			lock.readLock().unlock();
+		}
+	}
+
 	@Override
 	public void bookmarkPost(Post post) {
 		lock.writeLock().lock();
 		try {
 			bookmarkedPosts.add(post.getId());
+			saveBookmarkedPosts();
 		} finally {
 			lock.writeLock().unlock();
 		}
@@ -52,6 +74,7 @@ public class MemoryBookmarkDatabase implements BookmarkDatabase {
 		lock.writeLock().lock();
 		try {
 			bookmarkedPosts.remove(post.getId());
+			saveBookmarkedPosts();
 		} finally {
 			lock.writeLock().unlock();
 		}