angry1980.audio.similarity.FindSimilarTracksImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of audio-fingerprint Show documentation
There is a newer version: 0.0.10
package angry1980.audio.similarity;

import angry1980.audio.dao.TrackSimilarityDAO;
import angry1980.audio.model.Fingerprint;
import angry1980.audio.model.FingerprintType;
import angry1980.audio.model.TrackSimilarity;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.Optional;
import java.util.function.LongFunction;
import java.util.stream.Collectors;

public class FindSimilarTracksImpl implements FindSimilarTracks {

    private static Logger LOG = LoggerFactory.getLogger(FindSimilarTracksImpl.class);

    private TrackSimilarityDAO trackSimilarityDAO;
    private LongFunction fingerprintHandler;
    private Calculator calculator;
    private FingerprintType fingerprintType;

    public FindSimilarTracksImpl(TrackSimilarityDAO trackSimilarityDAO,
                                 LongFunction fingerprintHandler,
                                 Calculator calculator,
                                 FingerprintType fingerprintType) {
        this.trackSimilarityDAO = Objects.requireNonNull(trackSimilarityDAO);
        this.fingerprintHandler = Objects.requireNonNull(fingerprintHandler);
        this.calculator = Objects.requireNonNull(calculator);
        this.fingerprintType = Objects.requireNonNull(fingerprintType);
    }

    @Override
    public boolean test(FingerprintType fingerprintType) {
        return this.fingerprintType.equals(fingerprintType);
    }

    @Override
    public List apply(long trackId) {
        return trackSimilarityDAO.findByTrackIdAndFingerprintType(trackId, fingerprintType)
                    .orElseGet(() -> calculate(trackId).stream()
                                        .map(trackSimilarityDAO::create)
                                        .map(Optional::get)
                                        .collect(Collectors.toList())
                    );
    }

    private List calculate(long trackId){
        LOG.debug("Start calculation of {} similarities for track {}", fingerprintType, trackId);
        return Optional.of(trackId)
                .map(fingerprintHandler::apply)
                .map(fingerprint -> calculator.calculate(fingerprint))
                .orElseGet(() -> {
                    LOG.debug("It's not possible to calculate {} similarities for track {}", fingerprintType, trackId);
                    return Collections.emptyList();
                });
    }

    @Override
    public String toString() {
        return "FindSimilarTracks{" +
                "fingerprintType=" + fingerprintType +
                '}';
    }
}