use std::ops::Deref;
use bitcoin::Transaction;
use lightning::chain::chaininterface::BroadcasterInterface;
use tokio::sync::{mpsc, Mutex, MutexGuard};
use crate::logger::{log_error, LdkLogger};
const BCAST_PACKAGE_QUEUE_SIZE: usize = 50;
pub(crate) struct TransactionBroadcaster<L: Deref>
where
L::Target: LdkLogger,
{
queue_sender: mpsc::Sender<Vec<Transaction>>,
queue_receiver: Mutex<mpsc::Receiver<Vec<Transaction>>>,
logger: L,
}
impl<L: Deref> TransactionBroadcaster<L>
where
L::Target: LdkLogger,
{
pub(crate) fn new(logger: L) -> Self {
let (queue_sender, queue_receiver) = mpsc::channel(BCAST_PACKAGE_QUEUE_SIZE);
Self { queue_sender, queue_receiver: Mutex::new(queue_receiver), logger }
}
pub(crate) async fn get_broadcast_queue(
&self,
) -> MutexGuard<'_, mpsc::Receiver<Vec<Transaction>>> {
self.queue_receiver.lock().await
}
}
impl<L: Deref> BroadcasterInterface for TransactionBroadcaster<L>
where
L::Target: LdkLogger,
{
fn broadcast_transactions(&self, txs: &[&Transaction]) {
let package = txs.iter().map(|&t| t.clone()).collect::<Vec<Transaction>>();
self.queue_sender.try_send(package).unwrap_or_else(|e| {
log_error!(self.logger, "Failed to broadcast transactions: {}", e);
});
}
}