Search in sources :

Example 6 with ArtistRelation

use of com.github.hakko.musiccabinet.domain.model.music.ArtistRelation in project musiccabinet by hakko.

the class JdbcArtistRelationDao method getArtistRelations.

@Override
public List<ArtistRelation> getArtistRelations(Artist sourceArtist) {
    final int sourceArtistId = jdbcTemplate.queryForInt("select * from music.get_artist_id(?)", sourceArtist.getName());
    String sql = "select artist_name_capitalization, weight" + " from music.artistrelation" + " inner join music.artist on music.artistrelation.target_id = music.artist.id" + " where music.artistrelation.source_id = ?";
    List<ArtistRelation> artistRelations = jdbcTemplate.query(sql, new Object[] { sourceArtistId }, new RowMapper<ArtistRelation>() {

        @Override
        public ArtistRelation mapRow(ResultSet rs, int rowNum) throws SQLException {
            String artistName = rs.getString(1);
            float match = rs.getFloat(2);
            return new ArtistRelation(new Artist(artistName), match);
        }
    });
    return artistRelations;
}
Also used : ArtistRelation(com.github.hakko.musiccabinet.domain.model.music.ArtistRelation) Artist(com.github.hakko.musiccabinet.domain.model.music.Artist) SQLException(java.sql.SQLException) ResultSet(java.sql.ResultSet)

Example 7 with ArtistRelation

use of com.github.hakko.musiccabinet.domain.model.music.ArtistRelation in project musiccabinet by hakko.

the class JdbcArtistRelationDao method getSimilarArtistsByTags.

/*
	 * TODO
	 *
	 * on "Related artists" page, add an "Edit" possibility
	 * Display top tags for main artist
	 * Allow user to select which ones to use
	 * Return matching artists (see below)
	 * Add selected matching artists as related
	 *
	 * somehow mark as updated to avoid overwriting
	 */
public List<ArtistRelation> getSimilarArtistsByTags(int artistId) {
    String topTagsTable = settingsService.getArtistTopTagsTable();
    String topTagsSql = "select tag_id, tag_count from " + topTagsTable + " where artist_id = " + artistId + " order by tag_count desc limit 5";
    String similarityTableSql = "create temporary table similarity" + " (artist_id integer not null, count integer not null";
    String insertSimilaritySql = "insert into similarity (artist_id, count)" + " select artist_id, 100-abs(?-tag_count) from" + " " + topTagsTable + " where tag_id = ?";
    String topArtistsSql = "select a.artist_name_capitalization, m.sum/(100.0*?) from music.artist a" + " inner join (select artist_id, sum(count) from similarity group by artist_id) s" + " on a.id = s.artist_id order by m.sum desc";
    List<TagCount> topTags = jdbcTemplate.query(topTagsSql, new RowMapper<TagCount>() {

        @Override
        public TagCount mapRow(ResultSet rs, int rowNum) throws SQLException {
            return new TagCount(rs.getInt(1), rs.getInt(2));
        }
    });
    jdbcTemplate.execute(similarityTableSql);
    for (TagCount topTag : topTags) {
        jdbcTemplate.update(insertSimilaritySql, topTag.tagCount, topTag.tagId);
    }
    return jdbcTemplate.query(topArtistsSql, new Object[] { topTags.size() }, new RowMapper<ArtistRelation>() {

        @Override
        public ArtistRelation mapRow(ResultSet rs, int rowNum) throws SQLException {
            return new ArtistRelation(new Artist(rs.getString(1)), rs.getFloat(2));
        }
    });
}
Also used : ArtistRelation(com.github.hakko.musiccabinet.domain.model.music.ArtistRelation) Artist(com.github.hakko.musiccabinet.domain.model.music.Artist) SQLException(java.sql.SQLException) ResultSet(java.sql.ResultSet)

Example 8 with ArtistRelation

use of com.github.hakko.musiccabinet.domain.model.music.ArtistRelation in project musiccabinet by hakko.

the class JdbcArtistRelationDao method batchInsert.

private void batchInsert(Artist sourceArtist, List<ArtistRelation> ArtistRelations) {
    int sourceArtistId = jdbcTemplate.queryForInt("select * from music.get_artist_id(?)", sourceArtist.getName());
    String sql = "insert into music.artistrelation_import (source_id, target_artist_name, weight) values (?,?,?)";
    BatchSqlUpdate batchUpdate = new BatchSqlUpdate(jdbcTemplate.getDataSource(), sql);
    batchUpdate.setBatchSize(1000);
    batchUpdate.declareParameter(new SqlParameter("source_id", Types.INTEGER));
    batchUpdate.declareParameter(new SqlParameter("target_artist_name", Types.VARCHAR));
    batchUpdate.declareParameter(new SqlParameter("weight", Types.FLOAT));
    for (ArtistRelation ar : ArtistRelations) {
        batchUpdate.update(new Object[] { sourceArtistId, ar.getTarget().getName(), ar.getMatch() });
    }
    batchUpdate.flush();
}
Also used : ArtistRelation(com.github.hakko.musiccabinet.domain.model.music.ArtistRelation) SqlParameter(org.springframework.jdbc.core.SqlParameter) BatchSqlUpdate(org.springframework.jdbc.object.BatchSqlUpdate)

Aggregations

ArtistRelation (com.github.hakko.musiccabinet.domain.model.music.ArtistRelation)8 Artist (com.github.hakko.musiccabinet.domain.model.music.Artist)3 Test (org.junit.Test)3 ResultSet (java.sql.ResultSet)2 SQLException (java.sql.SQLException)2 TagTopArtists (com.github.hakko.musiccabinet.domain.model.aggr.TagTopArtists)1 File (com.github.hakko.musiccabinet.domain.model.library.File)1 ArtistInfo (com.github.hakko.musiccabinet.domain.model.music.ArtistInfo)1 Track (com.github.hakko.musiccabinet.domain.model.music.Track)1 ResourceUtil (com.github.hakko.musiccabinet.util.ResourceUtil)1 UnittestLibraryUtil.getFile (com.github.hakko.musiccabinet.util.UnittestLibraryUtil.getFile)1 ArrayList (java.util.ArrayList)1 Before (org.junit.Before)1 SqlParameter (org.springframework.jdbc.core.SqlParameter)1 BatchSqlUpdate (org.springframework.jdbc.object.BatchSqlUpdate)1