com.algorithmia.algo.AlgoSuccess Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of algorithmia-client Show documentation
algorithmia-client
There is a newer version: 1.0.16
package com.algorithmia.algo;

import com.algorithmia.AlgorithmException;
import com.algorithmia.TypeToken;

import com.google.gson.Gson;
import com.google.gson.JsonElement;
import com.google.gson.JsonPrimitive;
import org.apache.commons.codec.binary.Base64;

import java.lang.UnsupportedOperationException;
import java.lang.reflect.Type;

/**
 * A result representing success, depending on how the algorithm was called it may have
 * an async response, rawOutput, or metadata + result.
 */
public final class AlgoSuccess extends AlgoResponse {

    private transient JsonElement result;
    private Metadata metadata;
    private String resultJson;
    private String rawOutput;
    private AlgoAsyncResponse asyncResponse;

    private static transient final Gson gson = new Gson();
    private static transient final Type byteType = new TypeToken(){}.getType();

    public AlgoSuccess(JsonElement result, Metadata metadata, String rawOutput, AlgoAsyncResponse asyncResponse) {
        this.result = result;
        this.metadata = metadata;
        if (result != null) {
            this.resultJson = result.toString();
        } else {
            this.resultJson = null;
        }
        this.rawOutput = rawOutput;
        this.asyncResponse = asyncResponse;
    }

    @Override
    public boolean isSuccess() {
        return true;
    }

    @Override
    public boolean isFailure() {
        return false;
    }

    @Override
    public Metadata getMetadata() {
        return metadata;
    }

    @Override
    public AlgoAsyncResponse getAsyncResponse() throws AlgorithmException {
        return asyncResponse;
    }

    @Override
    @SuppressWarnings("unchecked")
    protected  T as(Class returnClass) {
        if (result == null && resultJson != null) {
            result = gson.toJsonTree(resultJson);
        }
        if(metadata.getContentType() == ContentType.Void) {
            return null;
        } else if(metadata.getContentType() == ContentType.Text) {
            return gson.fromJson(new JsonPrimitive(result.getAsString()), returnClass);
        } else if(metadata.getContentType() == ContentType.Json) {
            return gson.fromJson(result, returnClass);
        } else if(metadata.getContentType() == ContentType.Binary) {
            if(byte[].class == returnClass) {
              return (T)Base64.decodeBase64(result.getAsString());
            } else {
              throw new UnsupportedOperationException("Only support returning as byte[] for Binary data");
            }

        } else {
            throw new UnsupportedOperationException("Unknown ContentType in response: " + metadata.getContentType().toString());
        }

    }

    @Override
    @SuppressWarnings("unchecked")
    protected  T as(Type returnType) {
        if (result == null && resultJson != null) {
            result = gson.toJsonTree(resultJson);
        }

        if(metadata.getContentType() == ContentType.Void) {
            return null;
        } else if(metadata.getContentType() == ContentType.Text) {
            return gson.fromJson(new JsonPrimitive(result.getAsString()), returnType);
        } else if(metadata.getContentType() == ContentType.Json) {
            return gson.fromJson(result, returnType);
        } else if(metadata.getContentType() == ContentType.Binary) {
            if(byteType.equals(returnType)) {
              return (T)Base64.decodeBase64(result.getAsString());
            } else {
              throw new UnsupportedOperationException("Only support returning as byte[] for Binary data");
            }
        } else {
            throw new UnsupportedOperationException("Unknown ContentType in response: " + metadata.getContentType().toString());
        }
    }

    @Override
    public String asJsonString() {
        return resultJson;
    }

    @Override
    public String asString() {
        return as(String.class);
    }

    @Override
    public String getRawOutput() throws AlgorithmException {
        return rawOutput;
    }

}