Test queue

This commit is contained in:
Myx
2024-10-12 17:08:43 +02:00
parent 5bf3041bb0
commit f38861c80c
32 changed files with 167 additions and 37 deletions

View File

@@ -0,0 +1,48 @@
using System.Collections.Concurrent;
using ImageMagick;
namespace bytefy.image;
public class ConversionTask
{
public byte[] ImageData { get; set; }
public MagickFormat Format { get; set; }
}
public class ConversionQueueService : BackgroundService
{
private readonly ConcurrentQueue<(ConversionTask Task, TaskCompletionSource<(byte[], string)> CompletionSource)> _queue = new();
public void QueueConversion(ConversionTask task, TaskCompletionSource<(byte[], string)> completionSource)
{
_queue.Enqueue((task, completionSource));
}
protected override async Task ExecuteAsync(CancellationToken stoppingToken)
{
while (!stoppingToken.IsCancellationRequested)
{
if (_queue.TryDequeue(out var item))
{
var result = await ProcessConversionAsync(item.Task);
item.CompletionSource.SetResult(result);
}
else
{
await Task.Delay(50, stoppingToken); // Shorten the delay to check the queue more frequently
}
}
}
private Task<(byte[], string)> ProcessConversionAsync(ConversionTask task)
{
using var magickImage = new MagickImage(task.ImageData);
magickImage.Format = task.Format;
var resultStream = new MemoryStream();
magickImage.Write(resultStream);
resultStream.Position = 0;
var mimeType = MimeTypes.MimeTypeMap.GetMimeType($"image/{task.Format.ToString().ToLower()}");
return Task.FromResult((resultStream.ToArray(), mimeType));
}
}

View File

@@ -1,42 +1,44 @@
using bytefy.image;
using ImageMagick;
using Microsoft.AspNetCore.Antiforgery;
var builder = WebApplication.CreateBuilder(args);
builder.Services.AddAntiforgery(options => options.HeaderName = "2311d8d8-607d-4747-8939-1bde65643254");
builder.Services.AddSingleton<ConversionQueueService>();
builder.Services.AddHostedService(provider => provider.GetRequiredService<ConversionQueueService>());
var app = builder.Build();
app.UseAntiforgery();
var conversionQueue = app.Services.GetRequiredService<ConversionQueueService>();
app.MapPost("/convert/{format}", async (IFormFile image, string format) =>
{
try
if (!Enum.TryParse(format, true, out MagickFormat magickFormat) || magickFormat == MagickFormat.Unknown)
return Results.BadRequest("Invalid format");
if (image == null || image.Length == 0)
return Results.BadRequest("No image provided");
if (image.Length > 20 * 1024 * 1024)
throw new Exception("Image size too large");
using var memoryStream = new MemoryStream();
await image.CopyToAsync(memoryStream);
var conversionTask = new ConversionTask
{
if (!Enum.TryParse(format, true, out MagickFormat magickFormat) || magickFormat == MagickFormat.Unknown)
return Results.BadRequest("Invalid format");
ImageData = memoryStream.ToArray(),
Format = magickFormat
};
if (image == null || image.Length == 0)
return Results.BadRequest("No image provided");
var tcs = new TaskCompletionSource<(byte[], string)>();
conversionQueue.QueueConversion(conversionTask, tcs);
if (image.Length > 20 * 1024 * 1024)
throw new Exception("Image size too large");
var (imageData, mimeType) = await tcs.Task;
using var memoryStream = new MemoryStream();
await image.CopyToAsync(memoryStream);
var magickImage = new MagickImage(memoryStream.ToArray());
magickImage.Format = magickFormat;
var resultStream = new MemoryStream();
magickImage.Write(resultStream);
resultStream.Position = 0;
var mimeType = MimeTypes.MimeTypeMap.GetMimeType($"image/{magickFormat.ToString().ToLower()}");
var fileName = Path.GetFileNameWithoutExtension(image.FileName);
return Results.File(resultStream, mimeType, fileDownloadName: $"{fileName}.{magickFormat.ToString().ToLower()}");
}
catch (NotSupportedException) {
return Results.BadRequest();
}
return Results.File(new MemoryStream(imageData), mimeType, $"{Path.GetFileNameWithoutExtension(image.FileName)}.{magickFormat.ToString().ToLower()}");
});
app.MapGet("/antiforgery/token", (IAntiforgery forgeryService, HttpContext context) =>
@@ -46,4 +48,12 @@ app.MapGet("/antiforgery/token", (IAntiforgery forgeryService, HttpContext conte
return TypedResults.Content(xsrfToken, "text/plain");
});
app.Run();
app.MapGet("/formats", () =>
{
var formats = Enum.GetNames<MagickFormat>().ToList();
formats.Remove("Unknown");
return Results.Ok(formats);
});
app.Run();

View File

@@ -5,7 +5,7 @@
<Nullable>enable</Nullable>
<ImplicitUsings>enable</ImplicitUsings>
<InvariantGlobalization>true</InvariantGlobalization>
<PublishAot>true</PublishAot>
<PublishAot>false</PublishAot>
<DockerDefaultTargetOS>Linux</DockerDefaultTargetOS>
</PropertyGroup>