X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fdemoscenemusic%2Fdata%2FDataManager.java;h=f53559e7bbe062a2b3a97fced6d6810d334b259f;hb=3f4c515611fd5711b2b1c4eb9130166a0673a4e1;hp=0c6f4220189809aad824209a9596c647c094cbf9;hpb=cc3a4940ce06e23e0d480d5db706878a11c007c3;p=demoscenemusic.git

diff --git a/src/main/java/net/pterodactylus/demoscenemusic/data/DataManager.java b/src/main/java/net/pterodactylus/demoscenemusic/data/DataManager.java
index 0c6f422..f53559e 100644
--- a/src/main/java/net/pterodactylus/demoscenemusic/data/DataManager.java
+++ b/src/main/java/net/pterodactylus/demoscenemusic/data/DataManager.java
@@ -47,6 +47,7 @@ public class DataManager {
 	private final ObjectCreator<Artist> artistCreator = new ArtistCreator();
 	private final ObjectCreator<Group> groupCreator = new GroupCreator();
 	private final ObjectCreator<Track> trackCreator = new TrackCreator();
+	private final ObjectCreator<Style> styleCreator = new StyleCreator();
 
 	private final Database database;
 
@@ -75,7 +76,16 @@ public class DataManager {
 		return database.getMultiple(query, artistCreator);
 	}
 
-	public Track trackById(String id) throws DatabaseException {
+	public List<Artist> getArtistsByTrack(String trackId) throws DatabaseException {
+		Query query = new Query(Type.SELECT, "ARTISTS");
+		query.addField(new Field("ARTISTS.*"));
+		query.addJoin(new Join(JoinType.INNER, "TRACK_ARTISTS", new Field("TRACK_ARTISTS.ARTIST"), new Field("ARTISTS.ID")));
+		query.addWhereClause(new ValueFieldWhereClause(new ValueField("TRACK_ARTISTS.TRACK", new StringParameter(trackId))));
+		query.addOrderField(new OrderField(new Field("TRACK_ARTISTS.DISPLAY_ORDER")));
+		return database.getMultiple(query, artistCreator);
+	}
+
+	public Track getTrackById(String id) throws DatabaseException {
 		Query query = new Query(Type.SELECT, "TRACKS");
 		query.addField(new Field("TRACKS.*"));
 		query.addWhereClause(new ValueFieldWhereClause(new ValueField("TRACKS.ID", new StringParameter(id))));
@@ -98,6 +108,14 @@ public class DataManager {
 		return database.getMultiple(query, groupCreator);
 	}
 
+	public Collection<Style> getStylesByTrack(String trackId) throws DatabaseException {
+		Query query = new Query(Type.SELECT, "STYLES");
+		query.addField(new Field("STYLES.*"));
+		query.addJoin(new Join(JoinType.INNER, "TRACK_STYLES", new Field("STYLES.ID"), new Field("TRACK_STYLES.STYLE")));
+		query.addWhereClause(new ValueFieldWhereClause(new ValueField("TRACK_STYLES.TRACK", new StringParameter(trackId))));
+		return database.getMultiple(query, styleCreator);
+	}
+
 	private class LazyArtist extends DefaultArtist {
 
 		private final Memoizer<Void> tracksMemoizer = new Memoizer<Void>(new Callable<Void>() {
@@ -208,6 +226,18 @@ public class DataManager {
 
 		});
 
+		private final Memoizer<Void> stylesMemoizer = new Memoizer<Void>(new Callable<Void>() {
+
+			@Override
+			public Void call() throws Exception {
+				if (!hasValue("styles")) {
+					value("styles", Collection.class).set(getStylesByTrack(id()));
+				}
+				return null;
+			}
+
+		});
+
 		/**
 		 * @param id
 		 */
@@ -229,7 +259,7 @@ public class DataManager {
 		 */
 		@Override
 		public Collection<Style> styles() {
-			// TODO Auto-generated method stub
+			stylesMemoizer.get();
 			return super.styles();
 		}
 
@@ -247,4 +277,16 @@ public class DataManager {
 
 	}
 
+	private class StyleCreator implements ObjectCreator<Style> {
+
+		/**
+		 * {@inheritDoc}
+		 */
+		@Override
+		public Style createObject(ResultSet resultSet) throws SQLException {
+			return new DefaultStyle(resultSet.getString("STYLES.ID")).name(resultSet.getString("STYLES.NAME"));
+		}
+
+	}
+
 }