/********************************************************************** Copyright (c) 2006 Andy Jefferson and others. 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. Contributors: ... **********************************************************************/ package org.datanucleus.store.fieldmanager; import java.util.Map.Entry; import org.datanucleus.ClassLoaderResolver; import org.datanucleus.api.ApiAdapter; import org.datanucleus.metadata.AbstractMemberMetaData; import org.datanucleus.metadata.RelationType; import org.datanucleus.state.ObjectProvider; import org.datanucleus.store.types.ContainerHandler; import org.datanucleus.store.types.ElementContainerAdapter; import org.datanucleus.store.types.ElementContainerHandler; import org.datanucleus.store.types.MapContainerAdapter; import org.datanucleus.store.types.SCO; import org.datanucleus.store.types.SCOUtils; import org.datanucleus.store.types.SequenceAdapter; import org.datanucleus.store.types.TypeManager; /** * Field manager that persists all unpersisted PC objects referenced from the source object. * If any collection/map fields are not currently using SCO wrappers they will be converted to do so. * Effectively provides "persistence-by-reachability" (at insert/update). */ public class PersistFieldManager extends AbstractFieldManager { /** ObjectProvider for the owning object. */ private final ObjectProvider op; /** Whether this manager will replace any SCO fields with SCO wrappers. */ private final boolean replaceSCOsWithWrappers; /** * Constructor. * @param op The ObjectProvider for the object. * @param replaceSCOsWithWrappers Whether to swap any SCO field objects for SCO wrappers */ public PersistFieldManager(ObjectProvider op, boolean replaceSCOsWithWrappers) { this.op = op; this.replaceSCOsWithWrappers = replaceSCOsWithWrappers; } /** * Utility method to process the passed persistable object. * @param pc The PC object * @param ownerFieldNum Field number of owner where this is embedded * @param objectType Type of object (see org.datanucleus.ObjectProvider) * @return The processed persistable object */ protected Object processPersistable(Object pc, int ownerFieldNum, int objectType) { // TODO Consider adding more of the functionality in SCOUtils.validateObjectForWriting ApiAdapter adapter = op.getExecutionContext().getApiAdapter(); if (!adapter.isPersistent(pc) || (adapter.isPersistent(pc) && adapter.isDeleted(pc))) { // Object is TRANSIENT/DETACHED and being persisted, or P_NEW_DELETED and being re-persisted if (objectType != ObjectProvider.PC) { return op.getExecutionContext().persistObjectInternal(pc, op, ownerFieldNum, objectType); } return op.getExecutionContext().persistObjectInternal(pc, null, -1, objectType); } return pc; } /** * Method to store an object field. * @param fieldNumber Number of the field (absolute) * @param value Value of the field */ public void storeObjectField(int fieldNumber, Object value) { if (value != null) { if (replaceSCOsWithWrappers) { // Replace any SCO field that isn't already a wrapper, with its wrapper object boolean[] secondClassMutableFieldFlags = op.getClassMetaData().getSCOMutableMemberFlags(); if (secondClassMutableFieldFlags[fieldNumber] && !(value instanceof SCO)) { // Replace the field with a SCO wrapper value = SCOUtils.wrapSCOField(op, fieldNumber, value, true); } } AbstractMemberMetaData mmd = op.getClassMetaData().getMetaDataForManagedMemberAtAbsolutePosition(fieldNumber); if (mmd.isCascadePersist()) { ClassLoaderResolver clr = op.getExecutionContext().getClassLoaderResolver(); RelationType relationType = mmd.getRelationType(clr); if (relationType != RelationType.NONE) { if (mmd.hasContainer()) { processContainer(fieldNumber, value, mmd); } else { // Process PC fields if (mmd.isEmbedded() || mmd.isSerialized()) { processPersistable(value, fieldNumber, ObjectProvider.EMBEDDED_PC); } else { processPersistable(value, -1, ObjectProvider.PC); } } } } } } private void processContainer(int fieldNumber, Object container, AbstractMemberMetaData mmd) { if (mmd.hasMap()) { processMapContainer(fieldNumber, container, mmd); } else { processElementContainer(fieldNumber, container, mmd); } } private void processMapContainer(int fieldNumber, Object container, AbstractMemberMetaData mmd) { TypeManager typeManager = op.getExecutionContext().getTypeManager(); ContainerHandler<Object, MapContainerAdapter<Object>> containerHandler = typeManager.getContainerHandler(mmd.getType()); ApiAdapter api = op.getExecutionContext().getApiAdapter(); // Process all keys, values of the Map that are PC MapContainerAdapter<Object> mapAdapter = containerHandler.getAdapter(container); for (Entry<Object, Object> entry : mapAdapter.entries()) { Object mapKey = entry.getKey(); Object mapValue = entry.getValue(); Object newMapKey = mapKey; Object newMapValue = mapValue; if (api.isPersistable(mapKey)) { // Persist (or attach) the key int mapKeyObjectType = mmd.getMap().isEmbeddedKey() || mmd.getMap().isSerializedKey() ? ObjectProvider.EMBEDDED_MAP_KEY_PC : ObjectProvider.PC; newMapKey = processPersistable(mapKey, fieldNumber, mapKeyObjectType); } if (api.isPersistable(mapValue)) { // Persist (or attach) the value int mapValueObjectType = mmd.getMap().isEmbeddedValue() || mmd.getMap().isSerializedValue() ? ObjectProvider.EMBEDDED_MAP_VALUE_PC : ObjectProvider.PC; newMapValue = processPersistable(mapValue, fieldNumber, mapValueObjectType); } if (newMapKey != mapKey || newMapValue != mapValue) { // Maybe we have just have attached key or value boolean updateKey = false; boolean updateValue = false; if (newMapKey != mapKey) { ObjectProvider keyOP = op.getExecutionContext().findObjectProvider(newMapKey); if (keyOP.getReferencedPC() != null) { // Attaching the key updateKey = true; } } if (newMapValue != mapValue) { ObjectProvider valOP = op.getExecutionContext().findObjectProvider(newMapValue); if (valOP.getReferencedPC() != null) { // Attaching the value updateValue = true; } } if (updateKey) { mapAdapter.remove(mapKey); mapAdapter.put(newMapKey, updateValue ? newMapValue : mapValue); } else if (updateValue) { mapAdapter.put(mapKey, newMapValue); } } } } private void processElementContainer(int fieldNumber, Object container, AbstractMemberMetaData mmd) { TypeManager typeManager = op.getExecutionContext().getTypeManager(); ElementContainerHandler<Object, ElementContainerAdapter<Object>> elementContainerHandler = typeManager.getContainerHandler(mmd.getType()); // Process all elements of the container that are PC ElementContainerAdapter containerAdapter = elementContainerHandler.getAdapter(container); ApiAdapter api = op.getExecutionContext().getApiAdapter(); int objectType = elementContainerHandler.getObjectType(mmd); if (objectType == ObjectProvider.PC) { int elementPosition = 0; for (Object element : containerAdapter) { if (api.isPersistable(element)) { Object newElement = processPersistable(element, -1, objectType); ObjectProvider elementSM = op.getExecutionContext().findObjectProvider(newElement); if (elementSM.getReferencedPC() != null) { // Must be attaching this element, so swap element (detached -> attached) if (containerAdapter instanceof SequenceAdapter) { ((SequenceAdapter) containerAdapter).update(newElement, elementPosition); } else { containerAdapter.remove(elementSM); containerAdapter.add(newElement); } } } elementPosition++; } } else { // Embedded/Serialized for (Object element : containerAdapter) { if (api.isPersistable(element)) { processPersistable(element, fieldNumber, objectType); } } } } public void storeBooleanField(int fieldNumber, boolean value) { // Do nothing } public void storeByteField(int fieldNumber, byte value) { // Do nothing } public void storeCharField(int fieldNumber, char value) { // Do nothing } public void storeDoubleField(int fieldNumber, double value) { // Do nothing } public void storeFloatField(int fieldNumber, float value) { // Do nothing } public void storeIntField(int fieldNumber, int value) { // Do nothing } public void storeLongField(int fieldNumber, long value) { // Do nothing } public void storeShortField(int fieldNumber, short value) { // Do nothing } public void storeStringField(int fieldNumber, String value) { // Do nothing } }