From 7a84060b41020db6bee9e05f788e23494f1bcc52 Mon Sep 17 00:00:00 2001
From: Matt Gibson <mgibson@bitwarden.com>
Date: Fri, 25 Jun 2021 14:24:10 -0400
Subject: [PATCH] USe MARS-enabled connections to fix connection issue

https://github.com/dotnet/SqlClient/issues/54
---
 .../SqlServer/OrganizationUserRepository.cs   | 21 +++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/src/Core/Repositories/SqlServer/OrganizationUserRepository.cs b/src/Core/Repositories/SqlServer/OrganizationUserRepository.cs
index 49979ae6be03..c204a5c5f189 100644
--- a/src/Core/Repositories/SqlServer/OrganizationUserRepository.cs
+++ b/src/Core/Repositories/SqlServer/OrganizationUserRepository.cs
@@ -16,9 +16,22 @@ namespace Bit.Core.Repositories.SqlServer
 {
     public class OrganizationUserRepository : Repository<OrganizationUser, Guid>, IOrganizationUserRepository
     {
+        /// <summary>
+        /// For use with methods with TDS stream issues.
+        /// This has been observed in Linux-hosted SqlServers with large table-valued-parameters
+        /// https://github.com/dotnet/SqlClient/issues/54
+        /// </summary>
+        private string _marsConnectionString;
+
         public OrganizationUserRepository(GlobalSettings globalSettings)
             : this(globalSettings.SqlServer.ConnectionString, globalSettings.SqlServer.ReadOnlyConnectionString)
-        { }
+        {
+            var builder = new SqlConnectionStringBuilder(ConnectionString)
+            {
+                MultipleActiveResultSets = true,
+            };
+            _marsConnectionString = builder.ToString();
+        }
 
         public OrganizationUserRepository(string connectionString, string readOnlyConnectionString)
             : base(connectionString, readOnlyConnectionString)
@@ -80,7 +93,7 @@ public async Task<int> GetCountByOrganizationAsync(Guid organizationId, string e
             bool onlyRegisteredUsers)
         {
             var emailsTvp = emails.ToArrayTVP("Email");
-            using (var connection = new SqlConnection(ConnectionString))
+            using (var connection = new SqlConnection(_marsConnectionString))
             {
                 var result = await connection.QueryAsync<string>(
                     "[dbo].[OrganizationUser_SelectKnownEmails]",
@@ -344,7 +357,7 @@ public async Task CreateManyAsync(IEnumerable<OrganizationUser> organizationUser
             }
 
             var orgUsersTVP = organizationUsers.ToTvp();
-            using (var connection = new SqlConnection(ConnectionString))
+            using (var connection = new SqlConnection(_marsConnectionString))
             {
                 var results = await connection.ExecuteAsync(
                     $"[{Schema}].[{Table}_CreateMany]",
@@ -361,7 +374,7 @@ public async Task ReplaceManyAsync(IEnumerable<OrganizationUser> organizationUse
             }
 
             var orgUsersTVP = organizationUsers.ToTvp();
-            using (var connection = new SqlConnection(ConnectionString))
+            using (var connection = new SqlConnection(_marsConnectionString))
             {
                 var results = await connection.ExecuteAsync(
                     $"[{Schema}].[{Table}_UpdateMany]",