/*
* Hibernate OGM, Domain model persistence for NoSQL datastores
*
* License: GNU Lesser General Public License (LGPL), version 2.1 or later
* See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
*/
package org.hibernate.ogm.datastore.map.impl;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import org.hibernate.ogm.datastore.spi.Association;
import org.hibernate.ogm.datastore.spi.AssociationOperation;
import org.hibernate.ogm.datastore.spi.Tuple;
import org.hibernate.ogm.datastore.spi.TupleOperation;
import org.hibernate.ogm.datastore.spi.TupleSnapshot;
import org.hibernate.ogm.grid.AssociationKey;
import org.hibernate.ogm.grid.RowKey;
/**
* @author Sanne Grinovero <sanne@hibernate.org> (C) 2011 Red Hat Inc.
*/
public final class MapHelpers {
private MapHelpers() {
// not meant to be instantiated
}
public static void applyTupleOpsOnMap(Tuple tuple, Map<String, Object> map) {
for ( TupleOperation action : tuple.getOperations() ) {
switch ( action.getType() ) {
case PUT_NULL:
case PUT:
map.put( action.getColumn(), action.getValue() );
break;
case REMOVE:
map.remove( action.getColumn() );
break;
}
}
}
public static Map<String, Object> tupleToMap(Tuple tuple) {
if (tuple == null) {
return null;
}
Map<String, Object> snapshot;
TupleSnapshot snapshotInstance = tuple.getSnapshot();
if ( snapshotInstance.isEmpty() ) {
//new assoc tuples are made of EmptyTupleSnapshot
snapshot = Collections.emptyMap();
}
else {
//loaded assoc tuples are made of MapTupleSnapshot
snapshot = ( (MapTupleSnapshot) snapshotInstance ).getMap();
}
Map<String, Object> map = new HashMap<String, Object>( snapshot );
MapHelpers.applyTupleOpsOnMap( tuple, map );
return map;
}
public static void updateAssociation(Association association, AssociationKey key) {
Map<RowKey, Map<String, Object>> atomicMap = ( (MapAssociationSnapshot) association.getSnapshot() ).getUnderlyingMap();
for ( AssociationOperation action : association.getOperations() ) {
switch ( action.getType() ) {
case CLEAR:
atomicMap.clear();
case PUT_NULL:
case PUT:
atomicMap.put( action.getKey(), MapHelpers.tupleToMap( action.getValue() ) );
break;
case REMOVE:
atomicMap.remove( action.getKey() );
break;
}
}
}
}