org.jboss.resteasy.reactive.common.util.MultiCollectors Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of resteasy-reactive-common Show documentation

There is a newer version: 3.17.5

Show newest version

package org.jboss.resteasy.reactive.common.util;

import java.util.stream.Collectors;

import io.smallrye.mutiny.Multi;
import io.smallrye.mutiny.Uni;

/**
 * Set of convenient collectors for dealing with {@link Multi} return types.
 */
// I've avoided creating collectors for silly things like Byte and Character because
// they can't possibly be useful, and even the JAX-RS spec doesn't mandate we have writers
// for them
public class MultiCollectors {

    public static Uni concatenateStrings(Multi multi) {
        return multi.collect().with(Collectors.joining());
    }

    public static Uni concatenateByteArrays(Multi multi) {
        // we could avoid the list and grow an array as we collect, but I doubt that's more efficient TBH
        return multi.collect().asList()
                .map(list -> {
                    int size = 0;
                    for (byte[] array : list) {
                        size += array.length;
                    }
                    byte[] ret = new byte[size];
                    int i = 0;
                    for (byte[] array : list) {
                        System.arraycopy(array, 0, ret, i, array.length);
                        i += array.length;
                    }
                    return ret;
                });
    }

    public static Uni concatenateCharArrays(Multi multi) {
        // we could avoid the list and grow an array as we collect, but I doubt that's more efficient TBH
        return multi.collect().asList()
                .map(list -> {
                    int size = 0;
                    for (char[] array : list) {
                        size += array.length;
                    }
                    char[] ret = new char[size];
                    int i = 0;
                    for (char[] array : list) {
                        System.arraycopy(array, 0, ret, i, array.length);
                        i += array.length;
                    }
                    return ret;
                });
    }
}