com.yahoo.parsec.web.JsonFormatter Maven / Gradle / Ivy

Go to download
// Copyright 2016 Yahoo Inc.
// Licensed under the terms of the Apache license. Please see LICENSE.md file distributed with this work for terms.

package com.yahoo.parsec.web;

import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.node.ObjectNode;

import java.io.IOException;
import java.util.Collection;
import java.util.Collections;
import java.util.Date;
import java.util.Enumeration;
import java.util.Map;
import java.util.stream.Collectors;

public class JsonFormatter implements ServletRequestResponseFomatter {

    private static final ObjectMapper _OBJECT_MAPPER = new ObjectMapper()
            .setSerializationInclusion(JsonInclude.Include.NON_NULL);

    @Override
    public String format(ParsecServletRequestWrapper req, ParsecServletResponseWrapper resp,
                         Map additionalArgs) {
        try {
            ObjectNode root = _OBJECT_MAPPER.createObjectNode();
            root.put("time", (System.currentTimeMillis() / 1000L));
            ObjectNode reqNode = root.with("request");
            fillRequestNode(req, reqNode);
            ObjectNode respNode = root.with("response");
            fillResponseNode(resp, respNode);
            return _OBJECT_MAPPER.writeValueAsString(root);
        } catch(IOException e) {
            throw new RuntimeException(e);
        }
    }

    private void fillRequestNode(ParsecServletRequestWrapper req, ObjectNode reqNode) throws IOException {
        reqNode.put("method", req.getMethod());
        StringBuilder uri = new StringBuilder(req.getRequestURI());
        if (req.getQueryString() != null) {
            uri.append("?").append(req.getQueryString());
        }
        reqNode.put("uri", uri.toString());
        reqNode.set("headers", getReqHeadersNode(req));
        reqNode.put("payload", req.getContent());
    }

    private void fillResponseNode(ParsecServletResponseWrapper resp, ObjectNode respNode) throws IOException {
        respNode.put("status", resp.getStatus());
        respNode.set("headers", getRespHeadersNode(resp));
        respNode.put("payload", resp.getContent());
    }

    ObjectNode getReqHeadersNode(ParsecServletRequestWrapper request) {
        ObjectNode headersNode = _OBJECT_MAPPER.createObjectNode();
        Enumeration headerNames = request.getHeaderNames();
        while(headerNames.hasMoreElements()) {
            String key = headerNames.nextElement();
            Enumeration headerValues = request.getHeaders(key);
            String val = Collections.list(headerValues).stream().collect(Collectors.joining(","));
            headersNode.put(key, val);
        }
        return headersNode;
    }

    ObjectNode getRespHeadersNode(ParsecServletResponseWrapper response) {
        ObjectNode headersNode = _OBJECT_MAPPER.createObjectNode();
        Collection headerNames = response.getHeaderNames();
        for (String header: headerNames) {
            String headerValue = response.getHeaders(header).stream().collect(Collectors.joining(","));
            headersNode.put(header, headerValue);
        }
        return headersNode;
    }
}