added StatementTimeout

abbgrade · Sep 5, 2023 · 10b4c50 · 10b4c50
1 parent db71b18
commit 10b4c50
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -7,6 +7,12 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [1.2.1] - 2023-09-05
+
+### Added
+
+- Added StatementTimeout to `Connect-Instance` command.
+
 ## [1.2.0] - 2023-08-24
 
 ### Changed

diff --git a/src/PsSmo/ConnectInstanceCommand.cs b/src/PsSmo/ConnectInstanceCommand.cs
@@ -97,6 +97,8 @@ public class ConnectInstanceCommand : PSCmdlet
         [ValidateNotNullOrEmpty()]
         public SecureString Password { get; set; }
 
+        [Parameter()]
+        public int StatementTimeout { get; set; } = 600;
         #endregion
 
         protected override void ProcessRecord()
@@ -111,6 +113,9 @@ protected override void ProcessRecord()
                         serverConnection: new ServerConnection(
                             sqlConnection: Connection
                         )
+                        {
+                            StatementTimeout = StatementTimeout
+                        }
                     );
                     break;
 
@@ -124,6 +129,9 @@ protected override void ProcessRecord()
                             serverConnection: new ServerConnection(
                                 sqlConnection: Connection
                             )
+                            {
+                                StatementTimeout = StatementTimeout
+                            }
                         );
                         break;
                     }
@@ -138,8 +146,7 @@ protected override void ProcessRecord()
                         if (DataSource.EndsWith("database.windows.net"))
                         {
                             Connection = new SqlConnection(connectionString: builder.ConnectionString);
-                            if (AccessToken == null)
-                                AccessToken = new AzureServiceTokenProvider().GetAccessTokenAsync("https://database.windows.net").Result;
+                            AccessToken ??= new AzureServiceTokenProvider().GetAccessTokenAsync("https://database.windows.net").Result;
                             Connection.AccessToken = AccessToken;
                         }
                         else
@@ -151,6 +158,9 @@ protected override void ProcessRecord()
                             serverConnection: new ServerConnection(
                                 sqlConnection: Connection
                             )
+                            {
+                                StatementTimeout = StatementTimeout
+                            }
                         );
                         break;
                     }
@@ -171,6 +181,9 @@ protected override void ProcessRecord()
                             serverConnection: new ServerConnection(
                                 sqlConnection: Connection
                             )
+                            {
+                                StatementTimeout = StatementTimeout
+                            }
                         );
                         break;
                     }

diff --git a/src/PsSmo/PsSmo.psd1 b/src/PsSmo/PsSmo.psd1
@@ -12,7 +12,7 @@
 RootModule = 'PsSmo.dll'
 
 # Version number of this module.
-ModuleVersion = '1.2.0'
+ModuleVersion = '1.2.1'
 
 # Supported PSEditions
 # CompatiblePSEditions = @()