adding conditions for Azure DB

.gitignore

@@ -21,3 +21,4 @@ samples/features/in-memory/ticket-reservations/DemoWorkload/obj/Release/DemoWork
 samples/features/in-memory/ticket-reservations/DemoWorkload/bin/Release/DemoWorkload.vshost.exe.config
 samples/databases/wide-world-importers/sample-scripts/always-encrypted/PopulateAlwaysEncryptedData/obj/Debug/PopulateAlwaysEncryptedData.csproj.FileListAbsolute.txt
 samples/databases/wide-world-importers/sample-scripts/always-encrypted/PopulateAlwaysEncryptedData/bin/Debug/PopulateAlwaysEncryptedData.vshost.exe.config
+*.zip

samples/databases/wide-world-importers/wwi-database-scripts/2-wwi-construct-database-from-metadata-tables.sql

@@ -82,6 +82,8 @@ DECLARE @PrimaryKeyColumn nvarchar(max) = N'';
 
 SET @SQL = N'';
 
+-- when not using Azure DB, add create database statement
+IF SERVERPROPERTY('EngineEdition') != 5
     SET @SQL += N'USE master;' + @CrLf + @CrLf
           + N'IF EXISTS(SELECT 1 FROM sys.databases WHERE name = N''WideWorldImporters'')' + @CrLf
           + N'BEGIN' + @CrLf
@@ -112,12 +114,14 @@ SET @SQL += N'USE master;' + @CrLf + @CrLf
           + N'    SIZE = 100MB,' + @CrLf
           + N'    MAXSIZE = UNLIMITED,' + @CrLf
           + N'    FILEGROWTH = 64MB' + @CrLf
-          + N')' + @CrLf
+          + N');' + @CrLf + @GO
-		  + N'COLLATE Latin1_General_100_CI_AS;' + @CrLf + @GO
-          + N'ALTER DATABASE WideWorldImporters SET RECOVERY SIMPLE;' + @CrLf + @GO
-          + N'ALTER DATABASE WideWorldImporters SET AUTO_UPDATE_STATISTICS_ASYNC ON;' + @CrLf + @GO
           + N'ALTER AUTHORIZATION ON DATABASE::WideWorldImporters to sa;' + @CrLf + @GO
-          + N'ALTER DATABASE WideWorldImporters' + @CrLf
+          + N'USE WideWorldImporters;' + @CrLf + @GO;
+
+ SET @SQL += N'ALTER DATABASE CURRENT COLLATE Latin1_General_100_CI_AS;' + @CrLf + @GO
+  	  + N'ALTER DATABASE CURRENT SET RECOVERY SIMPLE;' + @CrLf + @GO
+        + N'ALTER DATABASE CURRENT SET AUTO_UPDATE_STATISTICS_ASYNC ON;' + @CrLf + @GO
+        + N'ALTER DATABASE CURRENT' + @CrLf
         + N'SET QUERY_STORE' + @CrLf
         + N'(' + @CrLf
         + @Indent + N'OPERATION_MODE = READ_WRITE,' + @CrLf
@@ -129,7 +133,6 @@ SET @SQL += N'USE master;' + @CrLf + @CrLf
         + @Indent + N'QUERY_CAPTURE_MODE = AUTO,' + @CrLf
         + @Indent + N'MAX_PLANS_PER_QUERY = 1000' + @CrLf
         + N');' + @CrLf + @GO
-          + N'USE WideWorldImporters;' + @CrLf + @GO;
 
 DECLARE SchemaList CURSOR FAST_FORWARD READ_ONLY
 FOR

samples/databases/wide-world-importers/wwi-database-scripts/4-wwi-configure-required-database-objects.sql

@@ -4626,19 +4626,38 @@ FOR VALUES (N''20140101'', N''20150101'', N''20160101'', N''20170101'');';
 
         IF NOT EXISTS (SELECT * FROM sys.partition_schemes WHERE name = N'PS_TransactionDateTime')
         BEGIN
+
+            -- for Azure DB, assign to primary filegroup
+            IF SERVERPROPERTY('EngineEdition') = 5
+                SET @SQL =  N'
+CREATE PARTITION SCHEME PS_TransactionDateTime
+AS PARTITION PF_TransactionDateTime
+ALL TO ([PRIMARY]);';
+            -- for other engine editions, assign to user data filegroup
+            IF SERVERPROPERTY('EngineEdition') != 5
                 SET @SQL =  N'
 CREATE PARTITION SCHEME PS_TransactionDateTime
 AS PARTITION PF_TransactionDateTime
 ALL TO ([USERDATA]);';
