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

This commit is contained in:
Jon Cinque 2023-05-18 23:31:02 +02:00 committed by GitHub
parent a544553ac7
commit 936d2cf3c2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 27 additions and 4 deletions

View File

@ -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)
}
}

View File

@ -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>;
}

View File

@ -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()?;