io.hawkcd.agent.utilities.deserializers.TaskDefinitionAdapter.java Source code

Introduction

Here is the source code for io.hawkcd.agent.utilities.deserializers.TaskDefinitionAdapter.java

Source

/*
 * Copyright (C) 2016 R&D Solutions Ltd.
 *
 * 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 io.hawkcd.agent.utilities.deserializers;

import com.google.gson.*;
import io.hawkcd.agent.enums.TaskType;

import java.lang.reflect.Type;
import java.util.HashMap;
import java.util.Map;

import io.hawkcd.agent.models.ExecTask;
import io.hawkcd.agent.models.FetchArtifactTask;
import io.hawkcd.agent.models.FetchMaterialTask;
import io.hawkcd.agent.models.TaskDefinition;
import io.hawkcd.agent.models.UploadArtifactTask;

public class TaskDefinitionAdapter implements JsonSerializer<TaskDefinition>, JsonDeserializer<TaskDefinition> {
    private Map<String, Type> taskTypeMap;
    private Gson jsonConverter;

    public TaskDefinitionAdapter() {
        this.taskTypeMap = new HashMap() {
            {
                this.put(TaskType.EXEC.toString(), ExecTask.class);
                this.put(TaskType.FETCH_ARTIFACT.toString(), FetchArtifactTask.class);
                this.put(TaskType.FETCH_MATERIAL.toString(), FetchMaterialTask.class);
                this.put(TaskType.UPLOAD_ARTIFACT.toString(), UploadArtifactTask.class);
            }
        };
        this.jsonConverter = new Gson();
    }

    @Override
    public JsonElement serialize(TaskDefinition src, Type typeOfSrc, JsonSerializationContext context) {
        Type taskType = this.taskTypeMap.get(src.getType().toString());
        JsonObject element = (JsonObject) this.jsonConverter.toJsonTree(src, taskType);
        return element;
    }

    @Override
    public TaskDefinition deserialize(JsonElement json, Type typeOfT, JsonDeserializationContext context)
            throws JsonParseException {
        JsonObject jsonObject = json.getAsJsonObject();
        String type = jsonObject.get("type").getAsString();
        try {
            if (type.equals("EXEC")) {
                return context.deserialize(jsonObject, ExecTask.class);
            } else if (type.equals("UPLOAD_ARTIFACT")) {
                return context.deserialize(jsonObject, UploadArtifactTask.class);
            } else if (type.equals("FETCH_MATERIAL")) {
                return context.deserialize(jsonObject, FetchMaterialTask.class);
            } else if (type.equals("FETCH_ARTIFACT")) {
                return context.deserialize(jsonObject, FetchArtifactTask.class);
            }

            return context.deserialize(jsonObject, Class.forName("com.googlecode.whiteboard.model." + type));
        } catch (ClassNotFoundException cnfe) {
            throw new JsonParseException("Unknown element type: " + type, cnfe);
        }
    }
}