Upload file activity

2026-03-26 18:23:59 +01:00
parent 6c413f5174
commit 0384de792b
4 changed files with 119 additions and 1 deletions
@@ -65,7 +65,10 @@ namespace Musify.Infrastructure.MassTransit.Activities
                await fileStream.CopyToAsync(destinationStream, executeContext.CancellationToken);

                await processTrackingStore.CompleteStepAsync(processId, stepId, executeContext.CancellationToken);
-                return executeContext.Completed();
+                return executeContext.Completed(new DownloadFileFromBucketLog(
+                    executeContext.Arguments.BucketName,
+                    executeContext.Arguments.KeyName,
+                    destinationPath));
            }
            catch (Exception exception)
            {
@@ -0,0 +1,95 @@
+using MassTransit;
+using Microsoft.Extensions.Logging;
+using Musify.Application.Contracts.Infrastructure;
+using Musify.Domain.Entities;
+using Musify.Infrastructure.MassTransit.Arguments;
+using Musify.Infrastructure.MassTransit.Logs;
+
+namespace Musify.Infrastructure.MassTransit.Activities.Files
+{
+    public class UploadFileToBucketActivity(IStorageHandler storageHandler,
+        ILogger<UploadFileToBucketActivity> logger,
+        IProcessTrackingStore processTrackingStore)
+        : IActivity<UploadFileToBucketArguments, UploadFileToBucketLog>
+    {
+        public const string ExecuteEndpointName = "upload-file-to-bucket";
+
+        public async Task<ExecutionResult> Execute(ExecuteContext<UploadFileToBucketArguments> executeContext)
+        {
+            var processId = await processTrackingStore.GetOrCreateProcessAsync(
+                "RoutingSlip",
+                executeContext.CorrelationId ?? executeContext.TrackingNumber,
+                executeContext.ConversationId,
+                executeContext.MessageId,
+                executeContext.CancellationToken);
+
+            var stepId = await processTrackingStore.StartStepAsync(
+                processId,
+                nameof(UploadFileToBucketActivity),
+                ProcessStepComponentType.Activity,
+                0,
+                executeContext.CancellationToken);
+
+            try
+            {
+                using var fileStream = File.OpenRead(executeContext.Arguments.SourceFilePath);
+
+                var uploadResult = await storageHandler.UploadFileAsync(
+                    fileStream,
+                    executeContext.Arguments.ContentType,
+                    executeContext.Arguments.DestinationBucketName,
+                    executeContext.Arguments.DestinationKeyName,
+                    executeContext.CancellationToken);
+
+                if (!uploadResult.IsSuccess)
+                {
+                    var errors = string.Join("; ", uploadResult.Errors);
+                    logger.LogWarning("Failed to upload file to bucket. Bucket: {BucketName}, Key: {KeyName}, Errors: {Errors}",
+                        executeContext.Arguments.DestinationBucketName,
+                        executeContext.Arguments.DestinationKeyName,
+                        errors);
+                    throw new Exception(errors);
+                }
+
+                await processTrackingStore.CompleteStepAsync(processId, stepId, executeContext.CancellationToken);
+                return executeContext.Completed(new UploadFileToBucketLog(
+                    executeContext.Arguments.DestinationBucketName,
+                    executeContext.Arguments.DestinationKeyName));
+            }
+            catch (Exception exception)
+            {
+                logger.LogError(exception, "An error occurred while uploading file to bucket for process {ProcessId} at step {StepId}", processId, stepId);
+                await processTrackingStore.FailStepAsync(processId, stepId, exception.Message, executeContext.CancellationToken);
+                throw;
+            }
+        }
+
+        public async Task<CompensationResult> Compensate(CompensateContext<UploadFileToBucketLog> compensateContext)
+        {
+            try
+            {
+                var uploadResult = await storageHandler.RemoveFileAsync(
+                    compensateContext.Log.DestinationBucketName,
+                    compensateContext.Log.DestinationKeyName,
+                    compensateContext.CancellationToken);
+                
+                if (!uploadResult.IsSuccess)
+                {
+                    var errors = string.Join("; ", uploadResult.Errors);
+                    var errorMessage = $"Failed to remove file from bucket during compensation. Bucket: {compensateContext.Log.DestinationBucketName}, Key: {compensateContext.Log.DestinationKeyName}, Errors: {errors}";
+
+                    logger.LogError(errorMessage);
+                    return compensateContext.Failed(new Exception(errorMessage));
+                }
+                return compensateContext.Compensated();
+            }
+            catch (Exception exception)
+            {
+                logger.LogError(exception, "An error occurred while compensating upload file to bucket activity for bucket {BucketName} and key {KeyName}",
+                    compensateContext.Log.DestinationBucketName,
+                    compensateContext.Log.DestinationKeyName);
+                return compensateContext.Failed(exception);
+            }
+        }
+    }
+}