mirror of
https://github.com/g3rv4/FakeRelay.git
synced 2024-11-21 23:03:07 +01:00
Remove background tasks
This commit is contained in:
parent
032e03c823
commit
e05d91d93d
@ -1,7 +1,5 @@
|
||||
using System.Threading.Tasks;
|
||||
using FakeRelay.Core;
|
||||
using FakeRelay.Core.Helpers;
|
||||
using FakeRelay.Web.Services;
|
||||
using Microsoft.AspNetCore.Mvc;
|
||||
|
||||
namespace FakeRelay.Web.Controllers;
|
||||
@ -10,15 +8,15 @@ public class AcitivityPubController : Controller
|
||||
{
|
||||
|
||||
[Route("actor")]
|
||||
public async Task<ActionResult> Actor([FromServices] IBackgroundTaskQueue taskQueue) =>
|
||||
public async Task<ActionResult> Actor() =>
|
||||
Content(MastodonHelper.GetActorStaticContent(), "application/activity+json; charset=utf-8");
|
||||
|
||||
[Route("inbox"), HttpPost]
|
||||
public async Task<ActionResult> Inbox([FromBody] ActivityPubModel model, [FromServices] IBackgroundTaskQueue taskQueue)
|
||||
public async Task<ActionResult> Inbox([FromBody] ActivityPubModel model)
|
||||
{
|
||||
if (model.Type == "Follow")
|
||||
{
|
||||
await taskQueue.QueueBackgroundWorkItemAsync(_ => MastodonHelper.ProcessInstanceFollowAsync(model));
|
||||
await MastodonHelper.ProcessInstanceFollowAsync(model);
|
||||
}
|
||||
|
||||
return Content("{}", "application/activity+json");
|
||||
|
@ -1,5 +1,4 @@
|
||||
using FakeRelay.Core;
|
||||
using FakeRelay.Web.Services;
|
||||
using Microsoft.AspNetCore.HttpOverrides;
|
||||
using Prometheus;
|
||||
|
||||
@ -26,9 +25,6 @@ builder.Services.Configure<ForwardedHeadersOptions>(options =>
|
||||
options.ForwardLimit = 1;
|
||||
});
|
||||
|
||||
builder.Services.AddHostedService<QueuedHostedService>();
|
||||
builder.Services.AddSingleton<IBackgroundTaskQueue>(_ => new BackgroundTaskQueue(30));
|
||||
|
||||
var app = builder.Build();
|
||||
|
||||
// Configure the HTTP request pipeline.
|
||||
|
@ -1,102 +0,0 @@
|
||||
using System;
|
||||
using System.Threading;
|
||||
using System.Threading.Channels;
|
||||
using System.Threading.Tasks;
|
||||
using Microsoft.Extensions.Hosting;
|
||||
using Microsoft.Extensions.Logging;
|
||||
|
||||
namespace FakeRelay.Web.Services;
|
||||
|
||||
public interface IBackgroundTaskQueue
|
||||
{
|
||||
ValueTask QueueBackgroundWorkItemAsync(Func<CancellationToken, Task> workItem);
|
||||
|
||||
ValueTask<Func<CancellationToken, Task>> DequeueAsync(
|
||||
CancellationToken cancellationToken);
|
||||
}
|
||||
|
||||
public class BackgroundTaskQueue : IBackgroundTaskQueue
|
||||
{
|
||||
private readonly Channel<Func<CancellationToken, Task>> _queue;
|
||||
|
||||
public BackgroundTaskQueue(int capacity)
|
||||
{
|
||||
// Capacity should be set based on the expected application load and
|
||||
// number of concurrent threads accessing the queue.
|
||||
// BoundedChannelFullMode.Wait will cause calls to WriteAsync() to return a task,
|
||||
// which completes only when space became available. This leads to backpressure,
|
||||
// in case too many publishers/calls start accumulating.
|
||||
var options = new BoundedChannelOptions(capacity)
|
||||
{
|
||||
FullMode = BoundedChannelFullMode.Wait
|
||||
};
|
||||
_queue = Channel.CreateBounded<Func<CancellationToken, Task>>(options);
|
||||
}
|
||||
|
||||
public async ValueTask QueueBackgroundWorkItemAsync(
|
||||
Func<CancellationToken, Task> workItem)
|
||||
{
|
||||
if (workItem == null)
|
||||
{
|
||||
throw new ArgumentNullException(nameof(workItem));
|
||||
}
|
||||
|
||||
await _queue.Writer.WriteAsync(workItem);
|
||||
}
|
||||
|
||||
public async ValueTask<Func<CancellationToken, Task>> DequeueAsync(
|
||||
CancellationToken cancellationToken)
|
||||
{
|
||||
var workItem = await _queue.Reader.ReadAsync(cancellationToken);
|
||||
|
||||
return workItem;
|
||||
}
|
||||
}
|
||||
|
||||
public class QueuedHostedService : BackgroundService
|
||||
{
|
||||
private readonly ILogger<QueuedHostedService> _logger;
|
||||
|
||||
public QueuedHostedService(IBackgroundTaskQueue taskQueue,
|
||||
ILogger<QueuedHostedService> logger)
|
||||
{
|
||||
TaskQueue = taskQueue;
|
||||
_logger = logger;
|
||||
}
|
||||
|
||||
public IBackgroundTaskQueue TaskQueue { get; }
|
||||
|
||||
protected override async Task ExecuteAsync(CancellationToken stoppingToken)
|
||||
{
|
||||
_logger.LogInformation(
|
||||
$"Queued Hosted Service is running.");
|
||||
|
||||
await BackgroundProcessing(stoppingToken);
|
||||
}
|
||||
|
||||
private async Task BackgroundProcessing(CancellationToken stoppingToken)
|
||||
{
|
||||
while (!stoppingToken.IsCancellationRequested)
|
||||
{
|
||||
var workItem =
|
||||
await TaskQueue.DequeueAsync(stoppingToken);
|
||||
|
||||
try
|
||||
{
|
||||
await workItem(stoppingToken);
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
_logger.LogError(ex,
|
||||
"Error occurred executing {WorkItem}.", nameof(workItem));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public override async Task StopAsync(CancellationToken stoppingToken)
|
||||
{
|
||||
_logger.LogInformation("Queued Hosted Service is stopping.");
|
||||
|
||||
await base.StopAsync(stoppingToken);
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue
Block a user