me.prettyprint.hom.openjpa.CassandraStore Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hector-object-mapper Show documentation
There is a newer version: 3.0-04
package me.prettyprint.hom.openjpa;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.Map;

import org.apache.openjpa.kernel.OpenJPAStateManager;
import org.apache.openjpa.meta.ClassMetaData;
import org.apache.openjpa.meta.FieldMetaData;
import org.apache.openjpa.meta.JavaTypes;
import org.apache.openjpa.util.OpenJPAId;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.jdbc.core.ColumnMapRowMapper;

import me.prettyprint.cassandra.model.HColumnImpl;
import me.prettyprint.cassandra.model.MutatorImpl;
import me.prettyprint.cassandra.serializers.BytesArraySerializer;
import me.prettyprint.cassandra.serializers.LongSerializer;
import me.prettyprint.cassandra.serializers.StringSerializer;
import me.prettyprint.cassandra.utils.StringUtils;
import me.prettyprint.hector.api.Cluster;
import me.prettyprint.hector.api.Keyspace;
import me.prettyprint.hector.api.Serializer;
import me.prettyprint.hector.api.beans.ColumnSlice;
import me.prettyprint.hector.api.beans.HColumn;
import me.prettyprint.hector.api.factory.HFactory;
import me.prettyprint.hector.api.mutation.Mutator;
import me.prettyprint.hector.api.query.QueryResult;
import me.prettyprint.hector.api.query.SliceQuery;
import me.prettyprint.hom.EntityManagerConfigurator;
import me.prettyprint.hom.openjpa.EntityFacade.ColumnMeta;


/**
 * Holds the {@link Cluster} and {@link Keyspace} references needed
 * for accessing Cassandra
 * 
 * @author zznate
 */
public class CassandraStore {

  private static final Logger log = LoggerFactory.getLogger(CassandraStore.class);
  
  private final Cluster cluster;
  private final CassandraStoreConfiguration conf;
  private Keyspace keyspace;
  private MappingUtils mappingUtils;
  
  public CassandraStore(CassandraStoreConfiguration conf) {
    this.conf = conf;
    this.cluster = HFactory.getCluster(conf.getValue(EntityManagerConfigurator.CLUSTER_NAME_PROP)
        .getOriginalValue());
    // TODO needs passthrough of other configuration
    mappingUtils = new MappingUtils();
  }
  
  public CassandraStore open() {
    this.keyspace = HFactory.createKeyspace(conf.getValue(EntityManagerConfigurator.KEYSPACE_PROP)
        .getOriginalValue(), cluster);
    return this;
  }
  
  public boolean getObject(OpenJPAStateManager stateManager) {

    ClassMetaData metaData = stateManager.getMetaData();
    EntityFacade entityFacade = new EntityFacade(metaData);
    Object idObj = stateManager.getId();
    
    SliceQuery sliceQuery = mappingUtils.buildSliceQuery(idObj, entityFacade, keyspace);   

    //stateManager.storeObject(0, idObj);
    
    QueryResult> result = sliceQuery.execute();

    for (Map.Entry> entry : entityFacade.getColumnMeta().entrySet()) {      
      HColumn column = result.get().getColumnByName(entry.getKey());
      if ( column != null )
        stateManager.storeObject(entry.getValue().fieldId, entry.getValue().serializer.fromBytes(column.getValue()));  
    }            
    return true;
  }
  
  public Mutator storeObject(Mutator mutator, OpenJPAStateManager stateManager, Object idObj) {
    if ( mutator == null )
      mutator = new MutatorImpl(keyspace, BytesArraySerializer.get());
    if ( log.isDebugEnabled() ) {
      log.debug("Adding mutation (insertion) for class {}", stateManager.getManagedInstance().getClass().getName());
    }
    ClassMetaData metaData = stateManager.getMetaData();       
    EntityFacade entityFacade = new EntityFacade(metaData);
    Object field;
    for (Map.Entry> entry : entityFacade.getColumnMeta().entrySet()) {
      field = stateManager.fetch(entry.getValue().fieldId);
      if ( field != null ) {
        mutator.addInsertion(mappingUtils.getKeyBytes(idObj), entityFacade.getColumnFamilyName(), 
            new HColumnImpl(entry.getKey(), field, 
                keyspace.createClock(), StringSerializer.get(),
                entry.getValue().serializer));
      }
          
    }    
    return mutator;
  }
    
  public Mutator removeObject(Mutator mutator, OpenJPAStateManager stateManager, Object idObj) {
    if ( mutator == null )
      mutator = new MutatorImpl(keyspace, BytesArraySerializer.get());
    if ( log.isDebugEnabled() ) {
      log.debug("Adding mutation (deletion) for class {}", stateManager.getManagedInstance().getClass().getName());
    }
    ClassMetaData metaData = stateManager.getMetaData();       
    EntityFacade entityFacade = new EntityFacade(metaData);
    mutator.addDeletion(mappingUtils.getKeyBytes(idObj), entityFacade.getColumnFamilyName(),null,StringSerializer.get());                  
    return mutator;
  }
  

  public Cluster getCluster() {
    return cluster;
  }

  public Keyspace getKeyspace() {
    return keyspace;
  }


  
  
}