fix(mssql): support multiple result sets in raw SQL ExecuteReaderAsync

Author: vkuttyp

Directory.Build.props

@@ -7,6 +7,6 @@
     <Authors>vkuttyp</Authors>
     <PackageLicenseExpression>MIT</PackageLicenseExpression>
     <RepositoryUrl>https://github.com/vkuttyp/CosmoSQLClient-Dotnet</RepositoryUrl>
-    <Version>1.5.6</Version>
+    <Version>1.5.7</Version>
   </PropertyGroup>
 </Project>

src/CosmoSQLClient.MsSql/MsSqlCommand.cs

@@ -127,10 +127,10 @@ public override async Task<int> ExecuteNonQueryAsync(CancellationToken ct = defa
         }
         else
         {
-            // Internal RunQueryAsync returns MsSqlResultSet (Cols + Rows)
-            var rs = await ((MsSqlConnection)_connection).RunQueryAsync(_commandText, Parameters.ToCoreParameters(), cts.Token)
+            // Execute multiple result sets if present in the raw SQL batch
+            var rs = await ((MsSqlConnection)_connection).RunQueryAllAsync(_commandText, Parameters.ToCoreParameters(), cts.Token)
                                    .ConfigureAwait(false);
-            return new MsSqlDataReader([rs]);
+            return new MsSqlDataReader(rs);
         }
     }
 

src/CosmoSQLClient.MsSql/MsSqlConnection.cs

@@ -677,7 +677,20 @@ await WritePacketsAsync(TdsPacketType.PreLogin, TdsPreLogin.BuildRequest(wantEnc
         _isOpen = true;
     }
 
-    // ── Internal query helpers ────────────────────────────────────────────────
+    internal async Task<IReadOnlyList<SqlResultSet>> RunQueryAllAsync(
+        string sql, IReadOnlyList<SqlParameter>? parameters, CancellationToken ct)
+    {
+        await _lock.WaitAsync(ct).ConfigureAwait(false);
+        try
+        {
+            await SendSqlAsync(sql, parameters).ConfigureAwait(false);
+            var payload = await ReceiveAsync(ct).ConfigureAwait(false);
+            var tokens  = TdsDecoder.Decode(payload);
+            RaiseMessages(tokens);
+            return GroupResultSetsWithColumns(tokens);
+        }
+        finally { _lock.Release(); }
+    }
 
     internal async Task<SqlResultSet> RunQueryAsync(
         string sql, IReadOnlyList<SqlParameter>? parameters, CancellationToken ct)
@@ -741,7 +754,7 @@ private async Task<SqlProcResult> RunProcAsync(
     }
 
     /// <summary>Group a flat token list into per-result-set row lists using TdsDone as boundaries.</summary>
-    private static IReadOnlyList<SqlResultSet> GroupResultSetsWithColumns(List<object> tokens)
+    internal static IReadOnlyList<SqlResultSet> GroupResultSetsWithColumns(List<object> tokens)
     {
         var sets    = new List<SqlResultSet>();
         var current = new List<SqlRow>();