Wait for SQL

Author: ejsmith

tests/Foundatio.Parsers.SqlQueries.Tests/SqlQueryParserTests.cs

@@ -2,6 +2,7 @@
 using System.ComponentModel.DataAnnotations;
 using System.Linq;
 using System.Linq.Dynamic.Core;
+using System.Threading;
 using System.Threading.Tasks;
 using Foundatio.Parsers.LuceneQueries.Nodes;
 using Foundatio.Parsers.LuceneQueries.Visitors;
@@ -448,6 +449,9 @@ public static string TryGetNationalNumber(string phoneNumber, string regionCode
 
     public IServiceProvider GetServiceProvider()
     {
+        string sqlConnectionString = "Server=localhost;User Id=sa;Password=P@ssword1;Timeout=5;Initial Catalog=foundatio;Encrypt=False";
+        SqlWaiter.Wait(sqlConnectionString);
+
         var services = new ServiceCollection();
         services.AddDbContext<SampleContext>((_, x) =>
         {
@@ -536,6 +540,37 @@ ON ftCatalog
         return db;
     }
 
+    private static bool _checked;
+    private static readonly object _lock = new();
+
+    private static void WaitForSql(
+        string connectionString,
+        int maxRetries = 90,
+        int delayMs = 1000)
+    {
+        for (int i = 0; i < maxRetries; i++)
+        {
+            try
+            {
+                using var conn = new SqlConnection(connectionString);
+                conn.Open();
+
+                using var cmd = conn.CreateCommand();
+                cmd.CommandText = "SELECT 1";
+                cmd.ExecuteScalar();
+
+                return;
+            }
+            catch
+            {
+                if (i == maxRetries - 1)
+                    throw;
+
+                Thread.Sleep(delayMs);
+            }
+        }
+    }
+
     private async Task WaitForFullTextIndexAsync(DbContext db, string catalogName, int timeoutSeconds = 30)
     {
         var end = DateTime.UtcNow.AddSeconds(timeoutSeconds);

tests/Foundatio.Parsers.SqlQueries.Tests/SqlWaiter.cs

@@ -0,0 +1,63 @@
+using System;
+using System.Threading;
+using Microsoft.Data.SqlClient;
+
+namespace Foundatio.Parsers.SqlQueries.Tests;
+
+public static class SqlWaiter
+{
+    private static bool _checked;
+    private static readonly object _lock = new();
+
+    public static void Wait(
+        string connectionString,
+        TimeSpan? timeout = null,
+        int delayMs = 1000)
+    {
+        if (_checked)
+            return;
+
+        lock (_lock)
+        {
+            if (_checked)
+                return;
+
+            timeout ??= TimeSpan.FromSeconds(30);
+            var end = DateTime.UtcNow + timeout.Value;
+
+            string masterCs = BuildMasterConnectionString(connectionString);
+
+            while (DateTime.UtcNow < end)
+            {
+                try
+                {
+                    using var conn = new SqlConnection(masterCs);
+                    conn.Open();
+
+                    using var cmd = conn.CreateCommand();
+                    cmd.CommandText = "SELECT 1";
+                    cmd.ExecuteScalar();
+
+                    _checked = true;
+                    return;
+                }
+                catch
+                {
+                    Thread.Sleep(delayMs);
+                }
+            }
+
+            throw new Exception("Failed to connect to SQL Server within timeout.");
+        }
+    }
+
+    private static string BuildMasterConnectionString(string cs)
+    {
+        var builder = new SqlConnectionStringBuilder(cs)
+        {
+            InitialCatalog = "master"
+        };
+
+        return builder.ToString();
+    }
+}