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

mongodb update rating

Browse Source
This commit is contained in:
Quentin Rouland 2016-12-09 10:30:08 +01:00
parent e9627fa59a
commit 6bb851459d
2 changed files with 44 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/main/java/com/camillepradel/movierecommender/controller/MainController.java
View File

@ -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();
}

41
src/main/java/com/camillepradel/movierecommender/controller/MongoDBController.java
View File

@ -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);
}
}