com.aol.cyclops.javaslang.comprehenders.StreamComprehender Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cyclops-javaslang Show documentation

Converters and Comprehenders for Javaslang

There is a newer version: 8.4.2

package com.aol.cyclops.javaslang.comprehenders;

import java.util.Collection;
import java.util.function.Function;

import javaslang.collection.Stream;

import com.aol.cyclops.lambda.api.Comprehender;
import com.nurkiewicz.lazyseq.LazySeq;

public class StreamComprehender implements Comprehender {

	@Override
	public Object map(Stream t, Function fn) {
		return t.map(s -> fn.apply(s));
	}
	@Override
	public Object executeflatMap(Stream t, Function fn){
		return flatMap(t,input -> unwrapOtherMonadTypes(this,fn.apply(input)));
	}
	@Override
	public Object flatMap(Stream t, Function fn) {
		return t.flatMap(s->fn.apply(s));
	}

	@Override
	public Stream of(Object o) {
		return Stream.of(o);
	}

	@Override
	public Stream empty() {
		return Stream.of();
	}

	@Override
	public Class getTargetClass() {
		return Stream.class;
	}
	static Stream unwrapOtherMonadTypes(Comprehender comp,Object apply){
		if(apply instanceof java.util.stream.Stream)
			return Stream.ofAll( ((java.util.stream.Stream)apply).iterator());
		if(apply instanceof Iterable)
			return Stream.ofAll( ((Iterable)apply).iterator());
		if(apply instanceof LazySeq){
			apply = Stream.ofAll(((LazySeq)apply).iterator());
		}
		if(apply instanceof Collection){
			return Stream.ofAll((Collection)apply);
		}
		
		return Comprehender.unwrapOtherMonadTypes(comp,apply);
		
	}

}