com.linkedin.flashback.serializable.RecordedHttpMessage.java Source code

Introduction

Here is the source code for com.linkedin.flashback.serializable.RecordedHttpMessage.java

Source

/*
 * Copyright (c) LinkedIn Corporation. All rights reserved. Licensed under the BSD-2 Clause license.
 * See LICENSE in the project root for license information.
 */

package com.linkedin.flashback.serializable;

import com.google.common.base.Charsets;
import com.google.common.base.Strings;
import com.google.common.net.HttpHeaders;
import com.google.common.net.MediaType;
import java.io.IOException;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import org.apache.log4j.Logger;

/**
 * Abstract Recorded http message(Java bean) which contains properties that is used for both
 * http request and http response
 * @author shfeng
 * @author dvinegra
 */
public abstract class RecordedHttpMessage {
    private static final String DEFAULT_CHARSET = Charsets.UTF_8.toString();
    private static final Logger logger = Logger.getLogger("RecordedHttpMessage");
    private static final String DEFAULT_CONTENT_TYPE = MediaType.OCTET_STREAM.toString();

    private Map<String, String> _headers = Collections.EMPTY_MAP;
    private RecordedHttpBody _httpBody;

    public RecordedHttpMessage(Map<String, String> headers, RecordedHttpBody httpBody) {
        if (headers != null) {
            _headers = headers;
        }
        _httpBody = httpBody;

        // Update the Content-Length header if appropriate
        if (_headers.containsKey(HttpHeaders.CONTENT_LENGTH)) {
            try {
                int contentLength = _httpBody.getContent(getCharset()).length;
                _headers = new HashMap<>(_headers);
                _headers.put(HttpHeaders.CONTENT_LENGTH, Integer.toString(contentLength));
            } catch (IOException e) {
                logger.error("Caught exception " + e + " while updating Content-Length header");
            }
        }
    }

    public RecordedHttpBody getHttpBody() {
        return _httpBody;
    }

    public boolean hasHttpBody() {
        return _httpBody != null;
    }

    public Map<String, String> getHeaders() {
        return _headers;
    }

    public String getCharset() {
        String header = _headers.get(HttpHeaders.CONTENT_TYPE);
        if (Strings.isNullOrEmpty(header)) {
            return DEFAULT_CHARSET;
        } else {
            return MediaType.parse(header).charset().or(Charsets.UTF_8).toString();
        }
    }

    public String getContentType() {
        String header = _headers.get(HttpHeaders.CONTENT_TYPE);
        if (Strings.isNullOrEmpty(header)) {
            return DEFAULT_CONTENT_TYPE;
        } else {
            return MediaType.parse(header).withoutParameters().toString();
        }
    }
}