X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdatabase%2Fmemory%2FMemoryDatabase.java;h=bfea5e2fb153d357d865aafcec77037eb56c0869;hb=858555f647a1c39b2a6707192bdb8f32107aa5dd;hp=7ad40e0e36799f943975a610efc5ce5e51fb8700;hpb=3049b7e396ce20b5b6600ae2791334a6babb414e;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 7ad40e0..bfea5e2 100644
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
@@ -132,6 +132,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 MemorySoneDatabase soneDatabase;
 	private final MemoryPostDatabase postDatabase;
 	private final MemoryBookmarkDatabase memoryBookmarkDatabase;
 	private final MemoryFriendDatabase memoryFriendDatabase;
@@ -149,6 +150,7 @@ public class MemoryDatabase extends AbstractService implements Database {
 		this.soneProvider = soneProvider;
 		this.configuration = configuration;
 		this.configurationLoader = new ConfigurationLoader(configuration);
+		soneDatabase = new MemorySoneDatabase(configurationLoader);
 		postDatabase = new MemoryPostDatabase(this, configurationLoader);
 		memoryBookmarkDatabase =
 				new MemoryBookmarkDatabase(this, configurationLoader);
@@ -461,6 +463,7 @@ public class MemoryDatabase extends AbstractService implements Database {
 	/** {@inheritDocs} */
 	@Override
 	protected void doStart() {
+		soneDatabase.start();
 		postDatabase.start();
 		memoryBookmarkDatabase.start();
 		loadKnownPostReplies();
@@ -471,6 +474,7 @@ public class MemoryDatabase extends AbstractService implements Database {
 	@Override
 	protected void doStop() {
 		try {
+			soneDatabase.stop();
 			postDatabase.stop();
 			memoryBookmarkDatabase.stop();
 			save();
@@ -501,6 +505,16 @@ public class MemoryDatabase extends AbstractService implements Database {
 		}
 	}
 
+	@Override
+	public boolean isSoneKnown(Sone sone) {
+		return soneDatabase.isKnownSone(sone.getId());
+	}
+
+	@Override
+	public void setSoneKnown(Sone sone) {
+		soneDatabase.setSoneKnown(sone.getId());
+	}
+
 	private void storePosts(String soneId, Collection<Post> posts) {
 		postDatabase.storePosts(soneId, posts);
 	}
@@ -610,7 +624,7 @@ public class MemoryDatabase extends AbstractService implements Database {
 	}
 
 	@Override
-	public Collection<String> getFriends(Sone localSone) {
+	public Collection<String> getFriends(LocalSone localSone) {
 		if (!localSone.isLocal()) {
 			return Collections.emptySet();
 		}
@@ -618,7 +632,7 @@ public class MemoryDatabase extends AbstractService implements Database {
 	}
 
 	@Override
-	public boolean isFriend(Sone localSone, String friendSoneId) {
+	public boolean isFriend(LocalSone localSone, String friendSoneId) {
 		if (!localSone.isLocal()) {
 			return false;
 		}