Use different method to create a local Sone.

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

diff --git a/src/main/java/net/pterodactylus/sone/database/memory/MemorySoneBuilder.java b/src/main/java/net/pterodactylus/sone/database/memory/MemorySoneBuilder.java
index 3cdb120..e07999d 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemorySoneBuilder.java
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemorySoneBuilder.java
@@ -1,9 +1,13 @@
 package net.pterodactylus.sone.database.memory;
 
+import net.pterodactylus.sone.data.LocalSone;
 import net.pterodactylus.sone.data.Sone;
-import net.pterodactylus.sone.data.impl.SoneImpl;
 import net.pterodactylus.sone.data.impl.AbstractSoneBuilder;
+import net.pterodactylus.sone.data.impl.SoneImpl;
 import net.pterodactylus.sone.database.Database;
+import net.pterodactylus.sone.freenet.wot.OwnIdentity;
+
+import com.google.common.base.Preconditions;
 
 /**
  * Memory-based {@link AbstractSoneBuilder} implementation.
@@ -21,7 +25,13 @@ public class MemorySoneBuilder extends AbstractSoneBuilder {
        @Override
        public Sone build() throws IllegalStateException {
                validate();
-               return new SoneImpl(database, identity, local, lastUpdated, client, posts, postReplies);
+               return new SoneImpl(database, identity, false, lastUpdated, client, posts, postReplies);
+       }
+
+       @Override
+       public LocalSone buildLocal() throws IllegalStateException {
+               validateLocal();
+               return new SoneImpl(database, identity, true, lastUpdated, client, posts, postReplies);
        }
 
 }