Use database to return friends of a 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 c537bcb..49531a1 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemorySoneBuilder.java
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemorySoneBuilder.java
@@ -3,6 +3,7 @@ package net.pterodactylus.sone.database.memory;
 import net.pterodactylus.sone.data.Sone;
 import net.pterodactylus.sone.data.impl.SoneImpl;
 import net.pterodactylus.sone.data.impl.AbstractSoneBuilder;
+import net.pterodactylus.sone.database.Database;
 
 /**
  * Memory-based {@link AbstractSoneBuilder} implementation.
@@ -11,10 +12,16 @@ 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, local);
        }
 
 }