101 lines
4.0 KiB
Java
101 lines
4.0 KiB
Java
/*
|
|
* To change this license header, choose License Headers in Project Properties.
|
|
* To change this template file, choose Tools | Templates
|
|
* and open the template in the editor.
|
|
*/
|
|
package com.camillepradel.movierecommender.controller;
|
|
|
|
import com.camillepradel.movierecommender.model.Genre;
|
|
import com.camillepradel.movierecommender.model.Movie;
|
|
import com.camillepradel.movierecommender.model.Rating;
|
|
import java.net.ConnectException;
|
|
import java.util.LinkedList;
|
|
import java.util.List;
|
|
import java.util.logging.Level;
|
|
import java.util.logging.Logger;
|
|
import org.neo4j.driver.v1.Record;
|
|
import org.neo4j.driver.v1.StatementResult;
|
|
|
|
/**
|
|
*
|
|
* @author renando
|
|
*/
|
|
public class Neo4JController {
|
|
|
|
public static List<Movie>getMoviesNeo4JByUser (Integer userId){
|
|
List<Movie> movies = new LinkedList<Movie>();
|
|
int id=0;
|
|
StatementResult result = null;
|
|
try {
|
|
if (userId != null && userId >=0 ){
|
|
|
|
result = Neo4jConnector.getInstance().getConnection().run( "MATCH (me:User { id: "+userId+" })-[:RATED]->(movie:Movie) -[:CATEGORIZED_AS]->(g:Genre) RETURN movie.title as movies, collect(g.name) as genre;" );
|
|
}
|
|
else{
|
|
result = Neo4jConnector.getInstance().getConnection().run( "MATCH (m:Movie) -[:CATEGORIZED_AS]->(g:Genre) RETURN m.title as movies, collect(g.name) as genre;" );
|
|
}
|
|
} catch (ConnectException ex) {
|
|
Logger.getLogger(MainController.class.getName()).log(Level.SEVERE, null, ex);
|
|
}
|
|
finally{
|
|
Neo4jConnector.getInstance().close();
|
|
}
|
|
while ( result.hasNext() )
|
|
{
|
|
Record record = result.next();
|
|
String[] listGenre = record.get("genre").toString().split(",");
|
|
List<Genre> listTypeGenre = new LinkedList<Genre>();
|
|
for(String g : listGenre){
|
|
if(g != null)
|
|
listTypeGenre.add(new Genre(id, g.replace("\"", "").replace("[", "").replace("]", "")));
|
|
|
|
}
|
|
movies.add(new Movie(id, record.get("movies").asString(), (listTypeGenre.isEmpty())?null:listTypeGenre ) );
|
|
id+=1;
|
|
}
|
|
return movies;
|
|
}
|
|
|
|
|
|
public static List<Rating>GetMoviesRatingNeo4JByUser(Integer userId){
|
|
List<Rating> moviesRating = new LinkedList<Rating>();
|
|
|
|
StatementResult result = null;
|
|
try {
|
|
if (userId != null && userId >=0 ){
|
|
|
|
result = Neo4jConnector.getInstance().getConnection().run( "MATCH (u:User{id:"+userId+"})-[r:RATED]->(m:Movie)-[:CATEGORIZED_AS]->(g:Genre) RETURN m.id AS id, m.title AS title, collect(g.name) AS genre, r.note AS note ORDER BY id" );
|
|
}
|
|
else{
|
|
return null;
|
|
}
|
|
} catch (ConnectException ex) {
|
|
Logger.getLogger(MainController.class.getName()).log(Level.SEVERE, null, ex);
|
|
}
|
|
finally{
|
|
Neo4jConnector.getInstance().close();
|
|
}
|
|
Movie movie = null;
|
|
Rating rate = null;
|
|
|
|
while ( result.hasNext() )
|
|
{
|
|
Record record = result.next();
|
|
System.out.println(record.get("p").toString());
|
|
String[] listGenre = record.get("genre").toString().split(",");
|
|
List<Genre> listTypeGenre = new LinkedList<Genre>();
|
|
for(String g : listGenre){
|
|
if(g != null)
|
|
listTypeGenre.add(new Genre(0, g.replace("\"", "").replace("[", "").replace("]", "")));
|
|
|
|
}
|
|
movie = new Movie(record.get("id").asInt(), record.get("title").asString(),listTypeGenre );
|
|
rate = new Rating(movie, userId, record.get("note").asInt());
|
|
moviesRating.add(rate);
|
|
}
|
|
return moviesRating;
|
|
|
|
}
|
|
|
|
}
|