1st iteration

1 files changed, 14 insertions, 4 deletions

diff --git a/InvertedIndex.java b/InvertedIndex.java
index bc75ba6..b2e1fd0 100644
--- a/InvertedIndex.java
+++ b/InvertedIndex.java
@@ -1,3 +1,4 @@
+import java.util.HashMap;
 import java.util.LinkedList;
 import java.util.List;
 import java.util.Map;
@@ -12,12 +13,15 @@ public class InvertedIndex {
             inputs.add(new Document(filename, FileParser.parse(filename)));
         }
         // TODO: instantiate a MapReduce object with correct input, key, value, and output types
+        MapReduce<Document, String, String, List<String>> mapReduce = new MapReduce<>();
 
         // TODO: set the mapper and reducer suppliers, and set the inputs
+        mapReduce.setMapperSupplier(Mapper::new);
+        mapReduce.setReducerSupplier(Reducer::new);
+        mapReduce.setInput(inputs);
 
         // TODO: execute the MapReduce object and return the result
-
-        throw new UnsupportedOperationException("InvertedIndex.run() not implemented yet.");
+        return mapReduce.call();
     }
 
     class Document {
@@ -38,7 +42,11 @@ public class InvertedIndex {
         @Override
         public Map<String, String> compute() {
             // TODO: implement the Map function for inverted index
-            throw new UnsupportedOperationException("InvertedIndex map function not implemented yet.");
+            Map<String, String> map = new HashMap<>();
+            for (String word : input.words) {
+                map.put(word, input.name);
+            }
+            return map;
         }
 
     }
@@ -49,7 +57,9 @@ public class InvertedIndex {
         @Override
         public List<String> compute() {
             // TODO: implement the Reduce function for inverted index
-            throw new UnsupportedOperationException("InvertedIndex reduce function not implemented yet.");
+            List<String> list = new LinkedList<>();
+            list.addAll(valueList);
+            return list;
         }
     }