feat(core): TaskManagementService - отправка сообщений

This commit is contained in:
Mikhail Trubnikov
2026-04-15 16:48:23 +10:00
parent 6d792f04b1
commit 2a6fd7177d
8 changed files with 103 additions and 33 deletions

View File

@@ -1,8 +1,14 @@
using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Mvc;
using PARR.API.Contracts.V1;
using PARR.API.Contracts.V1.Responses.Base;
using PARR.API.Controllers.V1.Base;
using PARR.API.Services.Interfaces;
using PARR.API.Settings;
using PARR.Core.Services.Task.Interfaces;
using PARR.Domain.Common.Roles;
using PARR.Domain.Entities.Base.History;
using PARR.Domain.Enums;
namespace PARR.API.Controllers.V1.Statistics
{
@@ -12,9 +18,22 @@ namespace PARR.API.Controllers.V1.Statistics
[Authorize(Roles = ParrRoles.EsppRobot.RoleOrAdmin)]
public class StatWorkloadController : BaseApiController
{
public StatWorkloadController()
{
private readonly ITaskManagementService taskManagementService;
private readonly IClientService clientService;
private readonly MqSettings mqSettings;
private readonly ILogger<StatWorkloadController> logger;
public StatWorkloadController(
ITaskManagementService taskManagementService,
IClientService clientService,
MqSettings mqSettings,
ILogger<StatWorkloadController> logger
)
{
this.taskManagementService = taskManagementService;
this.clientService = clientService;
this.mqSettings = mqSettings;
this.logger = logger;
}
@@ -25,8 +44,31 @@ namespace PARR.API.Controllers.V1.Statistics
[HttpPost(ApiRoutes.Workload.Build)]
public async Task<IActionResult> Build()
{
var initiator = new HistoryInitiator
{
InitiatorIp = clientService.GetClientIp()?.ToString(),
InitiatorParrComponentId = ParrComponentsEnum.Api,
InitiatorComment = "Отправлен запрос из API на формирование отчета о загруженности"
};
return Ok();
try
{
mqSettings.Tasks.TryGetValue(TaskTypeEnum.Workload, out var queueSettings);
if (queueSettings == null)
{
logger.LogError("Не удалось получить настройки очереди для TaskType: {TaskType}", TaskTypeEnum.Workload);
throw new InvalidOperationException($"Не удалось получить настройки очереди для TaskTypeEnum.Workload");
}
var taskId = await taskManagementService.CreateTaskAsync<object?>(TaskTypeEnum.Workload, default, initiator, queueSettings);
return Ok(new Response<string?>(null, true, null!, $"Создана задача {taskId} на формирование отчета"));
}
catch (Exception ex)
{
return BadRequest(new Response(false, new List<ErrorModel> { new ErrorModel { Message = ex.Message } }));
}
}

View File

@@ -3,6 +3,8 @@ using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using PARR.Core.Common.Implementations;
using PARR.Core.Common.Interfaces;
using PARR.Core.Services.Task.Implementations;
using PARR.Core.Services.Task.Interfaces;
using PARR.Domain.Settings;
namespace PARR.Core
@@ -37,6 +39,7 @@ namespace PARR.Core
#region Services
services.AddScoped<ITaskManagementService, TaskManagementService>();
//services.AddScoped<IUserService, UserService>();
#endregion

View File

@@ -8,7 +8,6 @@
<ItemGroup>
<Folder Include="Infrastructure\" />
<Folder Include="Services\" />
</ItemGroup>
<ItemGroup>

View File

@@ -1,6 +1,9 @@
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Logging;
using PARR.Core.Common.Interfaces.RabbitServices;
using PARR.Core.Repositories.Interfaces.TaskRepositories;
using PARR.Core.Services.Task.Interfaces;
using PARR.Domain.Common.Rabbit.Messages;
using PARR.Domain.Entities.Base.History;
using PARR.Domain.Entities.TaskEntities;
using PARR.Domain.Enums;
@@ -8,29 +11,32 @@ using PARR.Domain.Settings;
using System.Text.Encodings.Web;
using System.Text.Json;
namespace PARR.DAL.TaskServices
namespace PARR.Core.Services.Task.Implementations
{
internal class TaskManagementService : ITaskManagementService
{
private readonly ILogger<TaskManagementService> logger;
private readonly ITaskTypeRepository taskTypeService;
private readonly ITaskRepository taskService;
private readonly ITaskTypeRepository taskTypeRepository;
private readonly ITaskRepository taskRepository;
private readonly IRabbitService rabbitService;
public TaskManagementService(
ILogger<TaskManagementService> logger,
ITaskTypeRepository taskTypeService,
ITaskRepository taskService
ITaskTypeRepository taskTypeRepository,
ITaskRepository taskRepository,
IRabbitService rabbitService
)
{
this.logger = logger;
this.taskTypeService = taskTypeService;
this.taskService = taskService;
this.taskTypeRepository = taskTypeRepository;
this.taskRepository = taskRepository;
this.rabbitService = rabbitService;
}
public async Task<Guid> CreateTaskAsync<T>(TaskTypeEnum typeCode, T payload, IHistoryInitiator initiator, IMqSettings mqSettings)
{
var taskType = await taskTypeService.Get().AsNoTracking().FirstOrDefaultAsync(t => t.Code == typeCode);
var taskType = await taskTypeRepository.Get().AsNoTracking().FirstOrDefaultAsync(t => t.Code == typeCode);
if (taskType == null)
{
@@ -45,10 +51,7 @@ namespace PARR.DAL.TaskServices
if (existingTask != null)
{
logger.LogInformation(
"Задача типа {TypeCode} уже активна (id: {ExistingId}). Возвращаем существующую.",
typeCode, existingTask.Id);
logger.LogInformation("Задача типа {TypeCode} уже активна (id: {ExistingId}). Возвращаем существующую.", typeCode, existingTask.Id);
return existingTask.Id;
}
}
@@ -63,26 +66,30 @@ namespace PARR.DAL.TaskServices
StatusCode = TaskItemStatusEnum.Pending,
Payload = payloadStr,
RetryCount = 0,
ProcessedAt = null,
InitiatorIp = initiator.InitiatorIp,
InitiatorParrComponentId = initiator.InitiatorParrComponentId,
InitiatorComment = initiator.InitiatorComment
ProcessedAt = null
};
if (!await taskService.CreateAsync(task) || !await taskService.CommitAsync())
if (!await taskRepository.CreateAsync(task) || !await taskRepository.CommitAsync(initiator))
{
logger.LogError("Ошибка при сохранении задачи в БД");
return default;
throw new DbUpdateException("Ошибка при сохранении задачи в БД");
}
logger.LogInformation("Задача {TaskId} типа {TypeCode} сохранена в БД со статусом Pending", task.Id, typeCode);
// Публикуме задачу в очередь. Если вдруг даже не получится,
// задача останется в БД со статусом pending, и Reconciliation Task позже ее возмет в работу
var message = new TaskMessage { TaskId = task.Id, TypeCode = task.TypeCode };
// todo: опубликовать в очередь
// вопросы, зачем scoped? может можно AddTransient?
// нужно придумать шаблонную модель для очереди
var sendResult = await rabbitService.SendAsync(mqSettings, new List<object> { message });
if (sendResult.IsSuccess)
{
logger.LogDebug("Задача {TaskId} типа {TypeCode} отправлена в очередь {QueueName}", task.Id, typeCode, mqSettings.QueueName);
}
else
{
// Не пробрасываем исключение дальше — задача создана, просто не в очереди.
// задача останется в БД со статусом pending, и Reconciliation Task позже ее возмет в работу
logger.LogError("Не удалось опубликовать задачу {TaskId} в очередь. Задача осталась в БД со статусом Pending.", task.Id);
}
return task.Id;
}
@@ -95,7 +102,7 @@ namespace PARR.DAL.TaskServices
/// <returns></returns>
private async Task<TaskItem?> GetActiveSingletonTaskAsync(TaskTypeEnum typeCode)
{
return await taskService.Get().AsNoTracking()
return await taskRepository.Get().AsNoTracking()
.FirstOrDefaultAsync(t => t.TypeCode == typeCode && (
t.StatusCode == TaskItemStatusEnum.Pending
|| t.StatusCode == TaskItemStatusEnum.Processing
@@ -109,13 +116,16 @@ namespace PARR.DAL.TaskServices
/// <typeparam name="T"></typeparam>
/// <param name="payload"></param>
/// <returns></returns>
private string PayloadToString<T>(T payload)
private string? PayloadToString<T>(T payload)
{
var jsonOptions = new JsonSerializerOptions
{
Encoder = JavaScriptEncoder.UnsafeRelaxedJsonEscaping
};
if (payload == null)
return null;
return JsonSerializer.Serialize(payload, jsonOptions);
}

View File

@@ -2,7 +2,7 @@
using PARR.Domain.Enums;
using PARR.Domain.Settings;
namespace PARR.DAL.TaskServices
namespace PARR.Core.Services.Task.Interfaces
{
/// <summary>
/// Управления задачами (очередями)

View File

@@ -22,7 +22,6 @@ using PARR.DAL.Services.Interfaces;
using PARR.DAL.Services.Interfaces.Job;
using PARR.DAL.Services.Interfaces.Schedule;
using PARR.DAL.Services.Interfaces.Unit;
using PARR.DAL.TaskServices;
using PARR.Domain.Settings;
namespace PARR.DAL
@@ -154,8 +153,6 @@ namespace PARR.DAL
services.AddScoped<ITaskRepository, TaskRepository>();
services.AddScoped<ITaskTypeRepository, TaskTypeRepository>();
services.AddTransient<ITaskManagementService, TaskManagementService>();
#endregion
//services.AddTransient<INextRunModifierService, NextRunModifierService>();

View File

@@ -99,6 +99,9 @@ namespace PARR.DAL.Repositories.Base
/// <param name="initiator"></param>
private void SetInitiator(IHistoryInitiator? initiator)
{
if (initiator == null)
return;
logger.LogDebug("Устанавливаю инициатора для изменений");
// Задаем инициатора только для новых и измененных записей

View File

@@ -0,0 +1,16 @@
using PARR.Domain.Enums;
using System.Text.Json.Serialization;
namespace PARR.Domain.Common.Rabbit.Messages
{
/// <summary>
/// Модель для отправки сообщений через ITaskManagement
/// </summary>
public record TaskMessage
{
public Guid TaskId { get; init; }
[JsonConverter(typeof(JsonStringEnumConverter))]
public TaskTypeEnum TypeCode { get; init; }
}
}