X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;ds=sidebyside;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2Fimpl%2FDefaultAlbumBuilder.java;h=fd4b5a514c95c45af8729c61145154e303c7feb8;hb=115487bc44d755f1ae836bafc610fb49a04d9d02;hp=cd6475a30c135fb2d06fe85787a1579aaba15a5d;hpb=414ee1b03f7155ccb5049ede303dd01020ddcd94;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 cd6475a..fd4b5a5 100644
--- a/src/main/java/net/pterodactylus/sone/data/impl/DefaultAlbumBuilder.java
+++ b/src/main/java/net/pterodactylus/sone/data/impl/DefaultAlbumBuilder.java
@@ -20,24 +20,34 @@ 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(Sone sone) {
+	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(), sone);
+		DefaultAlbum album = new DefaultAlbum(database, getId(), sone, parentId);
+		if (parentId != null) {
+			database.storeAlbum(album);
+		}
+		return album;
 	}
 
 }