UoW implementation based on ef core context.

UoW.cs

public class UnitOfWork : IUnitOfWork
{
    private readonly DbContext _dbContext;
    private readonly IMediator _mediator;

    public UnitOfWork(DbContext dbContext, IMediator mediator)
    {
        _dbContext = dbContext ?? throw new ArgumentNullException(nameof(dbContext));
        _mediator = mediator ?? throw new ArgumentNullException(nameof(mediator));
    }

    public async Task Commit(CancellationToken cancellationToken = default)
    {
        foreach (var entityEntry in _dbContext.ChangeTracker.Entries<IValidatable>().Where(x => x.State == EntityState.Added || x.State == EntityState.Modified))
            entityEntry.Entity.EnsureValid();

        var domainEvents = _dbContext.ChangeTracker.Entries<BaseEntity>()
            .SelectMany(x => x.Entity.GetEvents().Where(y => y is IDomainEvent))
            .ToArray();
        foreach (var @event in domainEvents)
            await _mediator.Publish(@event, cancellationToken);

        var integrationEvents = _dbContext.ChangeTracker.Entries<BaseEntity>()
            .SelectMany(x => x.Entity.GetEvents().Where(y => !(y is IDomainEvent)))
            .ToArray();

        await _dbContext.SaveChangesAsync(cancellationToken);

        foreach (var @event in integrationEvents)
            await _mediator.Publish(@event, cancellationToken);
    }
}