Enable streaming from Azure Sql DB

Author: spaghettidba

SharedAssemblyInfo.cs

@@ -29,5 +29,5 @@
 // You can specify all the values or you can default the Build and Revision Numbers
 // by using the '*' as shown below:
 // [assembly: AssemblyVersion("1.0.*")]
-[assembly: AssemblyVersion("1.7.1")]
-[assembly: AssemblyFileVersion("1.7.1")]
+[assembly: AssemblyVersion("1.7.2")]
+[assembly: AssemblyFileVersion("1.7.2")]

WorkloadTools/Consumer/Analysis/WorkloadAnalyzer.cs

@@ -13,6 +13,7 @@
 using WorkloadTools.Util;
 using System.Collections.Concurrent;
 using FastMember;
+using Microsoft.SqlServer.Management.SqlParser.SqlCodeDom;
 
 namespace WorkloadTools.Consumer.Analysis
 {
@@ -1126,8 +1127,14 @@ protected void CreateTargetDatabase()
 
 			try
 			{
-				using (var conn = new SqlConnection())
+                var databaseName = ConnectionInfo.DatabaseName;
+                using (var conn = new SqlConnection())
 				{
+                    // create a new connection to the target server 
+                    // for the analysis database, on the master db
+                    // then create the target database if not available
+                    var ci = new SqlConnectionInfo(ConnectionInfo);
+                    ci.DatabaseName = "master";
 					conn.ConnectionString = ConnectionInfo.ConnectionString();
 					conn.Open();
 
@@ -1147,11 +1154,9 @@ IF DB_ID(@name) IS NULL
 					}
 				}
 			}
-			finally
-			{
-				// restore original database name
-				ConnectionInfo.DatabaseName = databaseName;
-			}
+            catch(Exception e) {
+                logger.Warn("Unable to create the target database for the analysisy", e.Message);
+            }
 
 		}
 

WorkloadTools/Listener/ExtendedEvents/ExtendedEventsWorkloadListener.cs

@@ -61,10 +61,6 @@ public override void Initialize()
 
                 if (serverType == ServerType.AzureSqlDatabase)
                 {
-                    if (FileTargetPath == null)
-                    {
-                        throw new ArgumentException("Azure SqlDatabase does not support Extended Events streaming. Please specify a path for the FileTarget");
-                    }
                     if (ConnectionInfo.DatabaseName == null)
                     {
                         throw new ArgumentException("Azure SqlDatabase does not support starting Extended Events sessions on the master database. Please specify a database name.");
@@ -280,7 +276,7 @@ private void ReadEvents()
             try
             {
 
-                if (serverType != ServerType.AzureSqlDatabase && FileTargetPath == null)
+                if (FileTargetPath == null)
                 {
                     reader = new StreamXEventDataReader(ConnectionInfo.ConnectionString(), SessionName, Events);
                 }

WorkloadTools/SqlConnectionInfo.cs

@@ -19,6 +19,24 @@ public class SqlConnectionInfo
         public int MaxPoolSize { get; set; } = 500;
         public Dictionary<string, string> DatabaseMap { get; set; } = new Dictionary<string, string>();
 
+
+        public SqlConnectionInfo() { }
+
+        public SqlConnectionInfo(SqlConnectionInfo info)
+        {
+            this.ServerName = info.ServerName;
+            this.DatabaseName = info.DatabaseName;
+            this.SchemaName = info.SchemaName;
+            this.UseIntegratedSecurity = info.UseIntegratedSecurity;
+            this.UserName = info.UserName;
+            this.Password = info.Password;
+            this.Encrypt = info.Encrypt;
+            this.TrustServerCertificate = info.TrustServerCertificate;
+            this.ApplicationName = info.ApplicationName;
+            this.MaxPoolSize = info.MaxPoolSize;
+            this.DatabaseMap = info.DatabaseMap;
+        }
+
         public string ConnectionString()
         {
             return ConnectionString(ApplicationName);