天天看点

泛型用法和stream中collector接口使用

使用场景:在从一个集合对象转另一个集合对象

使用了jdk1.8后的stream处理

public class PrimaryUtil {
    public static <R extends IPrimaryKey<T>, T> Collection<T> primary(Collection<R> primaryKeyObjs){
        return primaryKeyObjs.stream().collect(new Collector<R, Collection<T>, Collection<T>>() {
            @Override
            public Supplier<Collection<T>> supplier() {
                return ArrayList::new;
            }
            @Override
            public BiConsumer<Collection<T>, R> accumulator() {
                return (c, e) -> {
                    c.add(e.primaryKey());
                };
            }
            @Override
            public BinaryOperator<Collection<T>> combiner() {
                return (a, b) -> {
                    a.addAll(b);
                    return a;
                };
            }
            @Override
            public Function<Collection<T>, Collection<T>> finisher() {
                return (a) -> a;
            }
            @Override
            public Set<Characteristics> characteristics() {
                return Collections.unmodifiableSet(EnumSet.of(Collector.Characteristics.IDENTITY_FINISH));
            }
        });
    }


    public static <R extends IPrimaryKey<T>, T> Collection<T> primary1(Collection<R> primaryKeyObjs){
        return primaryKeyObjs.stream().collect(new AbstractCollector<R, Collection<T>>() {
            @Override
            public Supplier<Collection<T>> supplier() {
                return ArrayList::new;
            }
            @Override
            public BiConsumer<Collection<T>, R> accumulator() {
                return (c, e) -> {
                    c.add(e.primaryKey());
                };
            }
        });
    }
}
abstract class AbstractCollector<R , T extends Collection>
        implements Collector<R, T, T> {

    @Override
    public BinaryOperator<T> combiner() {
        return (a, b) -> {
            a.addAll(b);
            return a;
        };
    }
    @Override
    public Function<T, T> finisher() {
        return (a) ->  a;
    }

    @Override
    public Set<Characteristics> characteristics() {
        return Collections.unmodifiableSet(EnumSet.of(Collector.Characteristics.IDENTITY_FINISH));
    }
}
           
public interface IPrimaryKey<T>{

    T primaryKey();
}
           

使用demo

propertys.add(new BasicNameValuePair("user_ids", JsonUtil.serialize(PrimaryUtil.primary(getMembers()))));
           

这个里面是在httpClient中使用,这儿作用是构造post的提交参数。改参数填充的是所有成员的id属性构成的list序列化得到的字符串。