pl.allegro.tech.hermes.common.http.ExtraRequestHeadersCollector Maven / Gradle / Ivy

Go to download

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

Fast and reliable message broker built on top of Kafka.

The newest version!

package pl.allegro.tech.hermes.common.http;

import java.util.Collections;
import java.util.Map;
import java.util.Set;
import java.util.function.BiConsumer;
import java.util.function.BinaryOperator;
import java.util.function.Function;
import java.util.function.Supplier;
import java.util.stream.Collector;

public class ExtraRequestHeadersCollector implements Collector, StringBuilder, String> {
    private ExtraRequestHeadersCollector() {
    }

    public static ExtraRequestHeadersCollector extraRequestHeadersCollector() {
        return new ExtraRequestHeadersCollector();
    }

    @Override
    public Supplier supplier() {
        return StringBuilder::new;
    }

    @Override
    public BiConsumer> accumulator() {
        return (StringBuilder accumulator, Map.Entry entry) -> {
            accumulator.append(entry.getKey());
            accumulator.append(": ");
            accumulator.append(entry.getValue());
            accumulator.append('\n');
        };
    }

    @Override
    public BinaryOperator combiner() {
        return StringBuilder::append;
    }

    @Override
    public Function finisher() {
        return (StringBuilder acc) -> {
            if (acc.length() > 0) {
                acc.setLength(acc.length() - 1);
            }
            return acc.toString();
        };
    }

    @Override
    public Set characteristics() {
        return Collections.emptySet();
    }
}