QRouland/M2MovieRecommander
Archived
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add mongo controller

Browse Source
This commit is contained in:
Quentin Rouland 2016-12-01 10:12:17 +01:00
parent a72a92de3c
commit 9f847c9e75
1 changed files with 117 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

117
src/main/java/com/camillepradel/movierecommender/controller/MongoDBController.java Normal file
View File

@ -0,0 +1,117 @@
/*
* 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 com.mongodb.BasicDBObject;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoCursor;
import com.mongodb.client.MongoDatabase;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import org.bson.Document;
/**
*
* @author sidya
*/
public class MongoDBController {
public static List<Movie>getMoviesMongoDBByUser(Integer user_id){
ArrayList<Movie> listMovies= new ArrayList<Movie>();
MongoCursor<Document> cursor;
MongoDatabase db = MongoDBConnector.getInstance().getConnexion();
MongoCollection<Document> movies = db.getCollection("movies");
if(user_id != null)
{
MongoCollection<Document> users = db.getCollection("users");
BasicDBObject whereQuery = new BasicDBObject();
whereQuery.put("_id", user_id);
cursor = users.find(whereQuery).iterator();
Document user;
try {
user = cursor.next();
} catch(Exception e)
{
return new ArrayList<Movie>();
}
ArrayList<Document> user_movies;
user_movies = (ArrayList) user.get("movies");
BasicDBObject inQuery = new BasicDBObject();
List<Integer> list = new ArrayList<Integer>();
for(Document movie : user_movies)
list.add(movie.getInteger("movieid"));
inQuery.put("_id", new BasicDBObject("$in", list));
cursor = movies.find(inQuery).iterator();
}
else
{
cursor = movies.find().iterator();
}
while (cursor.hasNext()) {
Document movie = cursor.next();
String[] genres;
genres = movie.getString("genres").split("\\|");
ArrayList<Genre> listGenres = new ArrayList<Genre>();
for(String genre : genres){
listGenres.add(new Genre(1,genre));
}
listMovies.add(new Movie(movie.getInteger("_id"), movie.getString("title"), listGenres));
}
return listMovies;
}
public static List<Rating>getRatingMongoDBByUser(Integer user_id){
ArrayList<Rating> listRating= new ArrayList<Rating>();
MongoCursor<Document> cursor;
MongoDatabase db = MongoDBConnector.getInstance().getConnexion();
MongoCollection<Document> users = db.getCollection("users");
MongoCollection<Document> movies = db.getCollection("movies");
BasicDBObject whereQuery = new BasicDBObject();
whereQuery.put("_id", user_id);
cursor = users.find(whereQuery).iterator();
Document user;
try {
user = cursor.next();
} catch(Exception e)
{
return new ArrayList<Rating>();
}
ArrayList<Document> user_movies = (ArrayList) user.get("movies");
BasicDBObject inQuery = new BasicDBObject();
HashMap<Integer,Integer> list = new HashMap<Integer,Integer>();
for(Document movie : user_movies)
list.put(movie.getInteger("movieid"),movie.getInteger("rating"));
inQuery.put("_id", new BasicDBObject("$in", list.keySet()));
cursor = movies.find(inQuery).iterator();
while (cursor.hasNext()) {
Document movie = cursor.next();
String[] genres;
genres = movie.getString("genres").split("\\|");
ArrayList<Genre> listGenres = new ArrayList<Genre>();
for(String genre : genres){
listGenres.add(new Genre(1,genre));
}
listRating.add(new Rating(new Movie(movie.getInteger("_id"), movie.getString("title"),
listGenres),user_id, list.get(movie.getInteger("_id"))));
}
return listRating;
}
}