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

Timestamp:

2020-01-31T23:49:11+13:00 (4 years ago)

Author:

ak19

Message:

1. Added support for writing out tables in csv format too. 2. Second table written out now. 3. Moved getFilePath() into Utility.

File:

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

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

@@ -17 +17 @@
 import static com.mongodb.client.model.Accumulators.*;
 
-
 //import org.bson.conversions.Bson;
 import com.mongodb.BasicDBObject;
@@ -57 +56 @@
 import org.greenstone.atea.morphia.*;
 import dev.morphia.*;
+
+import org.apache.commons.csv.*;
 
 /** 
@@ -83 +84 @@
  *   - https://www.mongodb.com/blog/post/6-rules-of-thumb-for-mongodb-schema-design-part-1
  *
+ * API:
+ * - https://mongodb.github.io/mongo-java-driver/3.4/javadoc/?com/mongodb/client/MongoCollection.html#find--
+ * - examples: https://mongodb.github.io/mongo-java-driver/3.4/driver/getting-started/quick-start/
  */
 public class MongoDBAccess implements AutoCloseable {
@@ -516 +520 @@
     /** Do the aggregates for writing out tables.
        Table1:
-       db.Websites.aggregate([
        
-       { $unwind: "$geoLocationCountryCode" },
-       {
-       $group: {
-       _id: "$geoLocationCountryCode",
-       count: { $sum: 1 },
-       //domain: { $addToSet: '$domain' },
-       numPagesInMRICount: { $sum: '$numPagesInMRI' },
-       numPagesContainingMRICount: { $sum: '$numPagesContainingMRI' }
-       }
-       },
-       { $sort : { count : -1} }
-       ]);
     */
     public void writeTables(File outFolder) {
@@ -535 +526 @@
     MongoCollection<Document> collection = this.database.getCollection(WEBSITES_COLLECTION);
 
-    // table 1
-    File outFile = new File(outFolder, "1table_allCrawledSites.csv");
-    try (
-         Writer writer = new BufferedWriter(new FileWriter(outFile));
-         ) {
+    String[] tableNames = { "", "1table_allCrawledSites", "2table_sitesWithPagesInMRI"};
+    for (int tableNum = 1; tableNum < tableNames.length; tableNum++) {
+        File outFile = new File(outFolder, tableNames[tableNum] + ".json");
+        File csvFile = new File(outFolder, tableNames[tableNum] + ".csv");
+        try (
+         Writer writer = new BufferedWriter(new FileWriter(outFile));
+         CSVPrinter csvWriter = new CSVPrinter(new FileWriter(csvFile), CSVFormat.DEFAULT);
+         ) {
+
+        // Write out the CSV column headings
+        // https://commons.apache.org/proper/commons-csv/apidocs/org/apache/commons/csv/CSVPrinter.html
+        csvWriter.printRecord("countryCode", "siteCount",
+              "numPagesInMRI count","numPagesContainingMRICount"/*, "domain"*/);
+        
+        AggregateIterable<Document> output = getTable(collection, tableNum); //doTable1().forEach((Block<Document>)doc -> writeDoc(doc, writer));
+        
+        int docNum = 0;
+        for (Document doc : output) {       
+            //System.out.println(doc);
+            writeDocAsJsonRecord(++docNum, doc, writer);
+            writeDocAsCSVRecord(++docNum, doc, csvWriter);
+        }       
+        logger.info("@@@ Wrote out table into file: " + Utility.getFilePath(outFile) + " and .csv");
+        } catch(UncheckedIOException ioe) {
+        logger.error("Caught UncheckedIOException: " + ioe.getMessage(), ioe);
+        }
+        catch(Exception e) {
+        logger.error("Could not write table to file " + outFile + " or .csv equivalent" , e);
+        }
+    }
+    }
+
+    public AggregateIterable<Document> getTable(MongoCollection<Document> collection, int tableNum) {
+    
+    AggregateIterable<Document> output = null;
+
+    switch(tableNum) {
         
-        
-        AggregateIterable<Document> output = collection.aggregate(Arrays.asList(
+    case 1:
+        /* 1table_allCrawledSites -
+
+           db.Websites.aggregate([       
+           { $unwind: "$geoLocationCountryCode" },
+           {
+           $group: {
+           _id: "$geoLocationCountryCode",
+           count: { $sum: 1 },
+           //domain: { $addToSet: '$domain' },
+           numPagesInMRICount: { $sum: '$numPagesInMRI' },
+           numPagesContainingMRICount: { $sum: '$numPagesContainingMRI' }
+           }
+           },
+           { $sort : { count : -1} }
+           ]);
+         */
+        output = collection.aggregate(Arrays.asList(
                            //match(BasicDBObject.parse("{urlContainsLangCodeInPath:true}")),
          unwind("$geoLocationCountryCode"),
@@ -550 +589 @@
                                 sum("numPagesInMRICount", "$numPagesInMRI"),
                                 sum("numPagesContainingMRICount", "$numPagesContainingMRI"))),
-         sort(BasicDBObject.parse("{count : -1}"))
-        ));//.forEach((Block<Document>)doc -> writeDoc(doc, writer));
-
-        int docNum = 0;
-        for (Document doc : output) {       
-        //System.out.println(doc);
-        writeDoc(++docNum, doc, writer);        
-        }
+         sort(BasicDBObject.parse("{count : -1}"))       
+        ));
+        break;
+
+    case 2:
+        /*
+          db.Websites.aggregate([
+          { $match: { numPagesInMRI: {$gt: 0} } },
+          { $unwind: "$geoLocationCountryCode" },
+          {
+          $group: {
+          _id: {$toLower: '$geoLocationCountryCode'}, // ignore toLower
+          count: { $sum: 1 },
+          //domain: { $addToSet: '$domain' },
+          numPagesInMRICount: { $sum: '$numPagesInMRI' },
+          numPagesContainingMRICount: { $sum: '$numPagesContainingMRI' }
+          }
+          },
+          { $sort : { count : -1} }
+          ]);
+         */
+         output = collection.aggregate(Arrays.asList(
+                           match(BasicDBObject.parse("{ numPagesInMRI: {$gt: 0} }")),
+         unwind("$geoLocationCountryCode"),
+         group("$geoLocationCountryCode", Arrays.asList(
+                                sum("count", 1),
+                                /*addToSet("domain", "$domain"),*/
+                                sum("numPagesInMRICount", "$numPagesInMRI"),
+                                sum("numPagesContainingMRICount", "$numPagesContainingMRI"))),
+         sort(BasicDBObject.parse("{count : -1}"))       
+        ));
+        break;
         
+    default: logger.error("Unknown table number: " + tableNum);
         
-    } catch(UncheckedIOException ioe) {
-        logger.error("Caught UncheckedIOException: " + ioe.getMessage(), ioe);
-    }
-    catch(Exception e) {
-        logger.error("Could not write table to file " + outFile, e);
-    }
-    }
-
-    public void doTable1() {
+    }
+
+     return output;
     
     }
