mirror of
https://github.com/Polaris-Entertainment/bytefy.git
synced 2026-04-11 18:29:38 +00:00
Test queue
This commit is contained in:
@@ -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));
|
||||
}
|
||||
}
|
||||
@@ -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();
|
||||
|
||||
@@ -5,7 +5,7 @@
|
||||
<Nullable>enable</Nullable>
|
||||
<ImplicitUsings>enable</ImplicitUsings>
|
||||
<InvariantGlobalization>true</InvariantGlobalization>
|
||||
<PublishAot>true</PublishAot>
|
||||
<PublishAot>false</PublishAot>
|
||||
<DockerDefaultTargetOS>Linux</DockerDefaultTargetOS>
|
||||
</PropertyGroup>
|
||||
|
||||
|
||||
Reference in New Issue
Block a user