/*
* Copyright (c) 2009-2011, Rickard Öberg. All Rights Reserved.
* Copyright (c) 2009-2013, Niclas Hedhman. All Rights Reserved.
* Copyright (c) 2014, Paul Merlin. 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.
*/
package org.qi4j.runtime.unitofwork;
import java.util.HashMap;
import java.util.Map;
import org.qi4j.api.common.QualifiedName;
import org.qi4j.api.entity.EntityDescriptor;
import org.qi4j.api.entity.EntityReference;
import org.qi4j.api.util.Classes;
import org.qi4j.spi.entity.EntityState;
import org.qi4j.spi.entity.EntityStatus;
import org.qi4j.spi.entity.ManyAssociationState;
import org.qi4j.spi.entity.NamedAssociationState;
/**
* Implementation of EntityState for use through EntityBuilder.
*/
public final class BuilderEntityState
implements EntityState
{
private final EntityDescriptor entityType;
private final EntityReference reference;
private final Map<QualifiedName, Object> properties = new HashMap<>();
private final Map<QualifiedName, EntityReference> associations = new HashMap<>();
private final Map<QualifiedName, ManyAssociationState> manyAssociations = new HashMap<>();
private final Map<QualifiedName, NamedAssociationState> namedAssociations = new HashMap<>();
public BuilderEntityState( EntityDescriptor type, EntityReference reference )
{
this.entityType = type;
this.reference = reference;
}
@Override
public EntityReference identity()
{
return reference;
}
@Override
public String version()
{
return "";
}
@Override
public long lastModified()
{
return 0;
}
@Override
public void remove()
{
}
@Override
public EntityStatus status()
{
return EntityStatus.NEW;
}
@Override
public boolean isAssignableTo( Class<?> type )
{
return Classes.exactTypeSpecification( type ).satisfiedBy( entityType );
}
@Override
public EntityDescriptor entityDescriptor()
{
return entityType;
}
@Override
public Object propertyValueOf( QualifiedName stateName )
{
return properties.get( stateName );
}
@Override
public EntityReference associationValueOf( QualifiedName stateName )
{
return associations.get( stateName );
}
@Override
public void setPropertyValue( QualifiedName stateName, Object newValue )
{
properties.put( stateName, newValue );
}
@Override
public void setAssociationValue( QualifiedName stateName, EntityReference newEntity )
{
associations.put( stateName, newEntity );
}
@Override
public ManyAssociationState manyAssociationValueOf( QualifiedName stateName )
{
ManyAssociationState state = manyAssociations.get( stateName );
if( state == null )
{
state = new BuilderManyAssociationState();
manyAssociations.put( stateName, state );
}
return state;
}
@Override
public NamedAssociationState namedAssociationValueOf( QualifiedName stateName )
{
NamedAssociationState state = namedAssociations.get( stateName );
if( state == null )
{
state = new BuilderNamedAssociationState();
namedAssociations.put( stateName, state );
}
return state;
}
public void copyTo( EntityState newEntityState )
{
for( Map.Entry<QualifiedName, Object> fromPropertyEntry : properties.entrySet() )
{
newEntityState.setPropertyValue( fromPropertyEntry.getKey(), fromPropertyEntry.getValue() );
}
for( Map.Entry<QualifiedName, EntityReference> fromAssociationEntry : associations.entrySet() )
{
newEntityState.setAssociationValue( fromAssociationEntry.getKey(), fromAssociationEntry.getValue() );
}
for( Map.Entry<QualifiedName, ManyAssociationState> fromManyAssociationEntry : manyAssociations.entrySet() )
{
QualifiedName qName = fromManyAssociationEntry.getKey();
ManyAssociationState fromManyAssoc = fromManyAssociationEntry.getValue();
ManyAssociationState toManyAssoc = newEntityState.manyAssociationValueOf( qName );
for( EntityReference entityReference : fromManyAssoc )
{
toManyAssoc.add( 0, entityReference );
}
}
for( Map.Entry<QualifiedName, NamedAssociationState> fromNamedAssociationEntry : namedAssociations.entrySet() )
{
QualifiedName qName = fromNamedAssociationEntry.getKey();
NamedAssociationState fromNamedAssoc = fromNamedAssociationEntry.getValue();
NamedAssociationState toNamedAssoc = newEntityState.namedAssociationValueOf( qName );
for( String name : fromNamedAssoc )
{
toNamedAssoc.put( name, fromNamedAssoc.get( name ) );
}
}
}
}