问题描述
使代码更具动态性的主要问题是,您事先不知道要分组多少个元素。在这种情况下,最好对List
所有元素进行分组。之所以可行,是因为如果两个列表的所有元素均相同且顺序相同,则它们相等。
在这种情况下,我们将按照由每种数据类型和模块组成的列表进行分组,而不是按类型和模块进行分组。
private static Map<List<String>, List<MyClass>> groupListBy(List<MyClass> data, String[] groupByFieldNames) {
final MethodHandles.Lookup lookup = MethodHandles.lookup();
List<MethodHandle> handles =
Arrays.stream(groupByFieldNames)
.map(field -> {
try {
return lookup.findGetter(MyClass.class, field, String.class);
} catch (Exception e) {
throw new RuntimeException(e);
}
}).collect(toList());
return data.stream().collect(groupingBy(
d -> handles.stream()
.map(handle -> {
try {
return (String) handle.invokeExact(d);
} catch (Throwable e) {
throw new RuntimeException(e);
}
}).collect(toList())
));
}
代码的第一部分将字段名称数组转换为List
的MethodHandle
。对于每个字段,将MethodHandle
为该字段检索a
:这是通过从中获取查询MethodHandles.lookup()
并使用来查找给定字段名称的句柄来完成的findGetter
:
产生方法句柄,以提供对非静态字段的读取访问权限。
其余代码创建分类器,以将其分组。在数据实例上调用所有句柄以返回String
值列表。这Stream
被收集到List
作为分类器。
样例代码:
public static void main(String[] args) {
List<MyClass> data = new ArrayList<>();
data.add(new MyClass("1", "A", "B"));
data.add(new MyClass("2", "A", "B"));
data.add(new MyClass("3", "A", "C"));
data.add(new MyClass("4", "B", "A"));
System.out.println(groupListBy(data, new String[] { "type", "module" }));
}
输出:
{[B, A]=[4], [A, B]=[1, 2], [A, C]=[3]}
MyClass.toString()
重写when 返回title
唯一的。
解决方法
我正在尝试使用Java 8 Collection-Stream按多个属性对对象列表进行分组。
效果很好:
public class MyClass
{
public String title;
public String type;
public String module;
public MyClass(String title,String type,String module)
{
this.type = type;
this.title = title;
this.module= module;
}
}
List<MyClass> data = new ArrayList();
data.add(new MyClass("1","A","B"));
data.add(new MyClass("2","B"));
data.add(new MyClass("3","C"));
data.add(new MyClass("4","B","A"));
Object result = data.stream().collect(Collectors.groupingBy((MyClass m)
-> m.type,Collectors.groupingBy((MyClass m) -> m.module)));
但我想使其更具动态性。我只想指定一个应该用于GroupBy的字符串数组(或列表)。
就像是:
Object groupListBy(List data,String[] groupByFieldNames)
{
//magic code
}
我想打电话给:
groupListBy(data,new String[]{"type","module"});
像我的示例一样,如何使groupBy-Method更动态?