com.basho.riak.client.query.IndexMapReduce.java Source code

Introduction

Here is the source code for com.basho.riak.client.query.IndexMapReduce.java

Source

/*
 * This file is provided to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */
package com.basho.riak.client.query;

import java.io.IOException;

import com.basho.riak.client.raw.RawClient;
import com.basho.riak.client.raw.query.indexes.IndexQuery;
import com.basho.riak.client.raw.query.indexes.IndexWriter;
import com.fasterxml.jackson.core.JsonGenerator;

/**
 * A {@link MapReduce} operation that takes a 2i index query as input
 * 
 * @author russell
 * 
 */
public class IndexMapReduce extends MapReduce {

    private static final String BUCKET = "bucket";
    private static final String INDEX = "index";
    private static final String KEY = "key";
    private static final String START = "start";
    private static final String END = "end";

    private final IndexQuery indexQuery;

    /**
     * @param client
     */
    public IndexMapReduce(RawClient client, IndexQuery indexQuery) {
        super(client);
        this.indexQuery = indexQuery;
    }

    /*
     * (non-Javadoc)
     * 
     * @see
     * com.basho.riak.client.query.MapReduce#writeInput(org.codehaus.jackson
     * .JsonGenerator)
     */
    @Override
    protected void writeInput(final JsonGenerator jsonGenerator) throws IOException {
        jsonGenerator.writeStartObject();

        IndexWriter e = new IndexWriter() {

            private void writeCommon(String bucket, String index) throws IOException {
                jsonGenerator.writeStringField(BUCKET, bucket);
                jsonGenerator.writeStringField(INDEX, index);
            }

            public void write(String bucket, String index, long from, long to) throws IOException {
                writeCommon(bucket, index);
                jsonGenerator.writeNumberField(START, from);
                jsonGenerator.writeNumberField(END, to);
            }

            public void write(String bucket, String index, long value) throws IOException {
                writeCommon(bucket, index);
                jsonGenerator.writeNumberField(KEY, value);
            }

            public void write(String bucket, String index, String from, String to) throws IOException {
                writeCommon(bucket, index);
                jsonGenerator.writeStringField(START, from);
                jsonGenerator.writeStringField(END, to);
            }

            public void write(String bucket, String index, String value) throws IOException {
                writeCommon(bucket, index);
                jsonGenerator.writeStringField(KEY, value);
            }
        };

        indexQuery.write(e);
        jsonGenerator.writeEndObject();
    }
}