+
             EXECUTE (@SQL);
         END;
 
         IF NOT EXISTS (SELECT 1 FROM sys.partition_schemes WHERE name = N'PS_TransactionDate')
         BEGIN
+        -- for Azure DB, assign to primary filegroup
+        IF SERVERPROPERTY('EngineEdition') = 5
+            SET @SQL =  N'
+CREATE PARTITION SCHEME PS_TransactionDate
+AS PARTITION PF_TransactionDate
+ALL TO ([PRIMARY]);';
+        -- for other engine editions, assign to user data filegroup
+        IF SERVERPROPERTY('EngineEdition') != 5
             SET @SQL =  N'
 CREATE PARTITION SCHEME PS_TransactionDate
 AS PARTITION PF_TransactionDate
 ALL TO ([USERDATA]);';
+
             EXECUTE (@SQL);
         END;
 

samples/databases/wide-world-importers/wwi-dw-database-scripts/2-wwi-dw-construct-database-from-metadata-tables.sql

@@ -72,6 +72,8 @@ DECLARE @PrimaryKeyColumn nvarchar(max) = N'';
 
 SET @SQL = N'';
 
+-- when not using Azure DB, add create database statement
+IF SERVERPROPERTY('EngineEdition') != 5
     SET @SQL += N'USE master;' + @CrLf + @CrLf
           + N'IF EXISTS(SELECT 1 FROM sys.databases WHERE name = N''WideWorldImportersDW'')' + @CrLf
           + N'BEGIN' + @CrLf
@@ -102,12 +104,15 @@ SET @SQL += N'USE master;' + @CrLf + @CrLf
           + N'    SIZE = 100MB,' + @CrLf
           + N'    MAXSIZE = UNLIMITED,' + @CrLf
           + N'    FILEGROWTH = 64MB' + @CrLf
-          + N')' + @CrLf
+          + N');' + @CrLf + @GO
-		  + N'COLLATE Latin1_General_100_CI_AS;' + @CrLf + @GO
-		  + N'ALTER DATABASE WideWorldImportersDW SET RECOVERY SIMPLE;' + @CrLf + @GO
-          + N'ALTER DATABASE WideWorldImporters SET AUTO_UPDATE_STATISTICS_ASYNC ON;' + @CrLf + @GO
           + N'ALTER AUTHORIZATION ON DATABASE::WideWorldImportersDW to sa;' + @CrLf + @GO
-          + N'ALTER DATABASE WideWorldImportersDW' + @CrLf
+          + N'USE WideWorldImportersDW;' + @CrLf + @GO;
+
+
+SET @SQL += N'ALTER DATABASE CURRENT COLLATE Latin1_General_100_CI_AS;' + @CrLf + @GO
+	  + N'ALTER DATABASE CURRENT SET RECOVERY SIMPLE;' + @CrLf + @GO
+      + N'ALTER DATABASE CURRENT SET AUTO_UPDATE_STATISTICS_ASYNC ON;' + @CrLf + @GO
+      + N'ALTER DATABASE CURRENT' + @CrLf
       + N'SET QUERY_STORE' + @CrLf
       + N'(' + @CrLf
       + @Indent + N'OPERATION_MODE = READ_WRITE,' + @CrLf
@@ -119,7 +124,7 @@ SET @SQL += N'USE master;' + @CrLf + @CrLf
       + @Indent + N'QUERY_CAPTURE_MODE = AUTO,' + @CrLf
       + @Indent + N'MAX_PLANS_PER_QUERY = 1000' + @CrLf
       + N');' + @CrLf + @GO
-          + N'USE WideWorldImportersDW;' + @CrLf + @GO;
+
 
 DECLARE SchemaList CURSOR FAST_FORWARD READ_ONLY
 FOR

samples/databases/wide-world-importers/wwi-dw-database-scripts/4-wwi-dw-configure-required-database-objects.sql

@@ -1552,6 +1552,14 @@ FOR VALUES (N''20120101'',N''20130101'',N''20140101'', N''20150101'', N''2016010
 
 			IF NOT EXISTS (SELECT 1 FROM sys.partition_schemes WHERE name = N'PS_Date')
 			BEGIN
+            -- for Azure DB, assign to primary filegroup
+            IF SERVERPROPERTY('EngineEdition') = 5
+                SET @SQL =  N'
+CREATE PARTITION SCHEME PS_Date
+AS PARTITION PF_Date
+ALL TO ([PRIMARY]);';
+            -- for other engine editions, assign to user data filegroup
+            IF SERVERPROPERTY('EngineEdition') != 5
                 SET @SQL =  N'
 CREATE PARTITION SCHEME PS_Date
 AS PARTITION PF_Date