Refactored Health Checks

src/Volvox.Helios.Web/HealthChecks/SqlServerHealthCheck.cs

@@ -0,0 +1,30 @@
+using System.Data.SqlClient;
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.Extensions.Diagnostics.HealthChecks;
+
+namespace Volvox.Helios.Web.HealthChecks
+{
+    public class SqlServerHealthCheck : IHealthCheck
+    {
+        SqlConnection _connection;
+
+        public SqlServerHealthCheck(SqlConnection connection)
+        {
+            _connection = connection;
+        }
+
+        public Task<HealthCheckResult> CheckHealthAsync(HealthCheckContext healthCheckContext, CancellationToken cancellationToken = default(CancellationToken))
+        {
+            try
+            {
+                _connection.Open();
+            }
+            catch (SqlException)
+            {
+                return Task.FromResult(HealthCheckResult.Unhealthy());
+            }
+            return Task.FromResult(HealthCheckResult.Healthy());
+        }
+    }
+}

src/Volvox.Helios.Web/Startup.cs

@@ -181,23 +181,9 @@ public void ConfigureServices(IServiceCollection services)
                 });
             });
 
-            //TODO: Refactor to make cleaner
+            // Health Checks
             services.AddHealthChecks()
-                .AddCheck("sql", () =>
-                {
-                    using (var connection = new SqlConnection(Configuration.GetConnectionString("VolvoxHeliosDatabase")))
-                    {
-                        try
-                        {
-                            connection.Open();
-                        }
-                        catch (SqlException)
-                        {
-                            return HealthCheckResult.Unhealthy();
-                        }
-                        return HealthCheckResult.Healthy();
-                    }
-                });
+                    .AddSqlServer(Configuration["ConnectionStrings:VolvoxHeliosDatabase"]);
         }
 
         // This method gets called by the runtime. Use this method to configure the HTTP request pipeline.

src/Volvox.Helios.Web/Volvox.Helios.Web.csproj

@@ -8,6 +8,7 @@
   </PropertyGroup>
   <ItemGroup>
     <PackageReference Include="AspNet.Security.OAuth.Discord" Version="2.0.0" />
+    <PackageReference Include="AspNetCore.HealthChecks.SqlServer" Version="2.2.1" />
     <PackageReference Include="AWS.Logger.AspNetCore" Version="1.4.1" />
     <PackageReference Include="AWS.Logger.Core" Version="1.3.1" />
     <PackageReference Include="BuildBundlerMinifier" Version="2.8.391" />