com.netflix.zeno.json.JsonSerializationFramework Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of netflix-zeno Show documentation
netflix-zeno
The newest version!
/*
 *
 *  Copyright 2013 Netflix, Inc.
 *
 *     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.netflix.zeno.json;

import java.io.IOException;
import java.io.StringWriter;
import java.util.Map;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.netflix.zeno.serializer.NFTypeSerializer;
import com.netflix.zeno.serializer.SerializationFramework;
import com.netflix.zeno.serializer.SerializerFactory;
import com.netflix.zeno.serializer.common.MapSerializer;

/**
 *
 * Serializes and deserializes JSON based on object instance contents.
 *
 * Usage is detailed in the documentation
 * on the page creating json data.
 *
 * Please see JSONSerializationExample in the source folder src/examples/java for example usage.
 *
 * @author tvaliulin
 *
 */
public class JsonSerializationFramework extends SerializationFramework {

    public JsonSerializationFramework(SerializerFactory factory) {
        super(factory);
        this.frameworkSerializer = new JsonFrameworkSerializer(this);
        this.frameworkDeserializer = new JsonFrameworkDeserializer(this);
    }

    public  String serializeAsJson(String type, T object) {
        return serializeAsJson(type, object, true);
    }

    public  String serializeAsJson(String type, T object, boolean pretty) {
        StringWriter writer = new StringWriter();

        JsonWriteGenericRecord record = new JsonWriteGenericRecord(writer, pretty);

        record.open();
        getSerializer(type).serialize(object, record);
        record.close();
        writer.flush();

        return writer.toString();
    }

    public  String serializeJsonMap(String keyType, String valueType, Map map, boolean pretty) {
        NFTypeSerializer keySerializer = getSerializer(keyType);
        NFTypeSerializer valueSerializer = getSerializer(valueType);

        MapSerializer mapSerializer = new MapSerializer(keySerializer, valueSerializer);
        mapSerializer.setSerializationFramework(this);

        StringWriter writer = new StringWriter();
        JsonWriteGenericRecord record = new JsonWriteGenericRecord(writer, pretty);

        record.open();
        mapSerializer.serialize(map, record);
        record.close();
        writer.flush();
        return writer.toString();
    }


    public  T deserializeJson(String type, String json) throws IOException {
        ObjectMapper mapper = new ObjectMapper();
        JsonNode node = mapper.readTree(json);
        NFTypeSerializer serializer = getSerializer(type);
        JsonReadGenericRecord readRecord = new JsonReadGenericRecord(serializer.getFastBlobSchema(), node);
        T object = serializer.deserialize(readRecord);

        return object;
    }

    public  Map deserializeJsonMap(String keyType, String valueType, String json) throws IOException {
        NFTypeSerializer keySerializer = getSerializer(keyType);
        NFTypeSerializer valueSerializer = getSerializer(valueType);

        MapSerializer mapSerializer = new MapSerializer(keySerializer, valueSerializer);
        mapSerializer.setSerializationFramework(this);

        JsonNode node = new ObjectMapper().readTree(json);
        JsonReadGenericRecord readRecord = new JsonReadGenericRecord(mapSerializer.getFastBlobSchema(), node);
        return mapSerializer.deserialize(readRecord);
    }

}