Changeset 33645 for other-projects/maori-lang-detection/src/org/greenstone/atea/MongoDBAccess.java

Timestamp:

2019-11-11T18:45:29+13:00 (4 years ago)

Author:

ak19

Message:

Fix to 2 bugs when sending data to MongoDB: 1. overlappingSentences was missing because overwritten by singleSentences. But it is still empty at present because of a bug. 2. Need to store an array of tuples not with each pair of a tuple stored as an array element.

File:

• other-projects/maori-lang-detection/src/org/greenstone/atea/MongoDBAccess.java (modified) (2 diffs)

other-projects/maori-lang-detection/src/org/greenstone/atea/MongoDBAccess.java

@@ -1 +1 @@
 package org.greenstone.atea;
 
-
+//import org.bson.BSONObject;
+    
 import com.mongodb.client.MongoCollection; 
 import com.mongodb.client.MongoDatabase; 
@@ -201 +202 @@
     // https://stackoverflow.com/questions/15371839/how-to-add-an-array-to-a-mongodb-document-using-java
     List<BasicDBObject> sentencesList = new ArrayList<>();
-    for(SentenceInfo sentence : webpage.singleSentences) {
-        sentencesList.add(new BasicDBObject("langCode", sentence.langCode));
-        sentencesList.add(new BasicDBObject("confidence", sentence.confidenceLevel));
-        sentencesList.add(new BasicDBObject("sentence", sentence));
+    for(SentenceInfo sentenceInfo : webpage.singleSentences) {
+        //sentencesList.add(new BasicDBObject("langCode", sentenceInfo.langCode));
+        //sentencesList.add(new BasicDBObject("confidence", sentenceInfo.confidenceLevel));
+        //sentencesList.add(new BasicDBObject("sentence", sentenceInfo.sentence));
+        BasicDBObject bsonRecord = new BasicDBObject("langCode", sentenceInfo.langCode);
+        bsonRecord.put("confidence", sentenceInfo.confidenceLevel);
+        bsonRecord.put("sentence", sentenceInfo.sentence);
+        
+        sentencesList.add(bsonRecord);
     }   
     document.put("singleSentences", sentencesList);
     
     List<BasicDBObject> overlappingSentencesList = new ArrayList<>();
-    for(SentenceInfo sentence : webpage.overlappingSentences) {
-        sentencesList.add(new BasicDBObject("langCode", sentence.langCode));
-        sentencesList.add(new BasicDBObject("confidence", sentence.confidenceLevel));
-        sentencesList.add(new BasicDBObject("sentence", sentence));
+    for(SentenceInfo sentenceInfo : webpage.overlappingSentences) {
+        BasicDBObject bsonRecord = new BasicDBObject("langCode", sentenceInfo.langCode);
+        bsonRecord.put("confidence", sentenceInfo.confidenceLevel);
+        bsonRecord.put("sentence", sentenceInfo.sentence);
+        
+        sentencesList.add(bsonRecord);
     }   
-    document.put("singleSentences", overlappingSentencesList);
-
+    document.put("overlappingSentences", overlappingSentencesList);
+    
     // also put the full text in there
     document.put("text", webpage.text);