org.zalando.logbook.spring.webflux.DataBufferCopyUtils Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of logbook-spring-webflux Show documentation

The newest version!

package org.zalando.logbook.spring.webflux;

import lombok.experimental.UtilityClass;
import org.reactivestreams.Publisher;
import org.springframework.core.io.buffer.DataBuffer;
import org.springframework.core.io.buffer.DataBufferUtils;
import org.springframework.core.io.buffer.DefaultDataBuffer;
import org.springframework.core.io.buffer.DefaultDataBufferFactory;

import java.util.function.Consumer;

@UtilityClass
class DataBufferCopyUtils {

    Publisher wrapAndBuffer(Publisher body, Consumer copyConsumer) {
        return DataBufferUtils
                .join(body)
                .defaultIfEmpty(DefaultDataBufferFactory.sharedInstance.wrap(new byte[0]))
                .map(dataBuffer -> {
                    byte[] bytes = new byte[dataBuffer.readableByteCount()];
                    dataBuffer.read(bytes);
                    DataBufferUtils.release(dataBuffer);
                    DefaultDataBuffer wrappedDataBuffer = DefaultDataBufferFactory.sharedInstance.wrap(bytes);
                    copyConsumer.accept(bytes);
                    return wrappedDataBuffer;
                });
    }
}