X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2Fimpl%2FDefaultAlbumBuilder.java;h=a2f198f33ce793b20dc188dfcce01101fba33456;hb=54e40cf0c1c650e9eca2d50762a92b5d13ccfa85;hp=5148471310d0d93f4c099b26d93a42f6611195c2;hpb=f0c0d175ae5058661d3af7cc99d0c188616a4d8f;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 5148471..a2f198f 100644
--- a/src/main/java/net/pterodactylus/sone/data/impl/DefaultAlbumBuilder.java
+++ b/src/main/java/net/pterodactylus/sone/data/impl/DefaultAlbumBuilder.java
@@ -20,26 +20,32 @@ 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 DefaultAlbum parent;
+	private final String parentId;
 
-	public DefaultAlbumBuilder(Sone sone, DefaultAlbum parent) {
+	public DefaultAlbumBuilder(Database database, Sone sone, String parentId) {
+		this.database = database;
 		this.sone = sone;
-		this.parent = parent;
+		this.parentId = parentId;
 	}
 
 	@Override
 	public Album build() throws IllegalStateException {
 		validate();
-		return new DefaultAlbum(getId(), sone, parent);
+		DefaultAlbum album = new DefaultAlbum(database, getId(), sone, parentId);
+		database.storeAlbum(album);
+		return album;
 	}
 
 }