X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2Fimpl%2FDefaultAlbumBuilder.java;h=fd4b5a514c95c45af8729c61145154e303c7feb8;hb=8d49fb185ee26853262a12f6d419ef2e39c1116c;hp=15d9e1f952c843b33d6658c3da8ec267a2f47860;hpb=fcf912e985e92d17766736551e69fa53a1730c4b;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/data/impl/DefaultAlbumBuilder.java b/src/main/java/net/pterodactylus/sone/data/impl/DefaultAlbumBuilder.java
index 15d9e1f..fd4b5a5 100644
--- a/src/main/java/net/pterodactylus/sone/data/impl/DefaultAlbumBuilder.java
+++ b/src/main/java/net/pterodactylus/sone/data/impl/DefaultAlbumBuilder.java
@@ -18,19 +18,36 @@
 package net.pterodactylus.sone.data.impl;
 
 import net.pterodactylus.sone.data.Album;
+import net.pterodactylus.sone.data.Sone;
 import net.pterodactylus.sone.database.AlbumBuilder;
+import net.pterodactylus.sone.database.Database;
 
 /**
- * {@link AlbumBuilder} implementation that creates {@link DefaultAlbum} objects.
+ * {@link AlbumBuilder} implementation that creates {@link DefaultAlbum}
+ * objects.
  *
  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  */
 public class DefaultAlbumBuilder extends AbstractAlbumBuilder {
 
+	private final Database database;
+	private final Sone sone;
+	private final String parentId;
+
+	public DefaultAlbumBuilder(Database database, Sone sone, String parentId) {
+		this.database = database;
+		this.sone = sone;
+		this.parentId = parentId;
+	}
+
 	@Override
 	public Album build() throws IllegalStateException {
 		validate();
-		return new DefaultAlbum(getId());
+		DefaultAlbum album = new DefaultAlbum(database, getId(), sone, parentId);
+		if (parentId != null) {
+			database.storeAlbum(album);
+		}
+		return album;
 	}
 
 }