工具方法
public static <T> Map<String,List<T>> groupByOneField(String keyName, List<T> source){
if (StringUtil.isEmpty(keyName)|| CollectionUtil.isEmptyArray(source)){
return null;
}
//key必须是source内对象中的一个属性
Class<?> sourceObj = source.get(0).getClass();
try {
sourceObj.getDeclaredField(keyName);
} catch (NoSuchFieldException e) {
throw new RuntimeException("不存在属性:" + keyName);
}
String getMethodName = "get" + keyName.substring(0, 1).toUpperCase() + keyName.substring(1);
Method getMethod;
try {
getMethod = sourceObj.getMethod(getMethodName);
} catch (SecurityException e) {
throw new RuntimeException("方法转化异常:" + getMethodName, e);
} catch (NoSuchMethodException e) {
throw new RuntimeException("没有找到此方法:" + getMethodName, e);
} catch (Exception e){
throw new RuntimeException("获取方法异常:" + getMethodName, e);
}
//根据key进行分组
Map<String,List<T>> map = new HashMap<>();
Set<String> keys = map.keySet();
for (T o : source) {
String key = "";
try {
key = getMethod.invoke(o).toString();
} catch (Exception e) {
throw new RuntimeException("获取属性值失败!");
}
if (keys.contains(key)){
List<T> values = map.get(key);
values.add(o);
} else{
List<T> values = new ArrayList();
values.add(o);
map.put(key,values);
}
}
return map;
}
测试
public static void main(String[] args) {
//下面是初始化的数据
List<Student> list = new ArrayList<Student>();
Student student1 = new Student("李四1", "女", "一班");
Student student2 = new Student("李四2", "女", "一班");
Student student3 = new Student("李四3", "女", "一班");
Student student4 = new Student("李四4", "男", "一班");
Student student5 = new Student("李四5", "男", "一班");
Student student6 = new Student("李四6", "男", "二班");
Student student7 = new Student("李四7", "男", "二班");
Student student8 = new Student("李四8", "男", "二班");
Student student9 = new Student("李四9", "男", "二班");
list.add(student1);
list.add(student2);
list.add(student3);
list.add(student4);
list.add(student5);
list.add(student6);
list.add(student7);
list.add(student8);
list.add(student9);
Map<String, List<Student>> results = MapUtil.groupByOneField("className", list);
for (String key : results.keySet()) {
System.out.println("====================key:" + key);
System.out.println(results.get(key).toString());
}
}