package com.zll.test.stream;
import java.util.*;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.function.Consumer;
import java.util.stream.Collectors;
import java.util.stream.Stream;
public class StreamDemo {
public static void main(String[] args) {
String[] names = {"陈昌杰", "王欣", "唐雪冰", "何泽江", "史美江", "杨苏川", "袁江", "邱杰", "张迪"};
List<String> collect = Stream.of(names).collect(Collectors.toList());
Map<String, User> userMap = listToMap(collect);
System.out.println("==============listToMapObject==============");
System.out.println(userMap);
System.out.println("\n");
Map<Integer, String> userM = listToMap2(collect);
System.out.println("==============listToMap==============");
System.out.println(userM);
System.out.println("\n");
List<User> users = mapToList(userM);
System.out.println("==============mapToList==============");
System.out.println(users);
System.out.println("\n");
List<User> filter = filter(users, "陈昌杰");
System.out.println("==============filter==============");
System.out.println(filter);
System.out.println("\n");
List<Group> groups = new ArrayList<>();
List<GroupByUser> pgs = new ArrayList<>();
groups.add(new Group(0, "his"));
groups.add(new Group(1, "wx"));
groups.add(new Group(2, "jk"));
Random rd = new Random();
for (int i = 0; i < 20; i++) {
GroupByUser pg = new GroupByUser(rd.nextInt(3), rd.nextInt(names.length));
pgs.add(pg);
}
Map<Integer, List<GroupByUser>> collect1 = pgs.stream().collect(Collectors.groupingBy(GroupByUser::getuId));
System.out.println("==============根据用户id分组==============");
System.out.println(collect1);
System.out.println("\n");
Map<Integer, Long> collect2 = pgs.stream().collect(Collectors.groupingBy(GroupByUser::getuId, Collectors.counting()));
System.out.println("==============根据用户id分组 统计个数==============");
System.out.println(collect2);
System.out.println("\n");
Map<Integer, Long> collect3 = pgs.stream().collect(Collectors.groupingBy(GroupByUser::getuId, Collectors.summingLong(GroupByUser::getgId)));
System.out.println("==============根据用户id分组 累加==============");
System.out.println(collect3);
System.out.println("\n");
List<User> users1 = users.stream().map(user -> {
List<GroupByUser> groupByUsers = collect1.get(user.id);
for (GroupByUser groupByUser : groupByUsers) {
for (Group group : groups) {
if (groupByUser.getgId() == group.id) {
user.groups.add(group);
}
}
}
return user;
}).collect(Collectors.toList());
System.out.println("==============获取user 的分组1==============");
System.out.println(users1);
System.out.println("\n");
}
private static Map<String, User> listToMap(List<String> collect) {
AtomicInteger i = new AtomicInteger();
//原理演示
Map<String, User> map = collect.stream().collect(Collectors.toMap(value -> {
//System.out.println(value); 这里的value其实就是 name
return value;
}, value -> {
//System.out.println(value); 这里的value其实就是 name
return new User(i.getAndIncrement(), value);
}));
return map;
}
private static Map<Integer, String> listToMap2(List<String> collect) {
AtomicInteger i = new AtomicInteger();
Map<Integer, String> map2 = collect.stream().collect(Collectors.toMap(value -> i.getAndIncrement(), value -> value));
return map2;
}
public static List<User> filter(List<User> users, String key) {
List<User> collect = users.stream().filter(s -> s.name.equals(key)).collect(Collectors.toList());
return collect;
}
public static List<User> mapToList(Map<Integer, String> map) {
List<User> collect = map.keySet().stream().map(key -> new User(key, map.get(key))).collect(Collectors.toList());
return collect;
}
private String createStream() {
List<String> list = new ArrayList<>();
Map<String, Object> map = new HashMap<>();
String[] arr = {};
Stream<String> stream = list.stream();
stream = map.keySet().stream();
Stream<Object> stream1 = map.values().stream();
Stream<Map.Entry<String, Object>> stream2 = map.entrySet().stream();
Stream<String> arr1 = Stream.of(arr);
Stream<String> arr2 = Stream.of("a", "b", "c", "d");
return "";
}
static class User {
int id;
String name;
List<Group> groups = new ArrayList<>();
public User(int id, String name) {
this.id = id;
this.name = name;
}
@Override
public String toString() {
return "User{" +
"id=" + id +
", name='" + name + '\'' +
", groups=" + groups +
'}';
}
}
static class Group {
int id;
String name;
public Group(int id, String name) {
this.id = id;
this.name = name;
}
@Override
public String toString() {
return "Group{" +
"id=" + id +
", name='" + name + '\'' +
'}';
}
}
static class GroupByUser {
int gId;
int uId;
public GroupByUser(int gId, int uId) {
this.gId = gId;
this.uId = uId;
}
@Override
public String toString() {
return "GroupByUser{" +
"gId=" + gId +
", uId=" + uId +
'}';
}
public int getgId() {
return gId;
}
public void setgId(int gId) {
this.gId = gId;
}
public int getuId() {
return uId;
}
public void setuId(int uId) {
this.uId = uId;
}
}
}
java Stream Demo
于 2020-09-26 16:06:49 首次发布