leafcutter.client.Serializer.java Source code

Java tutorial

Introduction

Here is the source code for leafcutter.client.Serializer.java

Source

package leafcutter.client;

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

import com.google.gson.*;
import com.google.gson.internal.Streams;
import com.google.gson.reflect.*;
import com.google.gson.stream.*;

import leafcutter.event.ActivityResult;
import leafcutter.event.DecisionResult;
import leafcutter.event.EndWorkflowDecision;
import leafcutter.event.Event;
import leafcutter.event.ScheduleActivityDecision;
import leafcutter.event.StartWorkflowDecision;
import leafcutter.task.ActivityTask;
import leafcutter.task.DecisionTask;
import leafcutter.task.Task;

public class Serializer {

    public static Gson getGSON() {
        RuntimeTypeAdapterFactory<Event> eventFactory = RuntimeTypeAdapterFactory.of(Event.class, "clazz")
                .registerSubtype(ActivityResult.class).registerSubtype(DecisionResult.class)
                .registerSubtype(ScheduleActivityDecision.class).registerSubtype(StartWorkflowDecision.class)
                .registerSubtype(EndWorkflowDecision.class);

        RuntimeTypeAdapterFactory<Task> taskFactory = RuntimeTypeAdapterFactory.of(Task.class, "clazz")
                .registerSubtype(ActivityTask.class).registerSubtype(DecisionTask.class);

        return new GsonBuilder().registerTypeAdapterFactory(eventFactory).registerTypeAdapterFactory(taskFactory)
                .create();

    }

}

/*
 * This is a blatant copy-paste of code from the Google GSON project. This class
 * is under their gson-extra module, and not yet released onto Maven Central as
 * a stand alone artifact or otherwise.
 * 
 * I've resorted to simply copying the latest version from here:
 * https://github.com/google/gson/blob/master/extras/src/main/java/com/google/
 * gson/typeadapters/RuntimeTypeAdapterFactory.java
 * 
 * License information:
 * 
 * Copyright (C) 2011 Google 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.
 */

final class RuntimeTypeAdapterFactory<T> implements TypeAdapterFactory {
    private final Class<?> baseType;
    private final String typeFieldName;
    private final Map<String, Class<?>> labelToSubtype = new LinkedHashMap<String, Class<?>>();
    private final Map<Class<?>, String> subtypeToLabel = new LinkedHashMap<Class<?>, String>();

    private RuntimeTypeAdapterFactory(Class<?> baseType, String typeFieldName) {
        if (typeFieldName == null || baseType == null) {
            throw new NullPointerException();
        }
        this.baseType = baseType;
        this.typeFieldName = typeFieldName;
    }

    /**
     * Creates a new runtime type adapter using for {@code baseType} using
     * {@code
     * typeFieldName} as the type field name. Type field names are case
     * sensitive.
     */
    public static <T> RuntimeTypeAdapterFactory<T> of(Class<T> baseType, String typeFieldName) {
        return new RuntimeTypeAdapterFactory<T>(baseType, typeFieldName);
    }

    /**
     * Creates a new runtime type adapter for {@code baseType} using
     * {@code "type"} as the type field name.
     */
    public static <T> RuntimeTypeAdapterFactory<T> of(Class<T> baseType) {
        return new RuntimeTypeAdapterFactory<T>(baseType, "type");
    }

    /**
     * Registers {@code type} identified by {@code label}. Labels are case
     * sensitive.
     *
     * @throws IllegalArgumentException
     *             if either {@code type} or {@code label} have already been
     *             registered on this type adapter.
     */
    public RuntimeTypeAdapterFactory<T> registerSubtype(Class<? extends T> type, String label) {
        if (type == null || label == null) {
            throw new NullPointerException();
        }
        if (subtypeToLabel.containsKey(type) || labelToSubtype.containsKey(label)) {
            throw new IllegalArgumentException("types and labels must be unique");
        }
        labelToSubtype.put(label, type);
        subtypeToLabel.put(type, label);
        return this;
    }

    /**
     * Registers {@code type} identified by its {@link Class#getSimpleName
     * simple name}. Labels are case sensitive.
     *
     * @throws IllegalArgumentException
     *             if either {@code type} or its simple name have already been
     *             registered on this type adapter.
     */
    public RuntimeTypeAdapterFactory<T> registerSubtype(Class<? extends T> type) {
        return registerSubtype(type, type.getSimpleName());
    }

    public <R> TypeAdapter<R> create(Gson gson, TypeToken<R> type) {
        if (type.getRawType() != baseType) {
            return null;
        }

        final Map<String, TypeAdapter<?>> labelToDelegate = new LinkedHashMap<String, TypeAdapter<?>>();
        final Map<Class<?>, TypeAdapter<?>> subtypeToDelegate = new LinkedHashMap<Class<?>, TypeAdapter<?>>();
        for (Map.Entry<String, Class<?>> entry : labelToSubtype.entrySet()) {
            TypeAdapter<?> delegate = gson.getDelegateAdapter(this, TypeToken.get(entry.getValue()));
            labelToDelegate.put(entry.getKey(), delegate);
            subtypeToDelegate.put(entry.getValue(), delegate);
        }

        return new TypeAdapter<R>() {
            @Override
            public R read(JsonReader in) throws IOException {
                JsonElement jsonElement = Streams.parse(in);
                JsonElement labelJsonElement = jsonElement.getAsJsonObject().remove(typeFieldName);
                if (labelJsonElement == null) {
                    throw new JsonParseException("cannot deserialize " + baseType
                            + " because it does not define a field named " + typeFieldName);
                }
                String label = labelJsonElement.getAsString();
                @SuppressWarnings("unchecked") // registration requires that
                // subtype extends T
                TypeAdapter<R> delegate = (TypeAdapter<R>) labelToDelegate.get(label);
                if (delegate == null) {
                    throw new JsonParseException("cannot deserialize " + baseType + " subtype named " + label
                            + "; did you forget to register a subtype?");
                }
                return delegate.fromJsonTree(jsonElement);
            }

            @Override
            public void write(JsonWriter out, R value) throws IOException {
                Class<?> srcType = value.getClass();
                String label = subtypeToLabel.get(srcType);
                @SuppressWarnings("unchecked") // registration requires that
                // subtype extends T
                TypeAdapter<R> delegate = (TypeAdapter<R>) subtypeToDelegate.get(srcType);
                if (delegate == null) {
                    throw new JsonParseException(
                            "cannot serialize " + srcType.getName() + "; did you forget to register a subtype?");
                }
                JsonObject jsonObject = delegate.toJsonTree(value).getAsJsonObject();
                if (jsonObject.has(typeFieldName)) {
                    throw new JsonParseException("cannot serialize " + srcType.getName()
                            + " because it already defines a field named " + typeFieldName);
                }
                JsonObject clone = new JsonObject();
                clone.add(typeFieldName, new JsonPrimitive(label));
                for (Map.Entry<String, JsonElement> e : jsonObject.entrySet()) {
                    clone.add(e.getKey(), e.getValue());
                }
                Streams.write(clone, out);
            }
        }.nullSafe();
    }
}