import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Collection;
+import java.util.EnumMap;
+import java.util.HashSet;
import java.util.List;
+import java.util.Map;
import java.util.concurrent.Callable;
+import net.pterodactylus.demoscenemusic.data.Track.Relationship;
import net.pterodactylus.util.collection.Memoizer;
import net.pterodactylus.util.database.Database;
import net.pterodactylus.util.database.DatabaseException;
import net.pterodactylus.util.database.Parameter.StringParameter;
import net.pterodactylus.util.database.Query;
import net.pterodactylus.util.database.Query.Type;
+import net.pterodactylus.util.database.ResultProcessor;
import net.pterodactylus.util.database.ValueField;
import net.pterodactylus.util.database.ValueFieldWhereClause;
}
/**
+ * Returns all remix artists involved in the track with the given ID.
+ *
+ * @param trackId
+ * The ID of the track
+ * @return All remix artists involved in the track, in preferred order
+ * @throws DatabaseException
+ * if a database error occurs
+ */
+ public List<Artist> getRemixArtistsByTrack(String trackId) throws DatabaseException {
+ Query query = new Query(Type.SELECT, "ARTISTS");
+ query.addField(new Field("ARTISTS.*"));
+ query.addJoin(new Join(JoinType.INNER, "TRACK_REMIX_ARTISTS", new Field("TRACK_REMIX_ARTISTS.ARTIST"), new Field("ARTISTS.ID")));
+ query.addWhereClause(new ValueFieldWhereClause(new ValueField("TRACK_REMIX_ARTISTS.TRACK", new StringParameter(trackId))));
+ query.addOrderField(new OrderField(new Field("TRACK_REMIX_ARTISTS.DISPLAY_ORDER")));
+ return database.getMultiple(query, artistCreator);
+ }
+
+ /**
+ * Returns all related tracks for the track with the given ID.
+ *
+ * @param trackId
+ * The ID of the tracks
+ * @return A mapping from relationship to all tracks that match the relation
+ * @throws DatabaseException
+ * if a database error occurs
+ */
+ public Map<Relationship, Collection<Track>> getRelatedTracksByTrack(String trackId) throws DatabaseException {
+ Query query = new Query(Type.SELECT, "TRACKS");
+ query.addField(new Field("TRACKS.*"));
+ query.addField(new Field("TRACK_RELATIONS.*"));
+ query.addJoin(new Join(JoinType.INNER, "TRACK_RELATIONS", new Field("TRACK_RELATIONS.TRACK"), new Field("TRACK_RELATIONS.RELATED_TRACK")));
+ query.addWhereClause(new ValueFieldWhereClause(new ValueField("TRACK_RELATIONS.TRACK", new StringParameter(trackId))));
+ final Map<Relationship, Collection<Track>> relatedTracks = new EnumMap<Relationship, Collection<Track>>(Relationship.class);
+ database.process(query, new ResultProcessor() {
+
+ @Override
+ @SuppressWarnings("synthetic-access")
+ public void processResult(ResultSet resultSet) throws SQLException {
+ Track track = trackCreator.createObject(resultSet);
+ Relationship relationship = Relationship.valueOf(resultSet.getString("TRACK_RELATIONS.RELATIONSHIP"));
+ if (!relatedTracks.containsKey(relationship)) {
+ relatedTracks.put(relationship, new HashSet<Track>());
+ }
+ relatedTracks.get(relationship).add(track);
+ }
+ });
+ return relatedTracks;
+ }
+
+ /**
* Returns the track with the given ID.
*
* @param id
});
+ /** Memoizer for the remix artists. */
+ private final Memoizer<Void> remixArtistsMemoizer = new Memoizer<Void>(new Callable<Void>() {
+
+ @Override
+ public Void call() throws Exception {
+ if (!hasValue("remixArtists")) {
+ getValue("remixArtists", List.class).set(getRemixArtistsByTrack(getId()));
+ }
+ return null;
+ }
+
+ });
+
+ /** Memoizer for the related tracks. */
+ private final Memoizer<Void> relatedTracksMemoizer = new Memoizer<Void>(new Callable<Void>() {
+
+ @Override
+ public Void call() throws Exception {
+ if (!hasValue("relatedTracks")) {
+ getValue("relatedTracks", Map.class).set(getRelatedTracksByTrack(getId()));
+ }
+ return null;
+ }
+ });
+
/**
* Creates a new track.
*
return super.getStyles();
}
+ /**
+ * {@inheritDoc}
+ */
+ @Override
+ public List<Artist> getRemixArtists() {
+ remixArtistsMemoizer.get();
+ return super.getRemixArtists();
+ }
+
+ /**
+ * {@inheritDoc}
+ */
+ @Override
+ public Map<Relationship, Collection<Track>> getRelatedTracks() {
+ relatedTracksMemoizer.get();
+ return super.getRelatedTracks();
+ }
+
}
/**
*/
@Override
public Track createObject(ResultSet resultSet) throws SQLException {
- return new LazyTrack(resultSet.getString("TRACKS.ID")).setName(resultSet.getString("TRACKS.NAME"));
+ return new LazyTrack(resultSet.getString("TRACKS.ID")).setName(resultSet.getString("TRACKS.NAME")).setRemix(resultSet.getString("TRACKS.REMIX"));
}
}