Add exception handling to blob store

equinor · Dec 18, 2024 · 9db9ce7 · 9db9ce7
1 parent 14feaaa
commit 9db9ce7
Show file tree

Hide file tree

Showing 7 changed files with 27 additions and 12 deletions.
diff --git a/backend/api.test/Mocks/BlobServiceMock.cs b/backend/api.test/Mocks/BlobServiceMock.cs
@@ -7,7 +7,7 @@ namespace Api.Test.Mocks
 {
     public class MockBlobService : IBlobService
     {
-        public async Task<byte[]> DownloadBlob(string blobName, string containerName, string accountName)
+        public async Task<byte[]?> DownloadBlob(string blobName, string containerName, string accountName)
         {
             using var memoryStream = new System.IO.MemoryStream();
             await Task.CompletedTask;

diff --git a/backend/api.test/Mocks/MapServiceMock.cs b/backend/api.test/Mocks/MapServiceMock.cs
@@ -21,7 +21,7 @@ public class MockMapService : IMapService
             return new MapMetadata();
         }
 
-        public async Task<byte[]> FetchMapImage(string mapName, string installationCode)
+        public async Task<byte[]?> FetchMapImage(string mapName, string installationCode)
         {
             await Task.Run(() => Thread.Sleep(1));
             string filePath = Directory.GetCurrentDirectory() + "Images/MockMapImage.png";

diff --git a/backend/api/Controllers/InspectionController.cs b/backend/api/Controllers/InspectionController.cs
@@ -96,7 +96,10 @@ public async Task<ActionResult<Inspection>> GetInspectionImageById([FromRoute] s
 
             try
             {
-                byte[] inspectionStream = await inspectionService.FetchInpectionImage(inspectionData.BlobName, inspectionData.BlobContainer, inspectionData.StorageAccount);
+                byte[]? inspectionStream = await inspectionService.FetchInpectionImage(inspectionData.BlobName, inspectionData.BlobContainer, inspectionData.StorageAccount);
+
+                if (inspectionStream == null) return NotFound($"Could not retrieve inspection with task Id {taskId}");
+
                 return File(inspectionStream, "image/png");
             }
             catch (Azure.RequestFailedException)

diff --git a/backend/api/Controllers/MapController.cs b/backend/api/Controllers/MapController.cs
@@ -26,12 +26,15 @@ public async Task<ActionResult<byte[]>> GetMap([FromRoute] string installationCo
         {
             try
             {
-                byte[] mapStream = await mapService.FetchMapImage(mapName, installationCode);
+                byte[]? mapStream = await mapService.FetchMapImage(mapName, installationCode);
+
+                if (mapStream == null) return NotFound($"Could not retrieve map '{mapName}' in installation {installationCode}");
+
                 return File(mapStream, "image/png");
             }
             catch (RequestFailedException)
             {
-                return NotFound("Could not find map for this area");
+                return NotFound($"Could not find map '{mapName}' in installation {installationCode}");
             }
         }
     }

diff --git a/backend/api/Services/BlobService.cs b/backend/api/Services/BlobService.cs
@@ -11,7 +11,7 @@ namespace Api.Services
 {
     public interface IBlobService
     {
-        public Task<byte[]> DownloadBlob(string blobName, string containerName, string accountName);
+        public Task<byte[]?> DownloadBlob(string blobName, string containerName, string accountName);
 
         public AsyncPageable<BlobItem> FetchAllBlobs(string containerName, string accountName);
 
@@ -20,13 +20,22 @@ public interface IBlobService
 
     public class BlobService(ILogger<BlobService> logger, IOptions<AzureAdOptions> azureOptions) : IBlobService
     {
-        public async Task<byte[]> DownloadBlob(string blobName, string containerName, string accountName)
+        public async Task<byte[]?> DownloadBlob(string blobName, string containerName, string accountName)
         {
             var blobContainerClient = GetBlobContainerClient(containerName, accountName);
             var blobClient = blobContainerClient.GetBlobClient(blobName);
 
             using var memoryStream = new MemoryStream();
-            await blobClient.DownloadToAsync(memoryStream);
+            try
+            {
+                await blobClient.DownloadToAsync(memoryStream);
+            }
+            catch (RequestFailedException)
+            {
+                logger.LogWarning("Failed to download blob {blobName} from container {containerName}", blobName, containerName);
+                return null;
+            }
+
 
             return memoryStream.ToArray();
         }

diff --git a/backend/api/Services/InspectionService.cs b/backend/api/Services/InspectionService.cs
@@ -13,7 +13,7 @@ namespace Api.Services
 {
     public interface IInspectionService
     {
-        public Task<byte[]> FetchInpectionImage(string inpectionName, string installationCode, string storageAccount);
+        public Task<byte[]?> FetchInpectionImage(string inpectionName, string installationCode, string storageAccount);
         public Task<Inspection> UpdateInspectionStatus(string isarTaskId, IsarTaskStatus isarTaskStatus);
         public Task<Inspection?> ReadByIsarTaskId(string id, bool readOnly = true);
         public Task<Inspection?> AddFinding(InspectionFindingQuery inspectionFindingsQuery, string isarTaskId);
@@ -31,7 +31,7 @@ public class InspectionService(FlotillaDbContext context, ILogger<InspectionServ
     {
         public const string ServiceName = "IDA";
 
-        public async Task<byte[]> FetchInpectionImage(string inpectionName, string installationCode, string storageAccount)
+        public async Task<byte[]?> FetchInpectionImage(string inpectionName, string installationCode, string storageAccount)
         {
             return await blobService.DownloadBlob(inpectionName, installationCode, storageAccount);
         }

diff --git a/backend/api/Services/MapService.cs b/backend/api/Services/MapService.cs
@@ -8,7 +8,7 @@ namespace Api.Services
 {
     public interface IMapService
     {
-        public Task<byte[]> FetchMapImage(string mapName, string installationCode);
+        public Task<byte[]?> FetchMapImage(string mapName, string installationCode);
         public Task<MapMetadata?> ChooseMapFromPositions(IList<Position> positions, string installationCode);
         public Task<MapMetadata?> ChooseMapFromMissionRunTasks(MissionRun mission);
     }
@@ -17,7 +17,7 @@ public class MapService(ILogger<MapService> logger,
             IOptions<MapBlobOptions> blobOptions,
             IBlobService blobService) : IMapService
     {
-        public async Task<byte[]> FetchMapImage(string mapName, string installationCode)
+        public async Task<byte[]?> FetchMapImage(string mapName, string installationCode)
         {
             return await blobService.DownloadBlob(mapName, installationCode, blobOptions.Value.StorageAccount);
         }