com.basistech.rosette.dm.jackson.ListAttributeSerializer.java Source code

Introduction

Here is the source code for com.basistech.rosette.dm.jackson.ListAttributeSerializer.java

Source

/*
* Copyright 2014 Basis Technology Corp.
*
* 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.basistech.rosette.dm.jackson;

import com.basistech.rosette.dm.ListAttribute;
import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.SerializerProvider;
import com.fasterxml.jackson.databind.jsontype.TypeSerializer;

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

/**
 * Jackson serializer that that handles polymorphism in lists of homogeneous type without
 * writing out the type every time.
 */
public class ListAttributeSerializer extends JsonSerializer<ListAttribute> {
    @Override
    public void serialize(ListAttribute value, JsonGenerator jgen, SerializerProvider provider) throws IOException {
        jgen.writeStartObject();

        jgen.writeStringField("itemType", KnownAttribute.getAttributeForClass(value.getItemClass()).key());
        jgen.writeObjectField("items", value.getItems());
        writeExtendedProperties(value, jgen);
        jgen.writeEndObject();
    }

    private void writeExtendedProperties(ListAttribute value, JsonGenerator jgen) throws IOException {
        Map<String, Object> extendedProperties = value.getExtendedProperties();
        if (extendedProperties != null && !extendedProperties.isEmpty()) {
            for (Map.Entry<String, Object> entry : extendedProperties.entrySet()) {
                jgen.writeObjectField(entry.getKey(), entry.getValue());
            }
        }
    }

    @Override
    public void serializeWithType(ListAttribute value, JsonGenerator jgen, SerializerProvider provider,
            TypeSerializer typeSer) throws IOException {
        typeSer.writeTypePrefixForObject(value, jgen);

        KnownAttribute attributeForClass = KnownAttribute.getAttributeForClass(value.getItemClass());
        assert attributeForClass != null;
        jgen.writeStringField("itemType", attributeForClass.key());
        jgen.writeObjectField("items", value.getItems());
        writeExtendedProperties(value, jgen);
        typeSer.writeTypeSuffixForObject(value, jgen);
    }
}