Add config setting and hash check

alertingest/storage.py

@@ -4,10 +4,19 @@
 
 import abc
 import gzip
+import hashlib
 import logging
 from typing import Any, Set
 
+
+def calculate_sha256(content):
+    sha256_hash = hashlib.sha256()
+    sha256_hash.update(content)
+    return sha256_hash.hexdigest()
+
+
 import boto3
+from botocore.config import Config
 
 
 class AlertDatabaseBackend(abc.ABC):
@@ -92,7 +101,14 @@ def __init__(
         self.known_schemas: Set[int] = set()
 
     def create_client(self, endpoint_url: str | None = None):
-        s3_client = boto3.client("s3", endpoint_url=endpoint_url)
+        s3_client = boto3.client(
+            "s3",
+            endpoint_url=endpoint_url,
+            config=Config(
+                request_checksum_calculation="when_required",
+                response_checksum_validation="when_required",
+            ),
+        )
         return s3_client
 
     def store_alert(
@@ -120,6 +136,11 @@ def store_alert(
 
     def store_schema(self, schema_id: int, encoded_schema: bytes):
         schema_key = f"/alert_archive/v1/schemas/{schema_id}.json"
+
+        data = encoded_schema  # this is the data you're trying to upload
+        calculated_hash = calculate_sha256(data)
+        print(f"Calculated SHA-256 hash: {calculated_hash}")
+
         try:
             self.object_store_client.put_object(
                 Bucket=self.schema_bucket, Key=schema_key, Body=encoded_schema