using HrynCo.NotificationService.DAL.Abstract;
using MediatR;

namespace HrynCo.NotificationService.Services.Behaviors;

public class TransactionBehavior<TRequest, TResponse> : IPipelineBehavior<TRequest, TResponse>
    where TRequest : notnull
{
    private readonly IUnitOfWork _unitOfWork;

    public TransactionBehavior(IUnitOfWork unitOfWork)
    {
        _unitOfWork = unitOfWork;
    }

    public Task<TResponse> Handle(TRequest request, RequestHandlerDelegate<TResponse> next, CancellationToken cancellationToken)
    {
        return _unitOfWork.ExecuteInTransactionAsync(async () =>
        {
            TResponse result = await next();
            await _unitOfWork.SaveChangesAsync(cancellationToken);
            return result;
        });
    }
}