com.github.simonpercic.oklog.LogDataInterceptor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of oklog-java Show documentation
Response logging interceptor for OkHttp. Logs a url link with url-encoded response for every OkHttp call.
The newest version!
package com.github.simonpercic.oklog;

import com.github.simonpercic.oklog.core.BaseLogDataInterceptor;
import com.squareup.okhttp.Connection;
import com.squareup.okhttp.Headers;
import com.squareup.okhttp.Interceptor.Chain;
import com.squareup.okhttp.MediaType;
import com.squareup.okhttp.Protocol;
import com.squareup.okhttp.Request;
import com.squareup.okhttp.Response;
import com.squareup.okhttp.internal.http.HttpEngine;

import org.jetbrains.annotations.Nullable;

import java.io.IOException;
import java.nio.charset.Charset;
import java.nio.charset.UnsupportedCharsetException;

import okio.Buffer;
import okio.BufferedSource;

/**
 * Log data interceptor util. Inspired by: https://github.com/square/okhttp/tree/master/okhttp-logging-interceptor.
 *
 * @author Simon Percic https://github.com/simonpercic
 */
class LogDataInterceptor extends BaseLogDataInterceptor {

    private static final String OKHTTP_SENT_MILLIS = "OkHttp-Sent-Millis";
    private static final String OKHTTP_RECEIVED_MILLIS = "OkHttp-Received-Millis";

    @Override protected Request request(Chain chain) {
        return chain.request();
    }

    @Override protected String protocol(Chain chain) {
        Connection connection = chain.connection();
        Protocol protocol = connection != null ? connection.getProtocol() : Protocol.HTTP_1_1;
        return protocol.toString();
    }

    @Override protected String requestMethod(Request request) {
        return request.method();
    }

    @Override protected String requestUrl(Request request) {
        return request.httpUrl().toString();
    }

    @Override protected String requestUrlPath(Request request) {
        return request.httpUrl().encodedPath();
    }

    @Override protected String responseUrl(Response response) {
        return response.request().httpUrl().toString();
    }

    @Override protected Headers requestHeaders(Request request) {
        return request.headers();
    }

    @Override protected Headers responseHeaders(Response response) {
        return response.headers();
    }

    @Override protected int headersCount(Headers headers) {
        return headers.size();
    }

    @Override protected String headerName(Headers headers, int index) {
        return headers.name(index);
    }

    @Override protected String headerValue(Headers headers, int index) {
        return headers.value(index);
    }

    @Override protected String headerValue(Headers headers, String name) {
        return headers.get(name);
    }

    @Override protected boolean hasRequestBody(Request request) {
        return request.body() != null;
    }

    @Override protected boolean hasResponseBody(Response response) {
        return HttpEngine.hasBody(response);
    }

    @Override protected int responseCode(Response response) {
        return response.code();
    }

    @Override protected String responseMessage(Response response) {
        return response.message();
    }

    @Override protected long requestContentLength(Request request) throws IOException {
        return request.body().contentLength();
    }

    @Override protected long responseContentLength(Response response) throws IOException {
        return response.body().contentLength();
    }

    @Override protected MediaType requestContentType(Request request) {
        return request.body().contentType();
    }

    @Override protected MediaType responseContentType(Response response) {
        return response.body().contentType();
    }

    @Override protected String contentTypeString(MediaType mediaType) {
        return mediaType.toString();
    }

    @Override protected Charset contentTypeCharset(MediaType mediaType, Charset charset) {
        return mediaType.charset(charset);
    }

    @Nullable @Override protected Charset responseContentTypeCharset(MediaType contentType, Charset charset) {
        try {
            return contentTypeCharset(contentType, charset);
        } catch (UnsupportedCharsetException e) {
            return null;
        }
    }

    @Override protected void writeRequestBody(Request request, Buffer buffer) throws IOException {
        request.body().writeTo(buffer);
    }

    @Override protected BufferedSource responseBodySource(Response response) throws IOException {
        return response.body().source();
    }

    @Override protected boolean skipResponseHeader(String headerName) {
        // skip OkHttp response header timestamps
        return OKHTTP_SENT_MILLIS.equalsIgnoreCase(headerName)
                || OKHTTP_RECEIVED_MILLIS.equalsIgnoreCase(headerName);
    }
}