Last active
September 21, 2020 14:15
-
-
Save spullara/4752663 to your computer and use it in GitHub Desktop.
JDK 8 experiment where I recreate Excel pivot tables for collections of structured objects.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
@Test | |
public void testPivotTable() { | |
List<Row> rows = new ArrayList<>(); | |
rows.add(new Row("East", "Boy", "Tee", 10, 12.00)); | |
rows.add(new Row("East", "Boy", "Golf", 15, 20.00)); | |
rows.add(new Row("East", "Girl", "Tee", 8, 14.00)); | |
rows.add(new Row("East", "Girl", "Golf", 20, 24.00)); | |
rows.add(new Row("West", "Boy", "Tee", 5, 12.00)); | |
rows.add(new Row("West", "Boy", "Golf", 12, 20.00)); | |
rows.add(new Row("West", "Girl", "Tee", 15, 14.00)); | |
rows.add(new Row("West", "Girl", "Golf", 10, 24.00)); | |
// groupBy region and gender, summing total sales | |
Map<String, Map<String, Double>> pivot = pivot(rows, | |
(r) -> r.region, | |
(r) -> r.gender, | |
(rs) -> rs.stream().map((row) -> row.price * row.units).reduce(0.0, (a, b) -> a + b)); | |
System.out.println(pivot); | |
} | |
public static <K, V, W> Map<K, W> transformValues(Map<K, V> oldmap, Function<V, W> transform) { | |
return oldmap.entrySet().stream().collect(HashMap::new, | |
(map, entry) -> map.put(entry.getKey(), transform.apply(entry.getValue())), | |
Map::putAll); | |
} | |
public static <K1, K2, IN, OUT> Map<K1, Map<K2, OUT>> pivot(Collection<IN> rows, Function<IN, K1> left, Function<IN, K2> top, Function<Collection<IN>, OUT> aggregate) { | |
Map<K1, Collection<IN>> leftGroup = rows.stream().collect(groupingBy(left)); | |
Map<K1, Map<K2, Collection<IN>>> grid = transformValues(leftGroup, (rs) -> rs.stream().collect(groupingBy(top))); | |
return transformValues(grid, (map) -> transformValues(map, aggregate)); | |
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
@Test | |
public void testPivotTable() { | |
List<Row> rows = new ArrayList<>(); | |
rows.add(new Row("East", "Boy", "Tee", 10, 12.00)); | |
rows.add(new Row("East", "Boy", "Golf", 15, 20.00)); | |
rows.add(new Row("East", "Girl", "Tee", 8, 14.00)); | |
rows.add(new Row("East", "Girl", "Golf", 20, 24.00)); | |
rows.add(new Row("West", "Boy", "Tee", 5, 12.00)); | |
rows.add(new Row("West", "Boy", "Golf", 12, 20.00)); | |
rows.add(new Row("West", "Girl", "Tee", 15, 14.00)); | |
rows.add(new Row("West", "Girl", "Golf", 10, 24.00)); | |
// groupBy region and gender, summing total sales | |
Map<String, Map<String, Double>> pivot = rows.stream().collect(groupingBy(r -> r.region, | |
groupingReduce(r -> r.gender, r -> r.price * r.units, Double::sum))); | |
System.out.println(pivot); | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Nice! What jdk8 build is that? I tried 1.8.0-ea-b76 (latest from http://jdk8.java.net/download.html) on osx but am missing .stream()