X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdatabase%2Fmemory%2FMemorySoneBuilder.java;h=e07999d09e13540181d5080041888bba29b6d9e1;hb=f8672b1385173a103d7f085d8e9cd43bc5762d71;hp=c537bcb66898d2b62f14adc0d45c49000650cf9c;hpb=1bc68398c8912a992d87087b79ad9bbea8964c76;p=Sone.git

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 c537bcb..e07999d 100644
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemorySoneBuilder.java
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemorySoneBuilder.java
@@ -1,8 +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.
@@ -11,10 +16,22 @@ import net.pterodactylus.sone.data.impl.AbstractSoneBuilder;
  */
 public class MemorySoneBuilder extends AbstractSoneBuilder {
 
+	private final Database database;
+
+	public MemorySoneBuilder(Database database) {
+		this.database = database;
+	}
+
 	@Override
 	public Sone build() throws IllegalStateException {
 		validate();
-		return new SoneImpl(identity, local);
+		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);
 	}
 
 }