DD.WellWorkover.Cloud/AsbCloudWebApi/SignalR/WellInfoHub.cs

using System.Threading;
using System.Threading.Tasks;
using AsbCloudApp.IntegrationEvents;
using AsbCloudApp.IntegrationEvents.Interfaces;
using AsbCloudApp.Services;
using AsbCloudInfrastructure.Services;
using Microsoft.AspNetCore.SignalR;

namespace AsbCloudWebApi.SignalR;

public class WellInfoHub : BaseHub, IIntegrationEventHandler<UpdateWellInfoEvent>
{
	private readonly IHubContext<WellInfoHub> hubContext;
	private readonly IWellService wellService;
	private readonly WellInfoService wellInfoService;

	public WellInfoHub(IHubContext<WellInfoHub> hubContext,
		IWellService wellService,
		WellInfoService wellInfoService)
	{
		this.hubContext = hubContext;
		this.wellService = wellService;
		this.wellInfoService = wellInfoService;
	}

	public override async Task AddToGroup(string groupName)
	{
		var idWell = int.Parse(groupName.Split('_')[2]);
		
		await Groups.AddToGroupAsync(Context.ConnectionId, groupName);

		await HandleAsync(new UpdateWellInfoEvent(idWell), CancellationToken.None);
	}

	public async Task HandleAsync(UpdateWellInfoEvent integrationEvent, CancellationToken cancellationToken)
	{
		const string method = "update_well_info";

		var well = await wellService.GetOrDefaultAsync(integrationEvent.IdWell, cancellationToken);

		if(well is null)
			return;
		
		var wellInfo = wellInfoService.FirstOrDefault(w => w.Id == well.Id);

		await hubContext.Clients.Group($"well_info_{integrationEvent.IdWell}")
			.SendAsync(method, new
			{
				Well = well,
				WellInfo = wellInfo 
			}, cancellationToken);
	}
}