Unbookmarking a post saves the database.

[Sone.git] / src / main / java / net / pterodactylus / sone / database / memory / MemoryDatabase.java

diff --git a/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java b/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
index 3c5ef55..0f795eb 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
@@ -116,8 +116,7 @@ public class MemoryDatabase extends AbstractService implements Database {
        private final Map<String, Image> allImages = new HashMap<String, Image>();
        private final Multimap<String, Image> soneImages = HashMultimap.create();
 
-       private final MemoryBookmarkDatabase memoryBookmarkDatabase =
-                       new MemoryBookmarkDatabase(this);
+       private final MemoryBookmarkDatabase memoryBookmarkDatabase;
 
        /**
         * Creates a new memory database.
@@ -132,6 +131,8 @@ public class MemoryDatabase extends AbstractService implements Database {
                this.soneProvider = soneProvider;
                this.configuration = configuration;
                this.configurationLoader = new ConfigurationLoader(configuration);
+               memoryBookmarkDatabase =
+                               new MemoryBookmarkDatabase(this, configurationLoader);
        }
 
        //
@@ -157,6 +158,7 @@ public class MemoryDatabase extends AbstractService implements Database {
        /** {@inheritDocs} */
        @Override
        protected void doStart() {
+               memoryBookmarkDatabase.start();
                loadKnownPosts();
                loadKnownPostReplies();
                notifyStarted();
@@ -166,6 +168,7 @@ public class MemoryDatabase extends AbstractService implements Database {
        @Override
        protected void doStop() {
                try {
+                       memoryBookmarkDatabase.stop();
                        save();
                        notifyStopped();
                } catch (DatabaseException de1) {
@@ -607,11 +610,6 @@ public class MemoryDatabase extends AbstractService implements Database {
        }
 
        @Override
-       public void bookmarkPost(String postId) {
-               memoryBookmarkDatabase.bookmarkPost(postId);
-       }
-
-       @Override
        public void bookmarkPost(Post post) {
                memoryBookmarkDatabase.bookmarkPost(post);
        }