aljaz/btcpayserver
1
0
Fork 0
mirror of https://github.com/aljazceru/btcpayserver.git synced 2026-01-05 23:24:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add thread limit for updating payment states for payment invoices in NBXplorerListener

Browse Source
This commit is contained in:
Igor Rylko
2020-10-27 21:09:53 +02:00
parent 2083954aa5
commit 5d9827fb60
1 changed files with 26 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
BTCPayServer/Payments/Bitcoin/NBXplorerListener.cs
View File

@@ -4,6 +4,7 @@ using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using System.Threading.Tasks.Dataflow;
using BTCPayServer;
using BTCPayServer.Events;
using BTCPayServer.HostedServices;
@@ -26,6 +27,8 @@ namespace BTCPayServer.Payments.Bitcoin
/// </summary>
public class NBXplorerListener : IHostedService
{
private const int UpdatePaymentStatesMaxDegreeOfParallelism = 32;
readonly EventAggregator _Aggregator;
private readonly PayJoinRepository _payJoinRepository;
readonly ExplorerClientProvider _ExplorerClients;
@@ -138,9 +141,7 @@ namespace BTCPayServer.Payments.Bitcoin
switch (newEvent)
{
case NBXplorer.Models.NewBlockEvent evt:
await Task.WhenAll((await _InvoiceRepository.GetPendingInvoices())
.Select(invoiceId => UpdatePaymentStates(wallet, invoiceId))
.ToArray());
await UpdatePaymentStatesForPendingInvoices(wallet);
_Aggregator.Publish(new Events.NewBlockEvent() { CryptoCode = evt.CryptoCode });
break;
case NBXplorer.Models.NewTransactionEvent evt:
@@ -206,6 +207,27 @@ namespace BTCPayServer.Payments.Bitcoin
}
}
private async Task UpdatePaymentStatesForPendingInvoices(BTCPayWallet wallet)
{
var pendingInvoices = await _InvoiceRepository.GetPendingInvoices();
var actionBlock = new ActionBlock<string>(invoiceId => UpdatePaymentStates(wallet, invoiceId),
new ExecutionDataflowBlockOptions
{
CancellationToken = _Cts.Token,
EnsureOrdered = false,
SingleProducerConstrained = true,
MaxDegreeOfParallelism = UpdatePaymentStatesMaxDegreeOfParallelism
});
foreach (var invoiceId in pendingInvoices)
{
await actionBlock.SendAsync(invoiceId);
}
actionBlock.Complete();
await actionBlock.Completion;
}
async Task<InvoiceEntity> UpdatePaymentStates(BTCPayWallet wallet, string invoiceId)
{
var invoice = await _InvoiceRepository.GetInvoice(invoiceId, false);
@@ -278,7 +300,7 @@ namespace BTCPayServer.Payments.Bitcoin
pj.CoinjoinTransactionHash == tx.TransactionHash)
{
// This payment is a coinjoin, so the value of
// the payment output is different from the real value of the payment
// the payment output is different from the real value of the payment
paymentData.Value = pj.CoinjoinValue;
payment.SetCryptoPaymentData(paymentData);
}