com.github.yongchristophertang.engine.web.response.PrintResultHandler Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2014-2015 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.github.yongchristophertang.engine.web.response;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.github.yongchristophertang.engine.web.HttpResult;
import com.github.yongchristophertang.engine.web.ResultHandler;
import com.google.common.collect.Lists;
import org.apache.http.HttpEntityEnclosingRequest;
import org.apache.http.RequestLine;
import org.apache.http.util.EntityUtils;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

import java.net.URLDecoder;

/**
 * An implementation of {@link ResultHandler} for performing print task
 *
 * @author Yong Tang
 * @since 0.4
 */
public class PrintResultHandler implements ResultHandler {
    private static final Logger logger = LogManager.getLogger();

    /**
     * Accessed via {@link HttpResultHandlers#print}
     */
    protected PrintResultHandler() {
    }

    /**
     * Print the given result to log system.
     *
     * @param result the result of the executed request
     * @throws Exception if a failure occurs
     */
    @Override
    public void handle(HttpResult result) throws Exception {
        RequestLine rl = result.getHttpRequest().getRequestLine();
        String body = null;
        String formatter = "HTTP Request&Response Log: \n\n \t API: {} \n\n\t Request URL: {} \n\n \t ";
        if (rl.getMethod().equals("POST") || rl.getMethod().equals("PUT") || rl.getMethod().equals("PATCH")) {
            try {
                body = URLDecoder
                    .decode(EntityUtils.toString(((HttpEntityEnclosingRequest) result.getHttpRequest()).getEntity()), "UTF-8");
                formatter += "Request Body (URL Decoded): {} \n\n \t ";
            } catch (UnsupportedOperationException e) {
                formatter += "Multipart Body Cannot Be Displayed which is {}. \n\n \t ";
            } catch (IllegalArgumentException e) {
                formatter += "Request Body: {} \n\n \t ";
            }
        } else {
            formatter += "Request Body Not Applicable: {}\n\n \t ";
        }
        formatter +=
            "Request Headers: {} \n\n \t Cost Time(ms): {} \n\n \t Response Status: {} \n\n \t Response Headers: {} " +
                "\n\n \t " +
                "Response Content: \n {} \n=======================================================================\n";

        logger.info(formatter, result.getRequestDescritpion(), rl, body, Lists.newArrayList(result.getHttpRequest().getAllHeaders()),
            result.getCostTime(), result.getHttpResponse().getStatusLine(),
            Lists.newArrayList(result.getHttpResponse().getAllHeaders()),
            getPrettyJsonPrint(result.getResponseStringContent()));
    }

    /**
     * Reformat Json string for better look in printing
     *
     * @param rawJson raw json string
     * @return formatted json string
     */
    private String getPrettyJsonPrint(String rawJson) {
        ObjectMapper mapper = new ObjectMapper();
        try {
            Object json = mapper.readValue(rawJson,
                Object.class);
            return mapper.writerWithDefaultPrettyPrinter()
                .writeValueAsString(json);
        } catch (Exception e) {
            return rawJson;
        }
    }
}