pl.piomin.logging.reactive.interceptor.RequestLoggingInterceptor Maven / Gradle / Ivy

Go to download
package pl.piomin.logging.reactive.interceptor;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.nio.channels.Channels;

import org.apache.commons.io.IOUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import reactor.core.publisher.Flux;

import org.springframework.core.io.buffer.DataBuffer;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpRequestDecorator;

import static net.logstash.logback.argument.StructuredArguments.value;

public class RequestLoggingInterceptor extends ServerHttpRequestDecorator {

	private static final Logger LOGGER = LoggerFactory.getLogger(RequestLoggingInterceptor.class);

	private boolean logHeaders;

	public RequestLoggingInterceptor(ServerHttpRequest delegate, boolean logHeaders) {
		super(delegate);
		this.logHeaders = logHeaders;
	}

	@Override
	public Flux getBody() {
		ByteArrayOutputStream baos = new ByteArrayOutputStream();
		return super.getBody().doOnNext(dataBuffer -> {
			try {
				Channels.newChannel(baos).write(dataBuffer.asByteBuffer().asReadOnlyBuffer());
				String body = IOUtils.toString(baos.toByteArray(), "UTF-8");
				if (logHeaders)
					LOGGER.info("Request: method={}, uri={}, headers={}, payload={}, audit={}", getDelegate().getMethod(),
							getDelegate().getPath(), getDelegate().getHeaders(), body, value("audit", true));
				else
					LOGGER.info("Request: method={}, uri={}, payload={}, audit={}", getDelegate().getMethod(),
							getDelegate().getPath(), body, value("audit", true));
			}
			catch (IOException e) {
				e.printStackTrace();
			}
			finally {
				try {
					baos.close();
				}
				catch (IOException e) {
					e.printStackTrace();
				}
			}
		});
	}

}