org.talend.components.webtest.JsonIo2HttpMessageConverter.java Source code

Introduction

Here is the source code for org.talend.components.webtest.JsonIo2HttpMessageConverter.java

Source

// ============================================================================
//
// Copyright (C) 2006-2016 Talend Inc. - www.talend.com
//
// This source code is available under agreement available at
// %InstallDIR%\features\org.talend.rcp.branding.%PRODUCTNAME%\%PRODUCTNAME%license.txt
//
// You should have received a copy of the agreement
// along with this program; if not, write to Talend SA
// 9 rue Pages 92150 Suresnes, France
//
// ============================================================================
package org.talend.components.webtest;

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

import org.springframework.http.HttpInputMessage;
import org.springframework.http.HttpOutputMessage;
import org.springframework.http.MediaType;
import org.springframework.http.converter.AbstractHttpMessageConverter;
import org.springframework.http.converter.HttpMessageNotReadableException;
import org.springframework.http.converter.HttpMessageNotWritableException;

import com.cedarsoftware.util.io.JsonReader;
import com.cedarsoftware.util.io.JsonWriter;

/**
 * Spring HttpMessageConverter implementation to use json-io for converting Objects to json.
 * TODO add json-io configuration support.
 * TODO contribute back to json-io project
 * TODO provide a way to configure swagger exclusion instead of hard code see {@link #supports(Class)}
 */

public final class JsonIo2HttpMessageConverter extends AbstractHttpMessageConverter<Object> {

    public static final Charset DEFAULT_CHARSET = Charset.forName("UTF-8");

    /**
     * Sets following mime type headers:
     * application/json;charset=UTF-8
     * application/*+json;charset=UTF-8
     */
    public JsonIo2HttpMessageConverter() {
        super(new MediaType("application", "json", DEFAULT_CHARSET),
                new MediaType("application", "*+json", DEFAULT_CHARSET));
    }

    /**
     * Serializes specified Object and writes JSON to HTTP message body 
     * 
     * @param t Object to serialize
     * @param outputMessage the HTTP output message to write to
     */
    @Override
    protected void writeInternal(Object t, HttpOutputMessage outputMessage)
            throws IOException, HttpMessageNotWritableException {
        OutputStreamWriter outputStreamWriter = new OutputStreamWriter(outputMessage.getBody());
        String objectToJson = JsonWriter.objectToJson(t);
        outputStreamWriter.write(objectToJson);
        outputStreamWriter.flush();
        outputStreamWriter.close();
    }

    /**
     * {@inheritDoc}
     * 
     * This implementation doesn't and shouldn't support swagger object.
     */
    @Override
    protected boolean supports(Class<?> clazz) {
        // the following exception is used to exclude this json message converter to be used for swagger.
        // otherwise the generated json is no more compatibile with swagger standard.
        return !clazz.getName().startsWith("com.mangofactory");
    }

    /**
     * Deserializes JSON to object
     * 
     * @param clazz type of object to return. Not used
     * @param inputMessage the HTTP input message to read from
     * @return deserialized object
     */
    @Override
    protected Object readInternal(Class<? extends Object> clazz, HttpInputMessage inputMessage)
            throws IOException, HttpMessageNotReadableException {
        return JsonReader.jsonToJava(inputMessage.getBody(), null);
    }
}