| /* |
| * 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.apache.zest.runtime.unitofwork; |
| |
| import java.util.HashMap; |
| import java.util.Map; |
| import org.apache.zest.api.common.QualifiedName; |
| import org.apache.zest.api.entity.EntityDescriptor; |
| import org.apache.zest.api.entity.EntityReference; |
| import org.apache.zest.api.util.Classes; |
| import org.apache.zest.spi.entity.EntityState; |
| import org.apache.zest.spi.entity.EntityStatus; |
| import org.apache.zest.spi.entity.ManyAssociationState; |
| import org.apache.zest.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 ) ); |
| } |
| } |
| } |
| } |