public static SqlTable AddForeignKey(this SqlTable table, string referredTableName, bool isNullable = false, string prefix = null, string fkName = null) { var referredTableNameWithSchema = new SchemaAndTableName(referredTableName); var fk = new ForeignKeyRegistrationToTableWithUniqueKey(table, referredTableNameWithSchema, isNullable, prefix, fkName); var placeHolderColumnName = $"*{referredTableNameWithSchema}.{table.Columns.Count.ToString("D", CultureInfo.InvariantCulture)}"; table.Columns.Add(new SqlColumnFKRegistration(placeHolderColumnName, fk)); table.Properties.Add(fk); Prepare(table); return(table); }
public static void UniqueKey(DatabaseDefinition definition, SqlTable sqlTable, ForeignKeyRegistrationToTableWithUniqueKey fkRegistration, IForeignKeyNamingStrategy fkNaming) { var referredTable = definition.GetTable(fkRegistration.ReferredTableName); var referredUniqueIndex = GetReferredUniqueIndex(referredTable); var fk = ReplaceFKRegistrationWithNewFK(sqlTable, fkRegistration, referredTable); var placeHolderColumn = sqlTable.Columns.OfType <SqlColumnFKRegistration>().FirstOrDefault(c => c.FKRegistration == fkRegistration); var order = sqlTable.Columns.GetOrder(placeHolderColumn.Name); sqlTable.Columns.Remove(placeHolderColumn.Name); foreach (var pkColumn in referredUniqueIndex.SqlColumns.Select(x => x.SqlColumn)) { var col = new SqlColumn(); pkColumn.CopyTo(col); col.Table = sqlTable; col.Types.SetAllNullable(fkRegistration.IsNullable); col.Name = fkNaming.GetFkToPkColumnName(pkColumn, fkRegistration.NamePrefix); sqlTable.Columns.Add(col.Name, col, order++); fk.ForeignKeyColumns.Add(new ForeignKeyColumnMap(col, pkColumn)); } }