流API--流的收集

前面的一系列博客中,我们都是从一个集合中拿到一个流,但是有时候需要执行反操作,就是从流中获得集合。实际编码中,当我们处理完流后,我们通常想查看下结果,而不是将他们聚合成一个值。我们可以调用iterator方法来生成一个传统风格的迭代器,用于访问元素。
  • 1,将流中的元素收集到一个数组中,调用toArray()方法即可
由于无法在运行时创建一个泛型数组,所以调用toArray()方法将返回一个object[]数组。如果我们希望得到一个正确的数组,可以将类型传递给数组的构造函数toArray(String[]::new)。
public static void main(String[] args)
	{
		Stream<String> stream = Stream.of("林肯", "冰儿");
		String[] array = stream.toArray(String[]::new);
		for (String str : array)
		{
			System.out.println(str);
		}
	}

  • 2,将流中的元素收集到一个集合中。
流API提供了一个collect()方法,我们可以对比stream()方法,来很简单的实现集合和流的互相转换。
翻一下API说明文档,上面是这么定义这2个方法的:
<R,A> R 	collect(Collector<? super T,A,R> collector):Performs a mutable reduction operation on the elements of this stream using a Collector. 
<R> R 		collect(Supplier<R> supplier, BiConsumer<R,? super T> accumulator, BiConsumer<R,R> combiner):Performs a mutable reduction operation on the elements of this stream. 

关于Collector接口我们了解下好了,这个接口的申明如下:
interface Collector<T,A,R>,R指定结果的类型,T指定调用流的元素类型,内部累积类型由A指定。
Collectors类为了我们定义了一些可以直接使用的静态收集器方法,这是一个不可变类,我们一般使用toList()和toSet()方法,如下所示:
public static <T> Collector<T, ?, List<T>> toList()
	{
		return new CollectorImpl<>((Supplier<List<T>>) ArrayList::new, List::add, (left, right) ->
		{
			left.addAll(right);
			return left;
		} , CH_ID);
	}


	public static <T> Collector<T, ?, Set<T>> toSet()
	{
		return new CollectorImpl<>((Supplier<Set<T>>) HashSet::new, Set::add, (left, right) ->
		{
			left.addAll(right);
			return left;
		} , CH_UNORDERED_ID);
	}

toList()方法返回的收集器可用于将元素收集到一个list中,toSet()方法返回的收集器可用于将元素收集到一个Set中。例如,我们现在想要把元素收集到一个list中,可以向下面这样子调用:
public static void main(String[] args) throws Exception
	{
		List<Double> list = new ArrayList<>(4);
		list.add(1.1);
		list.add(2.5);
		list.add(3.0);
		list.add(4.8);
		Stream<Double> parallelStream = list.parallelStream();
		List<Double> collect = parallelStream.collect(Collectors.toList());
		collect.forEach(System.out::println);
	}

上面的整理我们使用Collectors的静态方法比较方便,当然我们也看到了上面的collect方法还有另外一个重载,我们使用这个版本可以对收集过程做更多的控制。
collect(Supplier<R> supplier, BiConsumer<R,? super T> accumulator, BiConsumer<R,R> combiner)
关于这个方法签名多说几句,上面方法中的3个参数:
supplier:表示一个指向可变存储对象的引用;后面2个参数都是BiConsumer<R,R> 类型的,这个函数式接口表示对2个对象进行处理,它里面的抽象方法签名如下:void accept(T obj1,U obj2),对于上面第2个参数,obj1指定目标集合,obj2指定要添加到该集合中的元素,对于上面的第3个参数,obj1和obj2指定两个将要被合并的集合。
具体的使用看下面代码:
public static void main(String[] args) throws Exception
	{
		List<Double> list = new ArrayList<>(4);
		list.add(1.1);
		list.add(2.5);
		list.add(3.0);
		list.add(4.8);
		Stream<Double> parallelStream = list.parallelStream();
		//使用Lambda表达式
		parallelStream.collect(() -> new LinkedList<>(), (MyList, element) -> MyList.add(element), (listA, listB) -> listA.addAll(listB))
		.forEach(System.out::println);
		//使用方法引用,直接使用被消费过的流报错stream has already been operated upon or closed,所以要重新获得流
		List<Double> list1 = list.parallelStream().collect(LinkedList::new,LinkedList::add,LinkedList::addAll);
		list1.forEach(System.out::println);
	}

  • 3,将流中的元素收集到一个map中
和上面的处理list一样,使用Collectors的toMap()方法就可以将一个流中的元素收集到一个map中。该方法有2个参数,分别用来生成map的键和值。
public class Test
{


	public static void main(String[] args)
	{
		Stream<Person> stream = Stream.of(new Person("1", "林肯"), new Person("2", "冰儿"));
		Map<String, String> collect = stream.collect(Collectors.toMap(Person::getId, Person::getName));
		for (String string : collect.keySet())
		{
			System.out.println(string + "-->" + collect.get(string));
		}


		for (Map.Entry<String, String> entry : collect.entrySet())
		{
			System.out.println(entry.getKey() + "-->" + entry.getValue());
		}
		System.out.println("==========华丽丽的分割线============");


		Stream<Person> stream1 = Stream.of(new Person("1", "林肯"), new Person("2", "冰儿"));
		//如果要获得原来的元素,使用Function.identity()就OK啦
		Map<String, Person> map = stream1.collect(Collectors.toMap(Person::getId, Function.identity()));
		for (String string : map.keySet())
		{
			System.out.println(string + "-->" + map.get(string).toString());
		}
	}


}


class Person
{
	private String id;
	private String name;


	public Person(String id, String name)
	{
		this.id = id;
		this.name = name;
	}


	public String getId()
	{
		return id;
	}


	public void setId(String id)
	{
		this.id = id;
	}


	public String getName()
	{
		return name;
	}


	public void setName(String name)
	{
		this.name = name;
	}


	@Override
	public String toString()
	{
		return "Person [id=" + id + ", name=" + name + "]";
	}


}

关于这点要注意的是,如果有相同的元素拥有相同的键,那么收集方法就会报错:java.lang.IllegalStateException
public static void main(String[] args)
	{
		Stream<Person> stream = Stream.of(new Person("1", "林肯"), new Person("1", "冰儿"));
		//下面代码报错,java.lang.IllegalStateException:
		Map<String, String> collect = stream.collect(Collectors.toMap(Person::getId, Person::getName));
		for (String string : collect.keySet())
		{
			System.out.println(string + "-->" + collect.get(string));
		}


	}
那么怎么办呢?
我们可以提供第3个函数参数,根据已有的值和新值来决定键的值,从而重新该行为。
public static void main(String[] args)
	{
		Stream<Person> stream = Stream.of(new Person("1", "林肯"), new Person("1", "冰儿"));
		//下面代码输出:1-->林肯冰儿
		Map<String, String> collect = stream.collect(Collectors.toMap(Person::getId, Person::getName,(value1,value2)->value1+value2));
		for (String string : collect.keySet())
		{
			System.out.println(string + "-->" + collect.get(string));
		}


	}

原文地址:https://www.cnblogs.com/LinkinPark/p/5232962.html