com.basho.riak.client.api.commands.mapreduce.PhaseFunctionSerializer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of riak-client Show documentation

HttpClient-based client for Riak

The newest version!

package com.basho.riak.client.api.commands.mapreduce;

import com.basho.riak.client.core.query.functions.Function;
import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.SerializerProvider;

import java.io.IOException;

/**
 * Serializes a {@link PhaseFunction} to a Riak MapReduce JSON format.
 */
class PhaseFunctionSerializer extends JsonSerializer
{
    @Override
    public void serialize(PhaseFunction phaseFunction, JsonGenerator jg, SerializerProvider sp) throws IOException
    {
        Function function = phaseFunction.getFunction();

        jg.writeStartObject();

        jg.writeStringField("language", function.isJavascript() ? "javascript" : "erlang");

        if (function.isJavascript())
        {
            if (function.isNamed())
            {
                jg.writeStringField("name", function.getName());
            } else if (function.isStored())
            {
                jg.writeStringField("bucket", function.getBucket());
                jg.writeStringField("key", function.getKey());
            } else if (function.isAnonymous())
            {
                jg.writeStringField("source", function.getSource());
            } else
            {
                throw new IllegalStateException("Cannot determine function type");
            }
        } else if (!function.isJavascript())
        {
            jg.writeStringField("module", function.getModule());
            jg.writeStringField("function", function.getFunction());
        }

        jg.writeBooleanField("keep", phaseFunction.isKeep());

        jg.writeEndObject();
    }
}