| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you 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.polygene.runtime.unitofwork; |
| |
| import java.time.Instant; |
| import java.util.HashMap; |
| import java.util.Map; |
| import org.apache.polygene.api.common.QualifiedName; |
| import org.apache.polygene.api.entity.EntityDescriptor; |
| import org.apache.polygene.api.entity.EntityReference; |
| import org.apache.polygene.api.util.Classes; |
| import org.apache.polygene.spi.entity.EntityState; |
| import org.apache.polygene.spi.entity.EntityStatus; |
| import org.apache.polygene.spi.entity.ManyAssociationState; |
| import org.apache.polygene.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 entityReference() |
| { |
| return reference; |
| } |
| |
| @Override |
| public String version() |
| { |
| return ""; |
| } |
| |
| @Override |
| public Instant lastModified() |
| { |
| return Instant.MIN; |
| } |
| |
| @Override |
| public void remove() |
| { |
| } |
| |
| @Override |
| public EntityStatus status() |
| { |
| return EntityStatus.NEW; |
| } |
| |
| @Override |
| public boolean isAssignableTo( Class<?> type ) |
| { |
| return Classes.exactTypeSpecification( type ).test( 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 ) |
| { |
| return manyAssociations.computeIfAbsent( stateName, |
| qualifiedName -> new BuilderManyAssociationState() ); |
| } |
| |
| @Override |
| public NamedAssociationState namedAssociationValueOf( QualifiedName stateName ) |
| { |
| return namedAssociations.computeIfAbsent( stateName, |
| qualifiedName -> new BuilderNamedAssociationState() ); |
| } |
| |
| 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 ) ); |
| } |
| } |
| } |
| } |