Skip to content

Commit ebb1a97

Browse files
Merge pull request #154 from dotnetprojects/add-primary-key
Quote columns and table name
2 parents a7a01a1 + 14441c4 commit ebb1a97

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/Migrator/Providers/TransformationProvider.cs

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@
2323
using System.Data.Common;
2424
using System.IO;
2525
using System.Linq;
26-
using System.Reflection;
2726
using System.Text;
2827
using ForeignKeyConstraint = DotNetProjects.Migrator.Framework.ForeignKeyConstraint;
2928
using ForeignKeyConstraintType = DotNetProjects.Migrator.Framework.ForeignKeyConstraintType;
@@ -725,6 +724,9 @@ public virtual void AddColumn(string table, string column, MigratorDbType type,
725724
/// <param name="columns">Primary column names</param>
726725
public virtual void AddPrimaryKey(string name, string table, params string[] columns)
727726
{
727+
QuoteColumnNamesIfRequired(columns);
728+
table = QuoteTableNameIfRequired(table);
729+
728730
ExecuteNonQuery(
729731
string.Format("ALTER TABLE {0} ADD CONSTRAINT {1} PRIMARY KEY ({2}) ", table, name,
730732
string.Join(",", QuoteColumnNamesIfRequired(columns))));
@@ -735,8 +737,7 @@ public virtual void AddPrimaryKeyNonClustered(string name, string table, params
735737
}
736738
public virtual void AddUniqueConstraint(string name, string table, params string[] columns)
737739
{
738-
QuoteColumnNames(columns);
739-
740+
QuoteColumnNamesIfRequired(columns);
740741
table = QuoteTableNameIfRequired(table);
741742

742743
ExecuteNonQuery(string.Format("ALTER TABLE {0} ADD CONSTRAINT {1} UNIQUE({2}) ", table, name, string.Join(", ", columns)));

0 commit comments

Comments
 (0)