@@ -606 +664 @@
     }       
     }
-    public void writeDoc(int docNum, Document doc, Writer writer) throws UncheckedIOException {
+    
+    public void writeDocAsJsonRecord(int docNum, Document doc, Writer writer) throws UncheckedIOException {
     String jsonStr = prettyPrintJson(doc.toJson());
     //System.err.println(jsonStr);
@@ -616 +675 @@
     }
     }
-    
+
+    // TODO
+    //public void writeDocToJsonAndCSV(int docNum, Document doc, Writer writer, CSVPrinter csvWriter) throws UncheckedIOException {
+    public void writeDocAsCSVRecord(int docNum, Document doc, CSVPrinter csvWriter) throws UncheckedIOException {
+    String jsonStr = doc.toJson();
+    JsonParser parser = new JsonParser();
+    JsonElement json = parser.parse(jsonStr);
+
+    JsonObject jsonObj = (JsonObject)json;
+    
+    String countryCode = jsonObj.get("_id").getAsString();
+    int siteCount = jsonObj.get("count").getAsInt();
+    int numPagesInMRICount = jsonObj.get("numPagesInMRICount").getAsInt();
+    int numPagesContainingMRICount = jsonObj.get("numPagesContainingMRICount").getAsInt();  
+    
+    //System.err.println(jsonStr);
+    try {
+        //writer.write("/* " + docNum + " */\n" + prettyPrintJson(jsonStr) + NEWLINE);
+        csvWriter.printRecord(countryCode, siteCount, numPagesInMRICount, numPagesContainingMRICount);
+    } catch (IOException ex) {
+        //throw ex;
+        throw new UncheckedIOException(ex);
+    }
+    }
     
     public String prettyPrintJson(String jsonStr) {