update(SqlServerStorage): Add check if tables already exist

MiniProfiler · Aug 31, 2021 · 8758fd7 · 8758fd7
1 parent c309d26
commit 8758fd7
Showing 1 changed file with 58 additions and 49 deletions.
diff --git a/src/MiniProfiler.Providers.SqlServer/SqlServerStorage.cs b/src/MiniProfiler.Providers.SqlServer/SqlServerStorage.cs
@@ -399,55 +399,64 @@ protected override IEnumerable<string> GetTableCreationScripts()
 -- creating schema name if not exists
 IF NOT EXISTS ( SELECT * FROM sys.schemas WHERE name = N'{SchemaName}') EXEC('CREATE SCHEMA [{SchemaName}]');
 
-CREATE TABLE {SchemaName}.{MiniProfilersTable}
-(
-    RowId                                integer not null identity constraint PK_{SchemaName}_{MiniProfilersTable} primary key clustered, -- Need a clustered primary key for SQL Azure
-    Id                                   uniqueidentifier not null, -- don't cluster on a guid
-    RootTimingId                         uniqueidentifier null,
-    Name                                 nvarchar(200) null,
-    Started                              datetime not null,
-    DurationMilliseconds                 decimal(15,1) not null,
-    [User]                               nvarchar(100) null,
-    HasUserViewed                        bit not null,
-    MachineName                          nvarchar(100) null,
-    CustomLinksJson                      nvarchar(max),
-    ClientTimingsRedirectCount           int null
-);
--- displaying results selects everything based on the main MiniProfilers.Id column
-CREATE UNIQUE NONCLUSTERED INDEX IX_{SchemaName}_{MiniProfilersTable}_Id ON {SchemaName}.{MiniProfilersTable} (Id);
-                
--- speeds up a query that is called on every .Stop()
-CREATE NONCLUSTERED INDEX IX_{SchemaName}_{MiniProfilersTable}_User_HasUserViewed_Includes ON {SchemaName}.{MiniProfilersTable} ([User], HasUserViewed) INCLUDE (Id, [Started]); 
-
-CREATE TABLE {SchemaName}.{MiniProfilerTimingsTable}
-(
-    RowId                               integer not null identity constraint PK_{SchemaName}_{MiniProfilerTimingsTable} primary key clustered,
-    Id                                  uniqueidentifier not null,
-    MiniProfilerId                      uniqueidentifier not null,
-    ParentTimingId                      uniqueidentifier null,
-    Name                                nvarchar(200) not null,
-    DurationMilliseconds                decimal(15,3) not null,
-    StartMilliseconds                   decimal(15,3) not null,
-    IsRoot                              bit not null,
-    Depth                               smallint not null,
-    CustomTimingsJson                   nvarchar(max) null
-);
-
-CREATE UNIQUE NONCLUSTERED INDEX IX_{SchemaName}_{MiniProfilerTimingsTable}_Id ON {SchemaName}.{MiniProfilerTimingsTable} (Id);
-CREATE NONCLUSTERED INDEX IX_{SchemaName}_{MiniProfilerTimingsTable}_MiniProfilerId ON {SchemaName}.{MiniProfilerTimingsTable} (MiniProfilerId);
-
-CREATE TABLE {SchemaName}.{MiniProfilerClientTimingsTable}
-(
-    RowId                               integer not null identity constraint PK_{SchemaName}_{MiniProfilerClientTimingsTable} primary key clustered,
-    Id                                  uniqueidentifier not null,
-    MiniProfilerId                      uniqueidentifier not null,
-    Name                                nvarchar(200) not null,
-    Start                               decimal(9, 3) not null,
-    Duration                            decimal(9, 3) not null
-);
-
-CREATE UNIQUE NONCLUSTERED INDEX IX_{SchemaName}_{MiniProfilerClientTimingsTable}_Id on {SchemaName}.{MiniProfilerClientTimingsTable} (Id);
-CREATE NONCLUSTERED INDEX IX_{SchemaName}_{MiniProfilerClientTimingsTable}_MiniProfilerId on {SchemaName}.{MiniProfilerClientTimingsTable} (MiniProfilerId);             
+IF OBJECT_ID(N'{SchemaName}.{MiniProfilersTable}', N'U') IS NULL
+BEGIN
+    CREATE TABLE {SchemaName}.{MiniProfilersTable}
+    (
+        RowId                                integer not null identity constraint PK_{SchemaName}_{MiniProfilersTable} primary key clustered, -- Need a clustered primary key for SQL Azure
+        Id                                   uniqueidentifier not null, -- don't cluster on a guid
+        RootTimingId                         uniqueidentifier null,
+        Name                                 nvarchar(200) null,
+        Started                              datetime not null,
+        DurationMilliseconds                 decimal(15,1) not null,
+        [User]                               nvarchar(100) null,
+        HasUserViewed                        bit not null,
+        MachineName                          nvarchar(100) null,
+        CustomLinksJson                      nvarchar(max),
+        ClientTimingsRedirectCount           int null
+    );
+    -- displaying results selects everything based on the main MiniProfilers.Id column
+    CREATE UNIQUE NONCLUSTERED INDEX IX_{SchemaName}_{MiniProfilersTable}_Id ON {SchemaName}.{MiniProfilersTable} (Id);
+                    
+    -- speeds up a query that is called on every .Stop()
+    CREATE NONCLUSTERED INDEX IX_{SchemaName}_{MiniProfilersTable}_User_HasUserViewed_Includes ON {SchemaName}.{MiniProfilersTable} ([User], HasUserViewed) INCLUDE (Id, [Started]); 
+END;
+
+IF OBJECT_ID(N'{SchemaName}.{MiniProfilerTimingsTable}', N'U') IS NULL
+BEGIN
+    CREATE TABLE {SchemaName}.{MiniProfilerTimingsTable}
+    (
+        RowId                               integer not null identity constraint PK_{SchemaName}_{MiniProfilerTimingsTable} primary key clustered,
+        Id                                  uniqueidentifier not null,
+        MiniProfilerId                      uniqueidentifier not null,
+        ParentTimingId                      uniqueidentifier null,
+        Name                                nvarchar(200) not null,
+        DurationMilliseconds                decimal(15,3) not null,
+        StartMilliseconds                   decimal(15,3) not null,
+        IsRoot                              bit not null,
+        Depth                               smallint not null,
+        CustomTimingsJson                   nvarchar(max) null
+    );
+
+    CREATE UNIQUE NONCLUSTERED INDEX IX_{SchemaName}_{MiniProfilerTimingsTable}_Id ON {SchemaName}.{MiniProfilerTimingsTable} (Id);
+    CREATE NONCLUSTERED INDEX IX_{SchemaName}_{MiniProfilerTimingsTable}_MiniProfilerId ON {SchemaName}.{MiniProfilerTimingsTable} (MiniProfilerId);
+END;
+
+IF OBJECT_ID(N'{SchemaName}.{MiniProfilerClientTimingsTable}', N'U') IS NULL
+BEGIN
+    CREATE TABLE {SchemaName}.{MiniProfilerClientTimingsTable}
+    (
+        RowId                               integer not null identity constraint PK_{SchemaName}_{MiniProfilerClientTimingsTable} primary key clustered,
+        Id                                  uniqueidentifier not null,
+        MiniProfilerId                      uniqueidentifier not null,
+        Name                                nvarchar(200) not null,
+        Start                               decimal(9, 3) not null,
+        Duration                            decimal(9, 3) not null
+    );
+
+    CREATE UNIQUE NONCLUSTERED INDEX IX_{SchemaName}_{MiniProfilerClientTimingsTable}_Id on {SchemaName}.{MiniProfilerClientTimingsTable} (Id);
+    CREATE NONCLUSTERED INDEX IX_{SchemaName}_{MiniProfilerClientTimingsTable}_MiniProfilerId on {SchemaName}.{MiniProfilerClientTimingsTable} (MiniProfilerId);             
+END;             
 ";
         }
     }