Merge pull request #218 from HubSpot/group-by-raw

Add raw object to group in groupby filter.
HubSpot · Aug 10, 2018 · 97d1219 · 97d1219
2 parents 877760b + 7011934
commit 97d1219
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/src/main/java/com/hubspot/jinjava/lib/filter/GroupByFilter.java b/src/main/java/com/hubspot/jinjava/lib/filter/GroupByFilter.java
@@ -1,7 +1,9 @@
 package com.hubspot.jinjava.lib.filter;
 
 import java.util.ArrayList;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 import java.util.Objects;
 
 import com.google.common.collect.LinkedListMultimap;
@@ -50,36 +52,48 @@ public Object filter(Object var, JinjavaInterpreter interpreter, String... args)
 
     ForLoop loop = ObjectIterator.getLoop(var);
     Multimap<String, Object> groupBuckets = LinkedListMultimap.create();
+    Map<String, Object> groupMapRaw = new HashMap<>();
 
     while (loop.hasNext()) {
       Object val = loop.next();
 
-      String grouper = Objects.toString(interpreter.resolveProperty(val, attr));
+      Object resolvedProperty = interpreter.resolveProperty(val, attr);
+      String grouper = Objects.toString(resolvedProperty);
       groupBuckets.put(grouper, val);
+
+      if (!groupMapRaw.containsKey(grouper)) {
+        groupMapRaw.put(grouper, resolvedProperty);
+      }
     }
 
     List<Group> groups = new ArrayList<>();
     for (String grouper : groupBuckets.keySet()) {
       List<Object> list = Lists.newArrayList(groupBuckets.get(grouper));
-      groups.add(new Group(grouper, list));
+      groups.add(new Group(grouper, groupMapRaw.get(grouper), list));
     }
 
     return groups;
   }
 
   public static class Group {
     private final String grouper;
+    private final Object grouperObject;
     private final List<Object> list;
 
-    public Group(String grouper, List<Object> list) {
+    public Group(String grouper, Object grouperObject, List<Object> list) {
       this.grouper = grouper;
+      this.grouperObject = grouperObject;
       this.list = list;
     }
 
     public String getGrouper() {
       return grouper;
     }
 
+    public Object getGrouperObject() {
+      return grouperObject;
+    }
+
     public List<Object> getList() {
       return list;
     }