com.github.shyiko.jackson.module.advice.JsonAdviceModule Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2013 Stanley Shyiko
 *
 * 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.github.shyiko.jackson.module.advice;

import com.fasterxml.jackson.core.Version;
import com.fasterxml.jackson.databind.BeanDescription;
import com.fasterxml.jackson.databind.DeserializationConfig;
import com.fasterxml.jackson.databind.Module;
import com.fasterxml.jackson.databind.SerializationConfig;
import com.fasterxml.jackson.databind.deser.BeanDeserializerBuilder;
import com.fasterxml.jackson.databind.deser.BeanDeserializerModifier;
import com.fasterxml.jackson.databind.ser.BeanSerializerModifier;

/**
 * @author Stanley Shyiko
 */
public class JsonAdviceModule extends Module {

    private static final Version MODULE_VERSION = new Version(1, 1, 0, null,
            "com.github.shyiko", "jackson-module-advice");

    @Override
    public String getModuleName() {
        return this.getClass().getSimpleName();
    }

    @Override
    public Version version() {
        return MODULE_VERSION;
    }

    @Override
    public void setupModule(SetupContext context) {
        context.addBeanSerializerModifier(new BeanSerializerModifier() {

            @Override
            public com.fasterxml.jackson.databind.ser.BeanSerializerBuilder updateBuilder(SerializationConfig config,
                    BeanDescription beanDesc, com.fasterxml.jackson.databind.ser.BeanSerializerBuilder builder) {
                JsonSerializerAdvice advice = beanDesc.getClassAnnotations().get(JsonSerializerAdvice.class);
                return advice != null ? new AdvisedBeanSerializerBuilder(builder, advice.value()) : builder;
            }
        });
        context.addBeanDeserializerModifier(new BeanDeserializerModifier() {

            @Override
            public BeanDeserializerBuilder updateBuilder(DeserializationConfig config, BeanDescription beanDesc,
                    BeanDeserializerBuilder builder) {
                JsonDeserializerAdvice advice = beanDesc.getClassAnnotations().get(JsonDeserializerAdvice.class);
                return  advice != null ? new AdvisedBeanDeserializerBuilder(builder, advice.value()) : builder;
            }
        });
    }

}