X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;ds=sidebyside;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdatabase%2Fmemory%2FMemoryDatabase.java;h=3c1dc741068a1d727f3b10a9f4aac98c2d623741;hb=70ac785b339120420da70a25e1651cd7b81aec0f;hp=5a8fb0078daeef23baff18decf91c732ec3e5fb7;hpb=bcc5ab46b598395bd0b0dbcdb3a0d902fb92a74c;p=Sone.git

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 5a8fb00..3c1dc74 100644
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
@@ -180,6 +180,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());
@@ -199,24 +228,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();
 		}