private static void MoveDefaultDiscriminator( StorageMappingFragment fromFragment, StorageMappingFragment toFragment) { var discriminatorColumn = fromFragment.GetDefaultDiscriminator(); if (discriminatorColumn != null) { var discriminator = fromFragment.ColumnConditions.SingleOrDefault( cc => cc.ColumnProperty == discriminatorColumn); if (discriminator != null) { fromFragment.RemoveDefaultDiscriminatorAnnotation(); fromFragment.RemoveConditionProperty(discriminator); toFragment.AddDiscriminatorCondition(discriminator.ColumnProperty, discriminator.Value); toFragment.SetDefaultDiscriminator(discriminator.ColumnProperty); } } }
public static EdmProperty RemoveDefaultDiscriminatorCondition( this StorageMappingFragment entityTypeMappingFragment) { DebugCheck.NotNull(entityTypeMappingFragment); var discriminatorColumn = entityTypeMappingFragment.GetDefaultDiscriminator(); if (discriminatorColumn != null && entityTypeMappingFragment.ColumnConditions.Any()) { Debug.Assert(entityTypeMappingFragment.ColumnConditions.Count() == 1); entityTypeMappingFragment.RemoveConditionProperty( entityTypeMappingFragment.ColumnConditions.Single()); } entityTypeMappingFragment.RemoveDefaultDiscriminatorAnnotation(); return(discriminatorColumn); }
/// <summary> /// Makes sure only the required property mappings are present /// </summary> private static void ConfigureTypeMappings( TableMapping tableMapping, Dictionary <EntityType, StorageEntityTypeMapping> rootMappings, EntityType entityType, StorageMappingFragment propertiesTypeMappingFragment, StorageMappingFragment conditionTypeMappingFragment) { var existingPropertyMappings = new List <ColumnMappingBuilder>( propertiesTypeMappingFragment.ColumnMappings.Where(pm => !pm.ColumnProperty.IsPrimaryKeyColumn)); var existingConditions = new List <StorageConditionPropertyMapping>(propertiesTypeMappingFragment.ColumnConditions); foreach (var columnMapping in from cm in tableMapping.ColumnMappings from pm in cm.PropertyMappings where pm.EntityType == entityType select new { cm.Column, Property = pm }) { if (columnMapping.Property.PropertyPath != null && !IsRootTypeMapping( rootMappings, columnMapping.Property.EntityType, columnMapping.Property.PropertyPath)) { var existingPropertyMapping = propertiesTypeMappingFragment.ColumnMappings.SingleOrDefault( x => x.PropertyPath == columnMapping.Property.PropertyPath); if (existingPropertyMapping != null) { existingPropertyMappings.Remove(existingPropertyMapping); } else { existingPropertyMapping = new ColumnMappingBuilder(columnMapping.Column, columnMapping.Property.PropertyPath); propertiesTypeMappingFragment.AddColumnMapping(existingPropertyMapping); } } if (columnMapping.Property.Conditions != null) { foreach (var condition in columnMapping.Property.Conditions) { if (conditionTypeMappingFragment.ColumnConditions.Contains(condition)) { existingConditions.Remove(condition); } else if (!entityType.Abstract) { conditionTypeMappingFragment.AddConditionProperty(condition); } } } } // Any leftover mappings are removed foreach (var leftoverPropertyMapping in existingPropertyMappings) { propertiesTypeMappingFragment.RemoveColumnMapping(leftoverPropertyMapping); } foreach (var leftoverCondition in existingConditions) { conditionTypeMappingFragment.RemoveConditionProperty(leftoverCondition); } if (entityType.Abstract) { propertiesTypeMappingFragment.ClearConditions(); } }