com.basho.riak.client.api.commands.timeseries.Store 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.timeseries;

import com.basho.riak.client.api.AsIsRiakCommand;
import com.basho.riak.client.core.operations.ts.StoreOperation;
import com.basho.riak.client.core.query.timeseries.Row;

import java.util.Collection;
import java.util.LinkedList;
import java.util.List;

/**
 * Time Series Store Command
 * Allows you to store data into a Time Series table.
 * Each row to be stored must have it's cells ordered the same as the table definition.
 * To view the table definition, execute riak-admin bucket-type status 
 * on any Riak node in the cluster.
 *
 * @author Alex Moore 
 * @author Sergey Galkin 
 * @since 2.0.3
 */
public class Store extends AsIsRiakCommand
{
    private final Builder builder;

    private Store (Builder builder)
    {
        this.builder = builder;
    }

    @Override
    protected StoreOperation buildCoreOperation()
    {
        return new StoreOperation.Builder(builder.tableName)
                .withRows(builder.rows)
                .build();
    }

    /**
     * Used to construct a Time Series Store command.
     */
    public static class Builder
    {
        private final String tableName;
        // TODO: Think about using a flattening iterable here.
        private final List rows = new LinkedList<>();

        /**
         * Construct a Builder for a Time Series Store command.
         * @param tableName Required. The name of the table to store data to.
         */
        public Builder(String tableName)
        {
            this.tableName = tableName;
        }

        /**
         * Add a single Row object to the store command.
         * @param row Required. The row to add.
         * @return a reference to this object.
         */
        public Builder withRow(Row row)
        {
            this.rows.add(row);
            return this;
        }

        /**
         * Add a collection of Row objects to the store command.
         * @param rows Required. The rows to add.
         * @return a reference to this object.
         */
        public Builder withRows(Iterable rows)
        {
            if (rows instanceof Collection)
            {
                this.rows.addAll((Collection) rows);
            }
            else
            {
                // A bit weird but have no other ideas
                for (Row r : rows)
                {
                    this.rows.add(r);
                }
            }
            return this;
        }

        /**
         * Construct a Time Series Store object.
         * @return a new Time Series Store instance.
         */
        public Store build()
        {
            return new Store(this);
        }
    }
}