private void GetPrimaryKeyV3(LinqToEdmx.Model.StorageV3.EntityTypeStore table, DatabaseTable dbTable)
{
if (table.Key.PropertyRefs.Count() == 0)
{
return;
}
var pk = table.Key;
var primaryKey = new DatabasePrimaryKey
{
// We do not have information about the primary key name in the model.
// So we're making it up
Name = string.Concat("PK_", dbTable.Name),
Table = dbTable
};
foreach (var pkCol in table.Key.PropertyRefs)
{
var dbCol = dbTable.Columns
.Single(c => c.Name == pkCol.Name);
primaryKey.Columns.Add(dbCol);
}
dbTable.PrimaryKey = primaryKey;
}
private void GetColumnsV3(LinqToEdmx.Model.StorageV3.EntityTypeStore item, DatabaseTable dbTable /*, IReadOnlyDictionary<string, (string storeType, string typeName)> typeAliases, List<TSqlDefaultConstraint> defaultConstraints,*/, LinqToEdmx.EdmxV3 model)
{
var tableColumns = item.Properties;
foreach (var col in tableColumns)
{
string storeType = GetStoreTypeV3(col);
string defaultValue = null;
var dbColumn = new DatabaseColumn
{
Table = dbTable,
Name = col.Name,
IsNullable = col.Nullable,
StoreType = storeType,
DefaultValueSql = defaultValue,
};
if (storeType == "rowversion")
{
dbColumn["ConcurrencyToken"] = true;
}
dbTable.Columns.Add(dbColumn);
}
}
private void GetForeignKeysV3(EdmxV3 model, LinqToEdmx.Model.StorageV3.EntityTypeStore storeTable, DatabaseModel dbModel)
{
var table = dbModel.Tables
.Single(t => t.Name == storeTable.Name &&
t.Schema == GetSchemaNameV3(model, storeTable.Name));
// The foreign key informations are stored in the Association Object
var associations = model.GetItems <LinqToEdmx.Model.StorageV3.Association>().Where(a => a.ReferentialConstraint.Dependent.Role == storeTable.Name);
if (!associations.Any())
{
// Give a chance to a reflexive constraint
// TODO make sure that this is the right way to deal with this case.
var endType = string.Concat(@"Self.", storeTable.Name);
associations = model.GetItems <LinqToEdmx.Model.StorageV3.Association>().Where(a => a.Ends.All(e => e.Type == endType));
}
// No association ? No FK then.
if (!associations.Any())
{
return;
}
foreach (var association in associations)
{
// The entity name could be derived if present multiple times (a table has a foreign key to himself => See ProductCategory which has a parent ProductCategory)
// Te association set allows us to disambiguate the table name.
var associationSet = model.GetItems <LinqToEdmx.Model.StorageV3.EntityContainer>().First().AssociationSets.Where(@as => @as.Name == association.Name).SingleOrDefault();
var entityName = association.ReferentialConstraint.Principal.Role;
// Look for the table to which the columns are constrained
// Remember that we could be constrained to ourself :)
var principalTable = dbModel.Tables.SingleOrDefault(t => t.Name == entityName && t.Schema == GetSchemaNameV3(model, entityName));
if (principalTable == null)
{
entityName = associationSet.Ends.Single(e => e.Role == entityName).EntitySet;
principalTable = dbModel.Tables.SingleOrDefault(t => t.Name == entityName && t.Schema == GetSchemaNameV3(model, entityName));
}
if (principalTable == null)
{
throw new InvalidOperationException($"Unable to get foreign keys for entity with name [{entityName}]. This usually indicates a bug");
}
var foreignKey = new DatabaseForeignKey
{
// The name of the foreign key
Name = association.Name,
// The table that contains the foreign key constraint (Dependent in the Edmx)
Table = table,
// The table to which the columns are constrained (Principal in the Edmx)
PrincipalTable = principalTable,
};
var end = associationSet.Ends[0];
// Finish to populate the foreign key definition with principal and dependent columns
var rc = association.ReferentialConstraint;
foreach (var fkCol in association.ReferentialConstraint.Principal.PropertyRefs)
{
var dbCol = principalTable.Columns
.Single(c => c.Name == fkCol.Name);
if (dbCol != null)
{
foreignKey.PrincipalColumns.Add(dbCol);
}
}
foreach (var fkCol in association.ReferentialConstraint.Dependent.PropertyRefs)
{
// Best case scenario, the columns name are the same in both entities
var dbCol = table.Columns
.Single/*OrDefault*/ (c => c.Name == fkCol.Name);
if (dbCol != null)
{
foreignKey.Columns.Add(dbCol);
}
}
if (foreignKey.PrincipalColumns.Count > 0)
{
table.ForeignKeys.Add(foreignKey);
}
}
}
