org.springframework.social.partnercenter.http.logging.Slf4jHttpRequestResponseLogger Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-social-microsoft-partnercenter Show documentation

There is a newer version: 10.11.0

Show newest version

package org.springframework.social.partnercenter.http.logging;

import static java.util.Arrays.asList;

import java.io.IOException;
import java.time.Instant;
import java.util.List;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.http.HttpRequest;
import org.springframework.http.HttpStatus;
import org.springframework.http.client.ClientHttpResponse;

public class Slf4jHttpRequestResponseLogger implements HttpRequestResponseLogger {
	private final Logger logger;
	private final LogLevel logLevel;
	private final HttpRequestLogger requestLogger;
	private final ClientHttpResponseLogger responseLogger;
	private final List expectedStatus;

	public Slf4jHttpRequestResponseLogger(Class logSource, LogLevel logLevel, HttpStatus... expectedStatus) {
		this(logSource, logLevel, asList(expectedStatus));
	}

	public Slf4jHttpRequestResponseLogger(Class logSource, LogLevel logLevel, List expectedStatus) {
		this.logger = LoggerFactory.getLogger(logSource);
		this.logLevel  = logLevel;
		this.requestLogger = new Slf4jHttpRequestLogger(logger);
		this.responseLogger = new Slf4jClientHttpResponseLogger(logger);
		this.expectedStatus = expectedStatus;
	}

	@Override
	public void log(Instant startTime, Instant endTime, HttpRequest request, byte[] body, ClientHttpResponse response) throws IOException {
		if (expectedStatus.contains(response.getStatusCode())){
			if(logger.isDebugEnabled() && logLevel.equals(LogLevel.DEBUG)){
				requestLogger.logDedug(startTime, request, body);
				responseLogger.logDebug(endTime, request, response);
			} else if (logger.isInfoEnabled() && logLevel.equals(LogLevel.INFO)){
				requestLogger.logInfo(startTime, request, body);
				responseLogger.logInfo(endTime, request, response);
			}
		} else {
			requestLogger.logWarning(startTime, request, body);
			responseLogger.logWarning(endTime, request, response);
		}
	}
}