X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdatabase%2Fmemory%2FMemoryDatabase.java;h=c611eea01d4b7c24650ed65983ab5d532456a8e6;hp=d129d87fda5368026f31e2a3974cc5164337ced2;hb=d0170947a5968260c73b9b985466ede0f7c12336;hpb=fc618f3bd07fdb608dedc2ee80dc697ca108ea65

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 d129d87..c611eea 100644
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
@@ -82,6 +82,7 @@ public class MemoryDatabase extends AbstractService implements Database {
 
 	/** The configuration. */
 	private final Configuration configuration;
+	private final ConfigurationLoader configurationLoader;
 
 	private final Map<String, Sone> allSones = new HashMap<String, Sone>();
 
@@ -115,6 +116,8 @@ 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;
+
 	/**
 	 * Creates a new memory database.
 	 *
@@ -127,6 +130,9 @@ public class MemoryDatabase extends AbstractService implements Database {
 	public MemoryDatabase(SoneProvider soneProvider, Configuration configuration) {
 		this.soneProvider = soneProvider;
 		this.configuration = configuration;
+		this.configurationLoader = new ConfigurationLoader(configuration);
+		memoryBookmarkDatabase =
+				new MemoryBookmarkDatabase(this, configurationLoader);
 	}
 
 	//
@@ -152,6 +158,7 @@ public class MemoryDatabase extends AbstractService implements Database {
 	/** {@inheritDocs} */
 	@Override
 	protected void doStart() {
+		memoryBookmarkDatabase.start();
 		loadKnownPosts();
 		loadKnownPostReplies();
 		notifyStarted();
@@ -161,6 +168,7 @@ public class MemoryDatabase extends AbstractService implements Database {
 	@Override
 	protected void doStop() {
 		try {
+			memoryBookmarkDatabase.stop();
 			save();
 			notifyStopped();
 		} catch (DatabaseException de1) {
@@ -177,6 +185,35 @@ public class MemoryDatabase extends AbstractService implements Database {
 	public void storeSone(Sone sone) {
 		lock.writeLock().lock();
 		try {
+			removeSone(sone);
+
+			allSones.put(sone.getId(), sone);
+			sonePosts.putAll(sone.getId(), sone.getPosts());
+			for (Post post : sone.getPosts()) {
+				allPosts.put(post.getId(), post);
+			}
+			sonePostReplies.putAll(sone.getId(), sone.getReplies());
+			for (PostReply postReply : sone.getReplies()) {
+				allPostReplies.put(postReply.getId(), postReply);
+			}
+			soneAlbums.putAll(sone.getId(), toAllAlbums.apply(sone));
+			for (Album album : toAllAlbums.apply(sone)) {
+				allAlbums.put(album.getId(), album);
+			}
+			soneImages.putAll(sone.getId(), toAllImages.apply(sone));
+			for (Image image : toAllImages.apply(sone)) {
+				allImages.put(image.getId(), image);
+			}
+		} finally {
+			lock.writeLock().unlock();
+		}
+	}
+
+	@Override
+	public void removeSone(Sone sone) {
+		lock.writeLock().lock();
+		try {
+			allSones.remove(sone.getId());
 			Collection<Post> removedPosts = sonePosts.removeAll(sone.getId());
 			for (Post removedPost : removedPosts) {
 				allPosts.remove(removedPost.getId());
@@ -196,24 +233,6 @@ public class MemoryDatabase extends AbstractService implements Database {
 			for (Image removedImage : removedImages) {
 				allImages.remove(removedImage.getId());
 			}
-
-			allSones.put(sone.getId(), sone);
-			sonePosts.putAll(sone.getId(), sone.getPosts());
-			for (Post post : sone.getPosts()) {
-				allPosts.put(post.getId(), post);
-			}
-			sonePostReplies.putAll(sone.getId(), sone.getReplies());
-			for (PostReply postReply : sone.getReplies()) {
-				allPostReplies.put(postReply.getId(), postReply);
-			}
-			soneAlbums.putAll(sone.getId(), toAllAlbums.apply(sone));
-			for (Album album : toAllAlbums.apply(sone)) {
-				allAlbums.put(album.getId(), album);
-			}
-			soneImages.putAll(sone.getId(), toAllImages.apply(sone));
-			for (Image image : toAllImages.apply(sone)) {
-				allImages.put(image.getId(), image);
-			}
 		} finally {
 			lock.writeLock().unlock();
 		}
@@ -338,51 +357,6 @@ public class MemoryDatabase extends AbstractService implements Database {
 		}
 	}
 
-	/** {@inheritDocs} */
-	@Override
-	public void storePosts(Sone sone, Collection<Post> posts) throws IllegalArgumentException {
-		checkNotNull(sone, "sone must not be null");
-		/* verify that all posts are from the same Sone. */
-		for (Post post : posts) {
-			if (!sone.equals(post.getSone())) {
-				throw new IllegalArgumentException(String.format("Post from different Sone found: %s", post));
-			}
-		}
-
-		lock.writeLock().lock();
-		try {
-			/* remove all posts by the Sone. */
-			Collection<Post> oldPosts = getPostsFrom(sone.getId());
-			for (Post post : oldPosts) {
-				allPosts.remove(post.getId());
-			}
-
-			/* add new posts. */
-			getPostsFrom(sone.getId()).addAll(posts);
-			for (Post post : posts) {
-				allPosts.put(post.getId(), post);
-			}
-		} finally {
-			lock.writeLock().unlock();
-		}
-	}
-
-	/** {@inheritDocs} */
-	@Override
-	public void removePosts(Sone sone) {
-		checkNotNull(sone, "sone must not be null");
-		lock.writeLock().lock();
-		try {
-			/* remove all posts by the Sone. */
-			getPostsFrom(sone.getId()).clear();
-			for (Post post : sone.getPosts()) {
-				allPosts.remove(post.getId());
-			}
-		} finally {
-			lock.writeLock().unlock();
-		}
-	}
-
 	//
 	// POSTREPLYPROVIDER METHODS
 	//
@@ -442,32 +416,6 @@ public class MemoryDatabase extends AbstractService implements Database {
 
 	/** {@inheritDocs} */
 	@Override
-	public void storePostReplies(Sone sone, Collection<PostReply> postReplies) {
-		checkNotNull(sone, "sone must not be null");
-		/* verify that all posts are from the same Sone. */
-		for (PostReply postReply : postReplies) {
-			if (!sone.equals(postReply.getSone())) {
-				throw new IllegalArgumentException(String.format("PostReply from different Sone found: %s", postReply));
-			}
-		}
-
-		lock.writeLock().lock();
-		try {
-			/* remove all post replies of the Sone. */
-			for (PostReply postReply : getRepliesFrom(sone.getId())) {
-				removePostReply(postReply);
-			}
-			for (PostReply postReply : postReplies) {
-				allPostReplies.put(postReply.getId(), postReply);
-				sonePostReplies.put(postReply.getSone().getId(), postReply);
-			}
-		} finally {
-			lock.writeLock().unlock();
-		}
-	}
-
-	/** {@inheritDocs} */
-	@Override
 	public void removePostReply(PostReply postReply) {
 		lock.writeLock().lock();
 		try {
@@ -477,21 +425,6 @@ public class MemoryDatabase extends AbstractService implements Database {
 		}
 	}
 
-	/** {@inheritDocs} */
-	@Override
-	public void removePostReplies(Sone sone) {
-		checkNotNull(sone, "sone must not be null");
-
-		lock.writeLock().lock();
-		try {
-			for (PostReply postReply : sone.getReplies()) {
-				removePostReply(postReply);
-			}
-		} finally {
-			lock.writeLock().unlock();
-		}
-	}
-
 	//
 	// ALBUMPROVDER METHODS
 	//
@@ -590,6 +523,26 @@ public class MemoryDatabase extends AbstractService implements Database {
 		}
 	}
 
+	@Override
+	public void bookmarkPost(Post post) {
+		memoryBookmarkDatabase.bookmarkPost(post);
+	}
+
+	@Override
+	public void unbookmarkPost(Post post) {
+		memoryBookmarkDatabase.unbookmarkPost(post);
+	}
+
+	@Override
+	public boolean isPostBookmarked(Post post) {
+		return memoryBookmarkDatabase.isPostBookmarked(post);
+	}
+
+	@Override
+	public Set<Post> getBookmarkedPosts() {
+		return memoryBookmarkDatabase.getBookmarkedPosts();
+	}
+
 	//
 	// PACKAGE-PRIVATE METHODS
 	//
@@ -692,16 +645,11 @@ public class MemoryDatabase extends AbstractService implements Database {
 
 	/** Loads the known posts. */
 	private void loadKnownPosts() {
+		Set<String> knownPosts = configurationLoader.loadKnownPosts();
 		lock.writeLock().lock();
 		try {
-			int postCounter = 0;
-			while (true) {
-				String knownPostId = configuration.getStringValue("KnownPosts/" + postCounter++ + "/ID").getValue(null);
-				if (knownPostId == null) {
-					break;
-				}
-				knownPosts.add(knownPostId);
-			}
+			this.knownPosts.clear();
+			this.knownPosts.addAll(knownPosts);
 		} finally {
 			lock.writeLock().unlock();
 		}
@@ -728,34 +676,13 @@ public class MemoryDatabase extends AbstractService implements Database {
 		}
 	}
 
-	/**
-	 * Returns all replies by the given Sone.
-	 *
-	 * @param id
-	 * 		The ID of the Sone
-	 * @return The post replies of the Sone, sorted by time (newest first)
-	 */
-	private Collection<PostReply> getRepliesFrom(String id) {
-		lock.readLock().lock();
-		try {
-			return unmodifiableCollection(sonePostReplies.get(id));
-		} finally {
-			lock.readLock().unlock();
-		}
-	}
-
 	/** Loads the known post replies. */
 	private void loadKnownPostReplies() {
+		Set<String> knownPostReplies = configurationLoader.loadKnownPostReplies();
 		lock.writeLock().lock();
 		try {
-			int replyCounter = 0;
-			while (true) {
-				String knownReplyId = configuration.getStringValue("KnownReplies/" + replyCounter++ + "/ID").getValue(null);
-				if (knownReplyId == null) {
-					break;
-				}
-				knownPostReplies.add(knownReplyId);
-			}
+			this.knownPostReplies.clear();
+			this.knownPostReplies.addAll(knownPostReplies);
 		} finally {
 			lock.writeLock().unlock();
 		}