com.spidertracks.datanucleus.CassandraTest.java Source code

Introduction

Here is the source code for com.spidertracks.datanucleus.CassandraTest.java

Source

/**********************************************************************
Copyright (c) 2010 Todd Nine. All rights reserved.
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.
    
Contributors :
...
 ***********************************************************************/
package com.spidertracks.datanucleus;

import java.util.List;
import java.util.Map;

import javax.jdo.JDOHelper;
import javax.jdo.PersistenceManagerFactory;

import org.apache.cassandra.thrift.Column;
import org.apache.cassandra.thrift.ConsistencyLevel;
import org.apache.cassandra.thrift.KeyRange;
import org.apache.cassandra.thrift.SlicePredicate;
import org.apache.cassandra.service.CassandraDaemon;
import org.apache.commons.codec.binary.Hex;
import org.junit.BeforeClass;
import org.scale7.cassandra.pelops.Bytes;
import org.scale7.cassandra.pelops.Pelops;
import org.scale7.cassandra.pelops.RowDeletor;
import org.scale7.cassandra.pelops.Selector;
import org.jboss.netty.logging.InternalLoggerFactory;
import org.jboss.netty.logging.Slf4JLoggerFactory;

/**
 * @author Todd Nine
 * 
 */
public abstract class CassandraTest {
    protected static PersistenceManagerFactory pmf;

    static {
        InternalLoggerFactory.setDefaultFactory(new Slf4JLoggerFactory());
    }

    /**
     * Set embedded cassandra up and spawn it in a new thread.
     * 
     * @throws Exception
     */
    @BeforeClass
    public static void setup() throws Exception {
        try {
            Server.INSTANCE.start();
        } catch (Exception e) {
            e.printStackTrace();
            System.exit(1);
        }
        pmf = Server.INSTANCE.getFactory();
    }

    private enum Server {

        INSTANCE;

        protected CassandraDaemon cassandraServer;
        protected PersistenceManagerFactory pmf;

        public void start() throws Exception {
            if (cassandraServer == null) {
                cassandraServer = new CassandraDaemon();
                cassandraServer.activate();

                // wait until cassandra server starts up. could wait less time,
                // but
                // 2 seconds to be sure.
                //Thread.sleep(2000);

                pmf = JDOHelper.getPersistenceManagerFactory("Test");
            }
        }

        public PersistenceManagerFactory getFactory() {
            return pmf;
        }
    }

    /**
     * Delete all rows in the CF
     * 
     * @param cfName
     */
    protected void deleteAllRows(String cfName) {

        int maxSize = 100;

        Selector selector = Pelops.createSelector("TestPool");

        SlicePredicate predicate = Selector.newColumnsPredicateAll(false);

        RowDeletor deletor = Pelops.createRowDeletor("TestPool", System.nanoTime() / 1000);

        Bytes lastKey = Bytes.fromByteArray(new byte[] {});

        Map<Bytes, List<Column>> results = null;

        do {

            KeyRange range = new KeyRange();
            range.setStart_key(lastKey.toByteArray());
            range.setEnd_key(new byte[] {});
            range.setCount(maxSize);

            results = selector.getColumnsFromRows(cfName, range, predicate, ConsistencyLevel.QUORUM);

            for (Bytes key : results.keySet()) {
                deletor.deleteRow(cfName, key, ConsistencyLevel.QUORUM);
                lastKey = key;
            }

        } while (results.size() == maxSize);

    }

    /**
     * Print all rows as hex in the cf name
     * @param cfName
     */
    protected void printAllRows(String cfName) {
        Selector selector = Pelops.createSelector("TestPool");

        SlicePredicate predicate = Selector.newColumnsPredicateAll(false);

        Bytes lastKey = Bytes.fromByteArray(new byte[] {});

        Map<Bytes, List<Column>> rawResults = null;

        do {

            KeyRange range = new KeyRange();
            range.setStart_key(lastKey.toByteArray());
            range.setEnd_key(new byte[] {});
            range.setCount(1000);

            rawResults = selector.getColumnsFromRows("InheritanceParent", range, predicate,
                    ConsistencyLevel.QUORUM);

            for (Bytes key : rawResults.keySet()) {
                System.out.println(String.format("Row: %s", new String(Hex.encodeHex(key.toByteArray()))));

                for (Column col : rawResults.get(key)) {
                    System.out.println(String.format("\tColumn: %s ; Value: %s",
                            new String(Hex.encodeHex(col.getName())), new String(Hex.encodeHex(col.getValue()))));
                }
            }

        } while (rawResults.size() == 1000);
    }
}