Enforce body storage max size

Author: johnsimons

src/ServiceControl.Audit.Persistence.PostgreSQL/UnitOfWork/PostgreSQLAuditIngestionUnitOfWork.cs

@@ -14,11 +14,13 @@ class PostgreSQLAuditIngestionUnitOfWork : IAuditIngestionUnitOfWork
 {
     readonly NpgsqlBatch batch;
     readonly NpgsqlConnection connection;
+    readonly DatabaseConfiguration databaseConfiguration;
 
-    public PostgreSQLAuditIngestionUnitOfWork(NpgsqlConnection connection)
+    public PostgreSQLAuditIngestionUnitOfWork(NpgsqlConnection connection, DatabaseConfiguration databaseConfiguration)
     {
         batch = new NpgsqlBatch(connection);
         this.connection = connection;
+        this.databaseConfiguration = databaseConfiguration;
     }
 
     public async ValueTask DisposeAsync()
@@ -47,7 +49,7 @@ INSERT INTO processed_messages (
                 );";
 
         processedMessage.MessageMetadata["ContentLength"] = body.Length;
-        if (!body.IsEmpty)
+        if (!body.IsEmpty && body.Length <= databaseConfiguration.MaxBodySizeToStore)
         {
             cmd.Parameters.AddWithValue("body", body);
         }

src/ServiceControl.Audit.Persistence.PostgreSQL/UnitOfWork/PostgreSQLAuditIngestionUnitOfWorkFactory.cs

@@ -5,19 +5,12 @@ namespace ServiceControl.Audit.Persistence.PostgreSQL.UnitOfWork;
 using ServiceControl.Audit.Persistence.UnitOfWork;
 using ServiceControl.Audit.Persistence.PostgreSQL;
 
-class PostgreSQLAuditIngestionUnitOfWorkFactory : IAuditIngestionUnitOfWorkFactory
+class PostgreSQLAuditIngestionUnitOfWorkFactory(PostgreSQLConnectionFactory connectionFactory, DatabaseConfiguration databaseConfiguration) : IAuditIngestionUnitOfWorkFactory
 {
-    readonly PostgreSQLConnectionFactory connectionFactory;
-
-    public PostgreSQLAuditIngestionUnitOfWorkFactory(PostgreSQLConnectionFactory connectionFactory)
-    {
-        this.connectionFactory = connectionFactory;
-    }
-
     public async ValueTask<IAuditIngestionUnitOfWork> StartNew(int batchSize, CancellationToken cancellationToken)
     {
         var connection = await connectionFactory.OpenConnection(cancellationToken);
-        return new PostgreSQLAuditIngestionUnitOfWork(connection);
+        return new PostgreSQLAuditIngestionUnitOfWork(connection, databaseConfiguration);
     }
 
     public bool CanIngestMore() => true;