CassandraGatewayImpl.java

/*
 * Copyright 2013 Netflix, Inc.
 *
 *      Licensed 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.netflix.nicobar.cassandra;

import java.util.Map;
import java.util.concurrent.Future;

import com.netflix.astyanax.Keyspace;
import com.netflix.astyanax.model.ColumnList;
import com.netflix.astyanax.model.Rows;
import com.netflix.nicobar.cassandra.internal.HystrixCassandraDeleteColumns;
import com.netflix.nicobar.cassandra.internal.HystrixCassandraDeleteRow;
import com.netflix.nicobar.cassandra.internal.HystrixCassandraGetRow;
import com.netflix.nicobar.cassandra.internal.HystrixCassandraGetRowsByKeys;
import com.netflix.nicobar.cassandra.internal.HystrixCassandraGetRowsByQuery;
import com.netflix.nicobar.cassandra.internal.HystrixCassandraPut;

/**
 * Concrete implementation of CassandraGateway, using Cassandra Hystrix commands.
 *
 * It is not ideal that this behavior is provided by nicobar-cassandra. Preferably,
 * Astyanax itself should expose hystrix protected cassandra operations that we
 * can directly use.
 *
 * @author Vasanth Asokan
 */
public class CassandraGatewayImpl implements CassandraGateway {

    private final Keyspace keyspace;
    private final String columnFamily;

    public CassandraGatewayImpl(Keyspace keyspace, String cf) {
        this.keyspace = keyspace;
        this.columnFamily = cf;
    }

    @Override
    public Keyspace getKeyspace() {
        return this.keyspace;
    }

    @Override
    public String getColumnFamily() {
        return this.columnFamily;
    }

    @Override
    public void upsert(String rowKey, Map<String, Object> attributes) {
        new HystrixCassandraPut<String>(keyspace, columnFamily, rowKey, attributes).execute();
    }

    @Override
    public void upsert(String rowKey, Map<String, Object> attributes, int ttlSeconds) {
        new HystrixCassandraPut<String>(keyspace, columnFamily, rowKey, attributes, ttlSeconds).execute();
    }

    @Override
    public ColumnList<String> getRow(String rowKey) {
        return new HystrixCassandraGetRow<String>(keyspace, columnFamily, rowKey).execute();
    }

    @Override
    public Rows<String, String> getRows(String... rowKeys) {
        return new HystrixCassandraGetRowsByKeys<String>(keyspace, columnFamily, rowKeys).execute();
    }

    @Override
    public Rows<String, String> select(String cql) {
        return new HystrixCassandraGetRowsByQuery<String>(keyspace, columnFamily, String.class, cql).execute();
    }

    @Override
    public Future<Rows<String, String>> selectAsync(String cql) {
        return new HystrixCassandraGetRowsByQuery<String>(keyspace, columnFamily, String.class, cql).queue();
    }

    @Override
    public ColumnList<String> getColumns(String rowKey, String... columns) {
        return new HystrixCassandraGetRow<String>(keyspace, columnFamily, rowKey).withColumns(columns).execute();
    }

    @Override
    public void deleteRow(String rowKey) {
        new HystrixCassandraDeleteRow<String>(keyspace, columnFamily, rowKey).execute();
    }

    @Override
    public void deleteColumn(String rowKey, String column) {
        new HystrixCassandraDeleteColumns<String>(keyspace, columnFamily, rowKey, column);
    }
}