From 6bb851459d9dbbf11ada10dfea57d86e41f4bb20 Mon Sep 17 00:00:00 2001
From: Quentin Rouland <quentin.rouland@rdrive.ovh>
Date: Fri, 9 Dec 2016 10:30:08 +0100
Subject: [PATCH] mongodb update rating

---
 .../controller/MainController.java            | 23 ++---------
 .../controller/MongoDBController.java         | 41 +++++++++++++++++++
 2 files changed, 44 insertions(+), 20 deletions(-)

diff --git a/src/main/java/com/camillepradel/movierecommender/controller/MainController.java b/src/main/java/com/camillepradel/movierecommender/controller/MainController.java
index 70854db..95589f5 100644
--- a/src/main/java/com/camillepradel/movierecommender/controller/MainController.java
+++ b/src/main/java/com/camillepradel/movierecommender/controller/MainController.java
@@ -13,23 +13,6 @@ import org.springframework.web.bind.annotation.RequestMethod;
 
 import com.camillepradel.movierecommender.model.Genre;
 import com.camillepradel.movierecommender.model.Movie;
-import com.mongodb.BasicDBList;
-import com.mongodb.BasicDBObject;
-import com.mongodb.DBCollection;
-import com.mongodb.DBCursor;
-import com.mongodb.DBObject;
-import com.mongodb.client.FindIterable;
-import com.mongodb.client.MongoCollection;
-import com.mongodb.client.MongoCursor;
-import com.mongodb.client.MongoDatabase;
-import java.util.ArrayList;
-import org.bson.Document;
-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;
 import com.camillepradel.movierecommender.model.Rating;
 
 @Controller
@@ -67,8 +50,8 @@ public class MainController {
 		System.out.println("GET /movieratings for user " + userId);
 
 		// write query to retrieve all movies from DB
-		List<Movie> allMovies = MongoDBController.getMoviesMongoDBByUser(userId);
-                //List<Movie> allMovies =Neo4JController.getMoviesNeo4JByUser(userId);
+		List<Movie> allMovies = MongoDBController.getMoviesMongoDBByUser(null);
+                //List<Movie> allMovies =Neo4JController.getMoviesNeo4JByUser(null);
 
 		// write query to retrieve all ratings from the specified user
 		List<Rating> ratings = MongoDBController.getRatingMongoDBByUser(userId);
@@ -91,7 +74,7 @@ public class MainController {
 		// TODO: add query which
 		//         - add rating between specified user and movie if it doesn't exist
 		//         - update it if it does exist
-
+                MongoDBController.updateMovieRating(rating);
 		return "redirect:/movieratings?user_id=" + rating.getUserId();
 	}
 
diff --git a/src/main/java/com/camillepradel/movierecommender/controller/MongoDBController.java b/src/main/java/com/camillepradel/movierecommender/controller/MongoDBController.java
index 9879beb..c70405b 100644
--- a/src/main/java/com/camillepradel/movierecommender/controller/MongoDBController.java
+++ b/src/main/java/com/camillepradel/movierecommender/controller/MongoDBController.java
@@ -9,10 +9,15 @@ 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.FindIterable;
 import com.mongodb.client.MongoCollection;
 import com.mongodb.client.MongoCursor;
 import com.mongodb.client.MongoDatabase;
+import static com.mongodb.client.model.Filters.eq;
+import com.mongodb.client.model.UpdateOptions;
 import java.util.ArrayList;
+import java.util.Calendar;
+import java.util.Date;
 import java.util.HashMap;
 import java.util.List;
 import org.bson.Document;
@@ -114,4 +119,40 @@ public class MongoDBController {
             }
             return listRating;
         }
+
+    public static void updateMovieRating(Rating rating) {
+        ArrayList<Rating> listRating= new  ArrayList<Rating>();
+        MongoCursor<Document> cursor;
+
+        MongoDatabase db = MongoDBConnector.getInstance().getConnexion();
+        MongoCollection<Document> users = db.getCollection("users");
+
+        BasicDBObject updateQuery = new BasicDBObject();
+        BasicDBObject newDocument = new BasicDBObject();
+        BasicDBObject deleteDocument = new BasicDBObject();
+        
+        deleteDocument.append("$pull", new BasicDBObject()
+            .append("movies", new BasicDBObject()
+                .append("movieid", rating.getMovieId())
+            )
+        );
+        
+        newDocument.append("$push", new BasicDBObject()
+            .append("movies", new BasicDBObject()
+                .append("movieid", rating.getMovieId())
+                .append("rating", rating.getScore())
+                .append("timestamp", (int) (new Date().getTime()/1000))
+            )
+        );
+        
+        BasicDBObject searchQuery = new BasicDBObject()
+            .append("_id", rating.getUserId());
+        
+        FindIterable<Document> movie = users.find(searchQuery);
+        
+
+        users.updateOne(searchQuery,deleteDocument);
+        users.updateOne(searchQuery, newDocument);
+        
+    }
 }