public static IWorkerApplicationBuilder UseSkipPollingMiddleware(this IWorkerApplicationBuilder app, ILoggerFactory loggerFactory)
{
// establish logger
var logger = loggerFactory.CreateLogger("SkipPollingMiddleware");
// get the polling service
var pollingService = app.ApplicationServices.GetService <IPollingService>();
// register the middleware
app.Use(async(WorkerApplicationOperation operation, IWorkerApplicationMiddlewareExecutionController context) =>
{
// log
logger.LogInformation("Skipping next polling...");
// done
pollingService.SkipNextPolling();
// done
await context.Invoke();
});
app.UseOnError(async(WorkerApplicationOperation operation, Exception error) => {
// log
logger.LogInformation("Skipping next polling...");
// done
pollingService.SkipNextPolling();
// done
await Task.CompletedTask;
});
return(app);
}
public static IWorkerApplicationBuilder UseStorageQueueProcessor(this IWorkerApplicationBuilder app, ILoggerFactory loggerFactory, IEnumerable <IAzureQueueClientPriorityConfiguration> queuesConfiguration, Func <WorkerApplicationOperation, String, IWorkerApplicationMiddlewareExecutionController, Task> queueMessageProcessor)
{
Task.Run(async() =>
{
// generate a logger
var logger = loggerFactory.CreateLogger("StorageQueueMiddleware");
// initialize the queues
var initializedQueues = await InitializeQueues(queuesConfiguration, logger);
// some state of our function
bool processedJobSinceLastMessage = true;
// register our middleware
logger.LogInformation("Registering StorageQueueMiddleware");
return(app.Use(async(operation, next) =>
{
// check the next queue task
if (processedJobSinceLastMessage)
{
logger.LogInformation("Waiting for new messages to process...");
processedJobSinceLastMessage = false;
}
// retrieve the message
var retrievedMessage = await GetMessageFromQueues(initializedQueues);
if (retrievedMessage != null)
{
// log
logger.LogInformation($"Received new message from queue {retrievedMessage.Item2.PriorityDisplayName} ({retrievedMessage.Item1.AsBytes.Length} bytes)");
// set the state
processedJobSinceLastMessage = true;
// process the next message as task
logger.LogInformation("Processing message...");
await queueMessageProcessor(operation, retrievedMessage.Item1.AsString, next);
// log
logger.LogInformation("Processing message finished...");
}
else
{
// Let the next middleware finish our job
await next.Skip();
}
}));
}).Wait();
return(app);
}
public void Configure(IWorkerApplicationBuilder app, IWorkerHostingEnvironment env, ILoggerFactory loggerFactory, IShutdownNotificationService shutdownService, IPollingService pollingService, ITimeoutService timeoutService)
{
shutdownService.OnShutdownNotification(async() =>
{
// get a logger
var logger = loggerFactory.CreateLogger("ShutdownHandler");
// delay
logger.LogInformation("Delaying shutdown by 10 seconds");
await Task.Delay(5000);
// done
logger.LogInformation("Finished delay");
});
app.Use((WorkerApplicationOperation operation, IWorkerApplicationMiddlewareExecutionController next) =>
{
// get a logger
var logger = loggerFactory.CreateLogger("Processor");
// lookup a scoped service
var scopedService = operation.Services.GetService <IScopedService>();
// log the message
logger.LogInformation("MW01 - InstanceId: {0}", scopedService.InstanceId);
return(next.Invoke());
});
app.Use((WorkerApplicationOperation operation, IWorkerApplicationMiddlewareExecutionController next) =>
{
// get a logger
var logger = loggerFactory.CreateLogger("Processor");
// lookup a scoped service
var scopedService = operation.Services.GetService <IScopedService>();
// log the message
logger.LogInformation("MW02 - InstanceId: {0}", scopedService.InstanceId);
return(next.Invoke());
});
/*app.Use((WorkerApplicationOperation operation, IWorkerApplicationMiddlewareExecutionController next) =>
* {
* // get a logger
* var logger = loggerFactory.CreateLogger("AbortNextPolling");
*
* logger.LogInformation("Abort...");
* pollingService.AbortDuringNextPolling();
*
* return next.Invoke();
* });*/
app.Use(async(WorkerApplicationOperation operation, IWorkerApplicationMiddlewareExecutionController next) =>
{
// get a logger
var logger = loggerFactory.CreateLogger("Processor");
logger.LogInformation("Delaying Job");
// delay
logger.LogInformation($"Delay 5sec - {DateTime.Now}");
await Task.Delay(5000);
// reset the timeout
await timeoutService.ResetExecutionTimeout();
// delay
logger.LogInformation($"Delay 5sec - {DateTime.Now}");
await Task.Delay(5000);
// reset the timeout
await timeoutService.ResetExecutionTimeout();
// delay
logger.LogInformation($"Delay 5sec - {DateTime.Now}");
await Task.Delay(5000);
// reset the timeout
await timeoutService.ResetExecutionTimeout();
// delay
logger.LogInformation($"Delay 5sec - {DateTime.Now}");
await Task.Delay(5000);
// delay
logger.LogInformation($"Delay 5sec - {DateTime.Now}");
await Task.Delay(5000);
// delay
logger.LogInformation($"Delay 5sec - {DateTime.Now}");
await Task.Delay(5000);
// next
await next.Invoke();
});
app.UseOnTimeout(async(WorkerApplicationOperation operation) =>
{
Console.WriteLine("Timeout Exceeded");
await Task.CompletedTask;
// abort
Console.WriteLine("Aborting Worker");
pollingService.AbortDuringNextPolling();
});
}