]> git.bitcoin.ninja Git - ldk-sample/commitdiff
Remove unneeded Target enum
authorbenthecarman <benthecarman@live.com>
Mon, 18 Sep 2023 20:30:32 +0000 (15:30 -0500)
committerbenthecarman <benthecarman@live.com>
Wed, 20 Sep 2023 18:08:40 +0000 (13:08 -0500)
src/bitcoind_client.rs

index a093e6d2eb7c6d1d85868b64c069181bada19759..3a41bc6e1ac74bb44cc2fec9a665e5a33d9f8c56 100644 (file)
@@ -33,19 +33,11 @@ pub struct BitcoindClient {
        port: u16,
        rpc_user: String,
        rpc_password: String,
-       fees: Arc<HashMap<Target, AtomicU32>>,
+       fees: Arc<HashMap<ConfirmationTarget, AtomicU32>>,
        handle: tokio::runtime::Handle,
        logger: Arc<FilesystemLogger>,
 }
 
-#[derive(Clone, Eq, Hash, PartialEq)]
-pub enum Target {
-       MempoolMinimum,
-       Background,
-       Normal,
-       HighPriority,
-}
-
 impl BlockSource for BitcoindClient {
        fn get_header<'a>(
                &'a self, header_hash: &'a BlockHash, height_hint: Option<u32>,
@@ -83,11 +75,11 @@ impl BitcoindClient {
                                std::io::Error::new(std::io::ErrorKind::PermissionDenied,
                                "Failed to make initial call to bitcoind - please check your RPC user/password and access settings")
                        })?;
-               let mut fees: HashMap<Target, AtomicU32> = HashMap::new();
-               fees.insert(Target::MempoolMinimum, AtomicU32::new(MIN_FEERATE));
-               fees.insert(Target::Background, AtomicU32::new(MIN_FEERATE));
-               fees.insert(Target::Normal, AtomicU32::new(2000));
-               fees.insert(Target::HighPriority, AtomicU32::new(5000));
+               let mut fees: HashMap<ConfirmationTarget, AtomicU32> = HashMap::new();
+               fees.insert(ConfirmationTarget::MempoolMinimum, AtomicU32::new(MIN_FEERATE));
+               fees.insert(ConfirmationTarget::Background, AtomicU32::new(MIN_FEERATE));
+               fees.insert(ConfirmationTarget::Normal, AtomicU32::new(2000));
+               fees.insert(ConfirmationTarget::HighPriority, AtomicU32::new(5000));
                let client = Self {
                        bitcoind_rpc_client: Arc::new(bitcoind_rpc_client),
                        host,
@@ -107,7 +99,7 @@ impl BitcoindClient {
        }
 
        fn poll_for_fee_estimates(
-               fees: Arc<HashMap<Target, AtomicU32>>, rpc_client: Arc<RpcClient>,
+               fees: Arc<HashMap<ConfirmationTarget, AtomicU32>>, rpc_client: Arc<RpcClient>,
                handle: tokio::runtime::Handle,
        ) {
                handle.spawn(async move {
@@ -171,14 +163,16 @@ impl BitcoindClient {
                                        }
                                };
 
-                               fees.get(&Target::MempoolMinimum)
+                               fees.get(&ConfirmationTarget::MempoolMinimum)
                                        .unwrap()
                                        .store(mempoolmin_estimate, Ordering::Release);
-                               fees.get(&Target::Background)
+                               fees.get(&ConfirmationTarget::Background)
                                        .unwrap()
                                        .store(background_estimate, Ordering::Release);
-                               fees.get(&Target::Normal).unwrap().store(normal_estimate, Ordering::Release);
-                               fees.get(&Target::HighPriority)
+                               fees.get(&ConfirmationTarget::Normal)
+                                       .unwrap()
+                                       .store(normal_estimate, Ordering::Release);
+                               fees.get(&ConfirmationTarget::HighPriority)
                                        .unwrap()
                                        .store(high_prio_estimate, Ordering::Release);
                                tokio::time::sleep(Duration::from_secs(60)).await;
@@ -267,20 +261,7 @@ impl BitcoindClient {
 
 impl FeeEstimator for BitcoindClient {
        fn get_est_sat_per_1000_weight(&self, confirmation_target: ConfirmationTarget) -> u32 {
-               match confirmation_target {
-                       ConfirmationTarget::MempoolMinimum => {
-                               self.fees.get(&Target::MempoolMinimum).unwrap().load(Ordering::Acquire)
-                       }
-                       ConfirmationTarget::Background => {
-                               self.fees.get(&Target::Background).unwrap().load(Ordering::Acquire)
-                       }
-                       ConfirmationTarget::Normal => {
-                               self.fees.get(&Target::Normal).unwrap().load(Ordering::Acquire)
-                       }
-                       ConfirmationTarget::HighPriority => {
-                               self.fees.get(&Target::HighPriority).unwrap().load(Ordering::Acquire)
-                       }
-               }
+               self.fees.get(&confirmation_target).unwrap().load(Ordering::Acquire)
        }
 }