public EntityMap GetEntityMap(Type type, ObjectContext objectContext)
{
var entityMap = new EntityMap(type);
var metadata = objectContext.MetadataWorkspace;
// Get the part of the model that contains info about the actual CLR types
var objectItemCollection = ((ObjectItemCollection)metadata.GetItemCollection(DataSpace.OSpace));
// Get the entity type from the model that maps to the CLR type
var entityType = metadata
.GetItems <EntityType>(DataSpace.OSpace)
.Single(e => objectItemCollection.GetClrType(e) == type);
// Get the entity set that uses this entity type
var entitySet = metadata
.GetItems <EntityContainer>(DataSpace.CSpace)
.Single()
.EntitySets
.Single(s => s.ElementType.Name == entityType.Name);
// Find the mapping between conceptual and storage model for this entity set
var mapping = metadata.GetItems <EntityContainerMapping>(DataSpace.CSSpace)
.Single()
.EntitySetMappings
.Single(s => s.EntitySet == entitySet);
// Find the storage entity set (table) that the entity is mapped
var mappingFragment =
(mapping.EntityTypeMappings.SingleOrDefault(a => a.IsHierarchyMapping) ?? mapping.EntityTypeMappings.Single())
.Fragments.Single();
entityMap.ModelType = entityType;
entityMap.ModelSet = entitySet;
entityMap.StoreSet = mappingFragment.StoreEntitySet;
entityMap.StoreType = mappingFragment.StoreEntitySet.ElementType;
// set table
SetTableName(entityMap);
// set properties
SetProperties(entityMap, mappingFragment);
// set keys
SetKeys(entityMap);
return(entityMap);
}
private static void SetTableName(EntityMap entityMap)
{
var builder = new StringBuilder(50);
EntitySet storeSet = entityMap.StoreSet;
string table = null;
string schema = null;
MetadataProperty tableProperty;
MetadataProperty schemaProperty;
storeSet.MetadataProperties.TryGetValue("Table", true, out tableProperty);
if (tableProperty == null || tableProperty.Value == null)
{
storeSet.MetadataProperties.TryGetValue("http://schemas.microsoft.com/ado/2007/12/edm/EntityStoreSchemaGenerator:Table", true, out tableProperty);
}
if (tableProperty != null)
{
table = tableProperty.Value as string;
}
// Table will be null if its the same as Name
if (table == null)
{
table = storeSet.Name;
}
//storeSet.MetadataProperties.TryGetValue("Schema", true, out schemaProperty);
//if (schemaProperty == null || schemaProperty.Value == null)
// storeSet.MetadataProperties.TryGetValue("http://schemas.microsoft.com/ado/2007/12/edm/EntityStoreSchemaGenerator:Schema", true, out schemaProperty);
//if (schemaProperty != null)
// schema = schemaProperty.Value as string;
//if (!string.IsNullOrWhiteSpace(schema))
//{
// builder.Append(QuoteIdentifier(schema));
// builder.Append(".");
//}
builder.Append(QuoteIdentifier(table));
entityMap.TableName = builder.ToString();
}
private static void SetTableName(EntityMap entityMap)
{
EntitySet storeSet = entityMap.StoreSet;
string table = null;
string schema = null;
MetadataProperty tableProperty;
MetadataProperty schemaProperty;
storeSet.MetadataProperties.TryGetValue("Table", true, out tableProperty);
if (tableProperty == null || tableProperty.Value == null)
{
storeSet.MetadataProperties.TryGetValue("http://schemas.microsoft.com/ado/2007/12/edm/EntityStoreSchemaGenerator:Table",
true, out tableProperty);
}
if (tableProperty != null)
{
table = tableProperty.Value as string;
}
// Table will be null if its the same as Name
if (table == null)
{
table = storeSet.Name;
}
storeSet.MetadataProperties.TryGetValue("Schema", true, out schemaProperty);
if (schemaProperty == null || schemaProperty.Value == null)
{
storeSet.MetadataProperties.TryGetValue("http://schemas.microsoft.com/ado/2007/12/edm/EntityStoreSchemaGenerator:Schema", true, out schemaProperty);
}
if (schemaProperty != null)
{
schema = schemaProperty.Value as string;
}
entityMap.SchemaName = schema;
entityMap.TableName = table;
}
private static void SetKeys(EntityMap entityMap)
{
var modelType = entityMap.ModelType;
foreach (var edmMember in modelType.KeyMembers)
{
var property = entityMap.PropertyMaps.FirstOrDefault(p => p.PropertyName == edmMember.Name);
if (property == null)
{
continue;
}
var map = new PropertyMap
{
PropertyName = property.PropertyName,
ColumnName = property.ColumnName
};
entityMap.KeyMaps.Add(map);
}
}
private static void SetProperties(EntityMap entityMap, MappingFragment mappingFragment)
{
foreach (var propertyMapping in mappingFragment.PropertyMappings)
{
var map = new PropertyMap();
map.PropertyName = propertyMapping.Property.Name;
entityMap.PropertyMaps.Add(map);
var scalarPropertyMapping = propertyMapping as ScalarPropertyMapping;
if (scalarPropertyMapping != null)
{
map.ColumnName = scalarPropertyMapping.Column.Name;
continue;
}
// TODO support complex mapping
var complexPropertyMapping = propertyMapping as ComplexPropertyMapping;
}
}
private static void SetProperties(EntityMap entityMap, dynamic mappingFragmentProxy)
{
var propertyMaps = mappingFragmentProxy.Properties;
foreach (var propertyMap in propertyMaps)
{
// StorageScalarPropertyMapping
dynamic propertyMapProxy = new DynamicProxy(propertyMap);
EdmProperty modelProperty = propertyMapProxy.EdmProperty;
EdmProperty storeProperty = propertyMapProxy.ColumnProperty;
var map = new PropertyMap
{
ColumnName = storeProperty.Name,
PropertyName = modelProperty.Name
};
entityMap.PropertyMaps.Add(map);
}
}
/// <summary>
/// Gets the <see cref="EntityMap" /> for the specified <paramref name="type" />.
/// </summary>
/// <param name="type">The type of the entity.</param>
/// <param name="objectContext">The object context to load metadata from.</param>
/// <returns>
/// An <see cref="EntityMap" /> with the mapping data.
/// </returns>
public EntityMap GetEntityMap(Type type, ObjectContext objectContext)
{
var entityMap = new EntityMap(type);
var metadata = objectContext.MetadataWorkspace;
// Get the part of the model that contains info about the actual CLR types
var objectItemCollection = ((ObjectItemCollection)metadata.GetItemCollection(DataSpace.OSpace));
var entityType = metadata.GetItems <EntityType>(DataSpace.OSpace).Single(e => objectItemCollection.GetClrType(e) == type);
var entitySet = metadata.GetItems <EntityContainer>(DataSpace.CSpace)
.SelectMany(a => a.EntitySets).FirstOrDefault(s => s.ElementType.Name == entityType.Name);
var entitySetMappings = metadata.GetItems <EntityContainerMapping>(DataSpace.CSSpace).Single().EntitySetMappings.ToList();
var mapping = GetMapping(entitySetMappings, metadata.GetItems(DataSpace.CSpace)
.Where(x => x.BuiltInTypeKind == BuiltInTypeKind.EntityType)
.Cast <EntityType>()
.Single(x => x.Name == entityType.Name));
// Find the storage entity set (table) that the entity is mapped
var mappingFragment =
(mapping.EntityTypeMappings.FirstOrDefault(a => a.IsHierarchyMapping) ??
mapping.EntityTypeMappings.First()).Fragments.First();
entityMap.ModelType = entityType;
entityMap.ModelSet = entitySet;
entityMap.StoreSet = mappingFragment.StoreEntitySet;
entityMap.StoreType = mappingFragment.StoreEntitySet.ElementType;
// set table
SetTableName(entityMap);
// set properties
SetProperties(entityMap, mapping, type);
// set keys
SetKeys(entityMap);
return(entityMap);
}
private static void SetProperties(EntityMap entityMap, MappingFragment mappingFragment)
{
foreach (var propertyMapping in mappingFragment.PropertyMappings)
{
var map = new PropertyMap();
var propertyName = propertyMapping.Property.Name;
map.PropertyName = propertyName;
entityMap.PropertyMaps.Add(map);
var scalarPropertyMapping = propertyMapping as ScalarPropertyMapping;
if (scalarPropertyMapping != null)
{
map.ColumnName = scalarPropertyMapping.Column.Name;
continue;
}
// TODO support complex mapping
var complexPropertyMapping = propertyMapping as ComplexPropertyMapping;
if (complexPropertyMapping == null)
{
continue;
}
map.IsComplex = true;
foreach (var mapping in complexPropertyMapping.TypeMappings)
{
foreach (var propertyMapping1 in mapping.PropertyMappings)
{
var scalarMapping = propertyMapping1 as ScalarPropertyMapping;
if (scalarMapping == null)
{
continue;
}
var complexType = entityMap.EntityType.GetProperty(propertyName).PropertyType;
var propertyType = complexType.GetProperty(propertyMapping1.Property.Name).PropertyType;
map.AddProperty(new PropertyMap(scalarMapping.Property.Name, scalarMapping.Column.Name, propertyType));
}
}
}
}
