com.basho.riak.client.api.commands.mapreduce.IndexInputSerializer 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.Namespace;
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 IndexInput} to a Riak MapReduce JSON format.
 */
public class IndexInputSerializer extends JsonSerializer
{
	@Override
	public void serialize(IndexInput input, JsonGenerator jg, SerializerProvider sp) throws IOException
	{
		jg.writeStartObject();
		
        // Riak bug ... explicitly specifying "default" as the type breaks things
        if (!input.getNamespace().getBucketTypeAsString().equals(Namespace.DEFAULT_BUCKET_TYPE))
        {
            jg.writeArrayFieldStart("bucket");
            jg.writeString(input.getNamespace().getBucketTypeAsString());
            jg.writeString(input.getNamespace().getBucketNameAsString());
            jg.writeEndArray();
        }
        else
        {
            jg.writeStringField("bucket", input.getNamespace().getBucketNameAsString());
        }
        
        jg.writeStringField("index", input.getIndex());
		IndexInput.IndexCriteria criteria = input.getCriteria();
		if (criteria instanceof IndexInput.MatchCriteria)
		{
			IndexInput.MatchCriteria match = (IndexInput.MatchCriteria) criteria;
			jg.writeObjectField("key", match.getValue());
		} else if (criteria instanceof IndexInput.RangeCriteria)
		{
			IndexInput.RangeCriteria range = (IndexInput.RangeCriteria) criteria;
			jg.writeObjectField("start", range.getBegin());
			jg.writeObjectField("end", range.getEnd());
		}
		jg.writeEndObject();
	}
}