84 lines
3.0 KiB
Java
84 lines
3.0 KiB
Java
package com.camillepradel.movierecommender.controller;
|
|
|
|
import java.util.Arrays;
|
|
import java.util.LinkedList;
|
|
import java.util.List;
|
|
|
|
import org.springframework.stereotype.Controller;
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
import org.springframework.web.bind.annotation.RequestParam;
|
|
import org.springframework.web.servlet.ModelAndView;
|
|
|
|
import com.camillepradel.movierecommender.model.Genre;
|
|
import com.camillepradel.movierecommender.model.Movie;
|
|
import java.net.ConnectException;
|
|
import java.util.logging.Level;
|
|
import java.util.logging.Logger;
|
|
import org.neo4j.driver.v1.Record;
|
|
import org.neo4j.driver.v1.Session;
|
|
import org.neo4j.driver.v1.StatementResult;
|
|
|
|
@Controller
|
|
public class MainController {
|
|
String message = "Welcome to Spring MVC!";
|
|
|
|
@RequestMapping("/hello")
|
|
public ModelAndView showMessage(
|
|
@RequestParam(value = "name", required = false, defaultValue = "World") String name) {
|
|
System.out.println("in controller");
|
|
|
|
ModelAndView mv = new ModelAndView("helloworld");
|
|
|
|
Session session = null;
|
|
try {
|
|
session =Neo4jConnector.getInstance().getConnection();
|
|
|
|
} catch (ConnectException ex) {
|
|
Logger.getLogger(MainController.class.getName()).log(Level.SEVERE, null, ex);
|
|
}
|
|
StatementResult result = session.run( "MATCH (u:User) RETURN COUNT( DISTINCT u.occupation) AS CPT" );
|
|
while ( result.hasNext() )
|
|
{
|
|
Record record = result.next();
|
|
System.out.println( record.toString() );
|
|
mv.addObject("message", record.get( "CPT" ));
|
|
}
|
|
|
|
try {
|
|
Neo4jConnector.getInstance().close();
|
|
} catch (ConnectException ex) {
|
|
Logger.getLogger(MainController.class.getName()).log(Level.SEVERE, null, ex);
|
|
}
|
|
|
|
|
|
mv.addObject("name", name);
|
|
return mv;
|
|
|
|
|
|
}
|
|
|
|
@RequestMapping("/movies")
|
|
public ModelAndView showMovies(
|
|
@RequestParam(value = "user_id", required = false) Integer userId) {
|
|
System.out.println("show Movies of user " + userId);
|
|
|
|
// TODO: write query to retrieve all movies from DB or all movies rated by user with id userId,
|
|
// depending on whether or not a value was given for userId
|
|
List<Movie> movies = new LinkedList<Movie>();
|
|
Genre genre0 = new Genre(0, "genre0");
|
|
Genre genre1 = new Genre(1, "genre1");
|
|
Genre genre2 = new Genre(2, "genre2");
|
|
movies.add(new Movie(0, "Titre 0", Arrays.asList(new Genre[] {genre0, genre1})));
|
|
movies.add(new Movie(1, "Titre 1", Arrays.asList(new Genre[] {genre0, genre2})));
|
|
movies.add(new Movie(2, "Titre 2", Arrays.asList(new Genre[] {genre1})));
|
|
movies.add(new Movie(3, "Titre 3", Arrays.asList(new Genre[] {genre0, genre1, genre2})));
|
|
|
|
ModelAndView mv = new ModelAndView("movies");
|
|
mv.addObject("userId", userId);
|
|
mv.addObject("movies", movies);
|
|
return mv;
|
|
|
|
|
|
}
|
|
}
|