com.basho.riak.client.api.commands.mapreduce.FunctionPhaseSerializer 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

There is a newer version: 2.1.1

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 FunctionPhase} to a Riak MapReduce JSON format.
 */
class FunctionPhaseSerializer extends JsonSerializer
{

	private void writePhaseFunction(JsonGenerator jg, FunctionPhase phase) throws IOException
	{
		Function function = phase.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", phase.isKeep());
		jg.writeObjectField("arg", phase.getArg());

		jg.writeEndObject();
	}

	@Override
	public void serialize(FunctionPhase phase, JsonGenerator jg, SerializerProvider sp) throws IOException
	{

		jg.writeStartObject();

		jg.writeFieldName(phase.getType().toString());
		writePhaseFunction(jg, phase);

		jg.writeEndObject();

	}
}