Pārlūkot izejas kodu

banks-client: Add `get_fee_for_message` (#31703)

Jon Cinque 2 gadi atpakaļ
vecāks
revīzija
936d2cf3c2

+ 25 - 2
banks-client/src/lib.rs

@@ -497,14 +497,37 @@ impl BanksClient {
             .map_err(Into::into)
     }
 
+    pub fn get_fee_for_message(
+        &mut self,
+        message: Message,
+    ) -> impl Future<Output = Result<Option<u64>, BanksClientError>> + '_ {
+        self.get_fee_for_message_with_commitment_and_context(
+            context::current(),
+            message,
+            CommitmentLevel::default(),
+        )
+    }
+
+    pub fn get_fee_for_message_with_commitment(
+        &mut self,
+        message: Message,
+        commitment: CommitmentLevel,
+    ) -> impl Future<Output = Result<Option<u64>, BanksClientError>> + '_ {
+        self.get_fee_for_message_with_commitment_and_context(
+            context::current(),
+            message,
+            commitment,
+        )
+    }
+
     pub fn get_fee_for_message_with_commitment_and_context(
         &mut self,
         ctx: Context,
-        commitment: CommitmentLevel,
         message: Message,
+        commitment: CommitmentLevel,
     ) -> impl Future<Output = Result<Option<u64>, BanksClientError>> + '_ {
         self.inner
-            .get_fee_for_message_with_commitment_and_context(ctx, commitment, message)
+            .get_fee_for_message_with_commitment_and_context(ctx, message, commitment)
             .map_err(Into::into)
     }
 }

+ 1 - 1
banks-interface/src/lib.rs

@@ -97,8 +97,8 @@ pub trait Banks {
         commitment: CommitmentLevel,
     ) -> Option<(Hash, u64)>;
     async fn get_fee_for_message_with_commitment_and_context(
-        commitment: CommitmentLevel,
         message: Message,
+        commitment: CommitmentLevel,
     ) -> Option<u64>;
 }
 

+ 1 - 1
banks-server/src/banks_server.rs

@@ -412,8 +412,8 @@ impl Banks for BanksServer {
     async fn get_fee_for_message_with_commitment_and_context(
         self,
         _: Context,
-        commitment: CommitmentLevel,
         message: Message,
+        commitment: CommitmentLevel,
     ) -> Option<u64> {
         let bank = self.bank(commitment);
         let sanitized_message = SanitizedMessage::try_from(message).ok()?;