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

The newest version!

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();
    }
}