angry1980.audio.ImportFromNetflixToNeo4j Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of audio-app Show documentation

There is a newer version: 0.0.10

package angry1980.audio;

import angry1980.audio.model.FingerprintType;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.context.annotation.Import;

@SpringBootApplication
@Import(ImportFromNetflixToNeo4jConfig.class)
public class ImportFromNetflixToNeo4j {

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

    @Autowired
    private DataImporter dataImporter;
    @Autowired
    private DataImporter.TrackDataEnvironment sourceEnvironment;

    public static void main(String[] args){
        SpringApplication sa = new SpringApplication(ImportFromNetflixToNeo4j.class);
        sa.setAdditionalProfiles(
                FingerprintType.CHROMAPRINT.name(),
                FingerprintType.PEAKS.name(),
                FingerprintType.LASTFM.name(),
                "NEO4J",
                "NETFLIX"
        );
        ConfigurableApplicationContext context = sa.run(args);
        context.registerShutdownHook();
        context.getBean(ImportFromNetflixToNeo4j.class)
                .importData(FingerprintType.CHROMAPRINT)
                //.importData(FingerprintType.LASTFM, FingerprintType.LASTFM_ER)
                //.importData(FingerprintType.PEAKS)
        ;
    }

    public ImportFromNetflixToNeo4j importData(FingerprintType type){
        return importData(type, type);
    }

    public ImportFromNetflixToNeo4j importData(FingerprintType type, FingerprintType goal){
        dataImporter.importTo(sourceEnvironment, type, goal);
        LOG.info("Similarities for {} was imported", type);
        return this;
    }

}