mirror of https://github.com/bvn13/imdb-spider.git
79 lines
2.7 KiB
Java
79 lines
2.7 KiB
Java
package ru.bvn13.imdbspider;
|
|
|
|
import ru.bvn13.imdbspider.exceptions.ImdbSpiderException;
|
|
import ru.bvn13.imdbspider.imdb.MovieDataType;
|
|
import ru.bvn13.imdbspider.imdb.MovieList;
|
|
import ru.bvn13.imdbspider.spider.api.v1_0.ApiFactory_1_0;
|
|
import ru.bvn13.imdbspider.spider.composer.ImdbObjectComposerFactory;
|
|
import ru.bvn13.imdbspider.spider.composer.MovieListComposer;
|
|
import ru.bvn13.imdbspider.spider.processor.JsoupHtmlProcessor;
|
|
import ru.bvn13.imdbspider.spider.tasker.Manager;
|
|
import ru.bvn13.imdbspider.spider.tasker.Task;
|
|
import ru.bvn13.imdbspider.spider.api.ApiFactory;
|
|
|
|
import java.time.Duration;
|
|
import java.time.LocalDateTime;
|
|
import java.util.*;
|
|
|
|
/**
|
|
* @author boyko_vn at 09.01.2019
|
|
*/
|
|
public class ImdbSpider {
|
|
|
|
private Manager manager;
|
|
|
|
private ApiFactory apiFactory;
|
|
private ImdbObjectComposerFactory imdbObjectComposerFactory;
|
|
|
|
public static ImdbSpider withApi_1_0() {
|
|
ApiFactory apiFactory = new ApiFactory_1_0(new JsoupHtmlProcessor());
|
|
return new ImdbSpider(apiFactory, new ImdbObjectComposerFactory(apiFactory));
|
|
}
|
|
|
|
|
|
public ImdbSpider(ApiFactory apiFactory, ImdbObjectComposerFactory imdbObjectComposerFactory) {
|
|
this.apiFactory = apiFactory;
|
|
this.imdbObjectComposerFactory = imdbObjectComposerFactory;
|
|
|
|
manager = new Manager();
|
|
}
|
|
|
|
public ImdbSpider addHttpRequestHeader(String key, String value) {
|
|
manager.addHttpRequestHeader(key, value);
|
|
return this;
|
|
}
|
|
|
|
public MovieList searchMovieByTitle(String title) throws ImdbSpiderException {
|
|
return searchMovieByTitle(title, 0);
|
|
}
|
|
|
|
public MovieList searchMovieByTitle(String title, int maxCount) throws ImdbSpiderException {
|
|
return searchMovieByTitle(title, maxCount, EnumSet.of(MovieDataType.TITLE));
|
|
}
|
|
|
|
public MovieList searchMovieByTitle(String title, int maxCount, MovieDataType... dataTypes) throws ImdbSpiderException {
|
|
return searchMovieByTitle(title, maxCount, EnumSet.copyOf(Arrays.asList(dataTypes)));
|
|
}
|
|
|
|
public MovieList searchMovieByTitle(String title, int maxCount, EnumSet<MovieDataType> dataTypes) throws ImdbSpiderException {
|
|
|
|
List<Task> tasks = apiFactory.createTasksForSearchMovieByTitle(title, maxCount, dataTypes);
|
|
|
|
LocalDateTime dateStart = LocalDateTime.now();
|
|
manager.processTasks(tasks);
|
|
LocalDateTime dateEnd = LocalDateTime.now();
|
|
Duration diff = Duration.between(dateStart, dateEnd);
|
|
System.out.println("TIME SPENT: "+(diff.toMillis())+" msec");
|
|
|
|
|
|
MovieListComposer movieListComposer = (MovieListComposer) imdbObjectComposerFactory.getComposer(MovieList.class);
|
|
MovieList movieList = movieListComposer.compose(tasks.get(0));
|
|
|
|
return movieList;
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|