package com.apigee.apix.cassandra.test;

import me.prettyprint.cassandra.serializers.CompositeSerializer;
import me.prettyprint.cassandra.serializers.LongSerializer;
import me.prettyprint.cassandra.serializers.StringSerializer;
import me.prettyprint.hector.api.beans.AbstractComposite;
import me.prettyprint.hector.api.beans.ColumnSlice;
import me.prettyprint.hector.api.beans.Composite;
import me.prettyprint.hector.api.beans.HColumn;
import me.prettyprint.hector.api.factory.HFactory;
import me.prettyprint.hector.api.query.QueryResult;
import me.prettyprint.hector.api.query.SliceQuery;

import java.util.SortedSet;
import java.util.TreeSet;

/**
 * Created with IntelliJ IDEA.
 * User: thameema
 * Date: 10/19/12
 * Time: 10:25 AM
 * To change this template use File | Settings | File Templates.
 */
public class CassTest extends Base {

    public static void main(String args[]){
        init();
        CassTest ct = new CassTest();
        long input = 168430082;

        String s = ct.query(input, input);
        System.out.println("Value is:"+s);
    }

    private String cf = "operator_ips2";

    public String query(long component1Value, long component2Value) {
        SliceQuery<String, Composite, String> sq = HFactory.createSliceQuery(tutorialKeyspace, StringSerializer.get(), new CompositeSerializer(), StringSerializer.get());

        sq.setColumnFamily(cf);
        sq.setKey("ip");

        Composite startRange = new Composite();
        startRange.addComponent(component1Value, LongSerializer.get(), "LongType", AbstractComposite.ComponentEquality.GREATER_THAN_EQUAL);
        startRange.addComponent(component2Value, LongSerializer.get(), "LongType", AbstractComposite.ComponentEquality.LESS_THAN_EQUAL);


        Composite endRange = new Composite();
        endRange.addComponent(component1Value, LongSerializer.get(), "LongType", AbstractComposite.ComponentEquality.GREATER_THAN_EQUAL);
        endRange.addComponent(1l, LongSerializer.get(), "LongType", AbstractComposite.ComponentEquality.EQUAL);

        sq.setRange(startRange, endRange, false, 10);

        QueryResult<ColumnSlice<Composite,String>> result = sq.execute();
        ColumnSlice<Composite, String> slice = result.get();

        return slice.getColumns().get(0).getValue();
    }

}