Adding more

Author: johnsimons

src/ServiceControl.Persistence.Sql.Core/DbContexts/ServiceControlDbContextBase.cs

@@ -17,6 +17,7 @@ protected ServiceControlDbContextBase(DbContextOptions options) : base(options)
     public DbSet<SubscriptionEntity> Subscriptions { get; set; }
     public DbSet<QueueAddressEntity> QueueAddresses { get; set; }
     public DbSet<KnownEndpointEntity> KnownEndpoints { get; set; }
+    public DbSet<CustomCheckEntity> CustomChecks { get; set; }
 
     protected override void OnModelCreating(ModelBuilder modelBuilder)
     {
@@ -29,6 +30,7 @@ protected override void OnModelCreating(ModelBuilder modelBuilder)
         modelBuilder.ApplyConfiguration(new SubscriptionConfiguration());
         modelBuilder.ApplyConfiguration(new QueueAddressConfiguration());
         modelBuilder.ApplyConfiguration(new KnownEndpointConfiguration());
+        modelBuilder.ApplyConfiguration(new CustomCheckConfiguration());
 
         OnModelCreatingProvider(modelBuilder);
     }

src/ServiceControl.Persistence.Sql.Core/Implementation/CustomChecksDataStore.cs

@@ -59,7 +59,7 @@ public async Task<CheckStateChange> UpdateCustomCheckStatus(CustomCheckDetail de
         return status;
     }
 
-    public async Task<QueryResult<IList<CustomCheck>>> GetStats(PagingInfo paging, string status = null)
+    public async Task<QueryResult<IList<CustomCheck>>> GetStats(PagingInfo paging, string? status = null)
     {
         using var scope = serviceProvider.CreateScope();
         var dbContext = scope.ServiceProvider.GetRequiredService<ServiceControlDbContextBase>();

src/ServiceControl.Persistence.Sql.SqlServer/Migrations/20241208000004_AddCustomChecks.cs

@@ -0,0 +1,44 @@
+namespace ServiceControl.Persistence.Sql.SqlServer.Migrations;
+
+using Microsoft.EntityFrameworkCore.Migrations;
+
+#nullable disable
+
+/// <inheritdoc />
+public partial class AddCustomChecks : Migration
+{
+    /// <inheritdoc />
+    protected override void Up(MigrationBuilder migrationBuilder)
+    {
+        migrationBuilder.CreateTable(
+            name: "CustomChecks",
+            columns: table => new
+            {
+                Id = table.Column<Guid>(type: "uniqueidentifier", nullable: false),
+                CustomCheckId = table.Column<string>(type: "nvarchar(500)", maxLength: 500, nullable: false),
+                Category = table.Column<string>(type: "nvarchar(500)", maxLength: 500, nullable: true),
+                Status = table.Column<int>(type: "int", nullable: false),
+                ReportedAt = table.Column<DateTime>(type: "datetime2", nullable: false),
+                FailureReason = table.Column<string>(type: "nvarchar(max)", nullable: true),
+                EndpointName = table.Column<string>(type: "nvarchar(500)", maxLength: 500, nullable: false),
+                HostId = table.Column<Guid>(type: "uniqueidentifier", nullable: false),
+                Host = table.Column<string>(type: "nvarchar(500)", maxLength: 500, nullable: false)
+            },
+            constraints: table =>
+            {
+                table.PrimaryKey("PK_CustomChecks", x => x.Id);
+            });
+
+        migrationBuilder.CreateIndex(
+            name: "IX_CustomChecks_Status",
+            table: "CustomChecks",
+            column: "Status");
+    }
+
+    /// <inheritdoc />
+    protected override void Down(MigrationBuilder migrationBuilder)
+    {
+        migrationBuilder.DropTable(
+            name: "CustomChecks");
+    }
+}